git.squeep.com Git - akkoma/blob - test/support/factory.ex

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Factory do
   6   use ExMachina.Ecto, repo: Pleroma.Repo
   7   alias Pleroma.Object
   8   alias Pleroma.User
   9
  10   def participation_factory do
  11     conversation = insert(:conversation)
  12     user = insert(:user)
  13
  14     %Pleroma.Conversation.Participation{
  15       conversation: conversation,
  16       user: user,
  17       read: false
  18     }
  19   end
  20
  21   def conversation_factory do
  22     %Pleroma.Conversation{
  23       ap_id: sequence(:ap_id, &"https://some_conversation/#{&1}")
  24     }
  25   end
  26
  27   def user_factory do
  28     user = %User{
  29       name: sequence(:name, &"Test テスト User #{&1}"),
  30       email: sequence(:email, &"user#{&1}@example.com"),
  31       nickname: sequence(:nickname, &"nick#{&1}"),
  32       password_hash: Comeonin.Pbkdf2.hashpwsalt("test"),
  33       bio: sequence(:bio, &"Tester Number #{&1}"),
  34       info: %{},
  35       last_digest_emailed_at: NaiveDateTime.utc_now()
  36     }
  37
  38     %{
  39       user
  40       | ap_id: User.ap_id(user),
  41         follower_address: User.ap_followers(user),
  42         following_address: User.ap_following(user)
  43     }
  44   end
  45
  46   def note_factory(attrs \\ %{}) do
  47     text = sequence(:text, &"This is :moominmamma: note #{&1}")
  48
  49     user = attrs[:user] || insert(:user)
  50
  51     data = %{
  52       "type" => "Note",
  53       "content" => text,
  54       "id" => Pleroma.Web.ActivityPub.Utils.generate_object_id(),
  55       "actor" => user.ap_id,
  56       "to" => ["https://www.w3.org/ns/activitystreams#Public"],
  57       "published" => DateTime.utc_now() |> DateTime.to_iso8601(),
  58       "likes" => [],
  59       "like_count" => 0,
  60       "context" => "2hu",
  61       "summary" => "2hu",
  62       "tag" => ["2hu"],
  63       "emoji" => %{
  64         "2hu" => "corndog.png"
  65       }
  66     }
  67
  68     %Pleroma.Object{
  69       data: merge_attributes(data, Map.get(attrs, :data, %{}))
  70     }
  71   end
  72
  73   def audio_factory(attrs \\ %{}) do
  74     text = sequence(:text, &"lain radio episode #{&1}")
  75
  76     user = attrs[:user] || insert(:user)
  77
  78     data = %{
  79       "type" => "Audio",
  80       "id" => Pleroma.Web.ActivityPub.Utils.generate_object_id(),
  81       "artist" => "lain",
  82       "title" => text,
  83       "album" => "lain radio",
  84       "to" => ["https://www.w3.org/ns/activitystreams#Public"],
  85       "published" => DateTime.utc_now() |> DateTime.to_iso8601(),
  86       "actor" => user.ap_id,
  87       "length" => 180_000
  88     }
  89
  90     %Pleroma.Object{
  91       data: merge_attributes(data, Map.get(attrs, :data, %{}))
  92     }
  93   end
  94
  95   def listen_factory do
  96     audio = insert(:audio)
  97
  98     data = %{
  99       "id" => Pleroma.Web.ActivityPub.Utils.generate_activity_id(),
 100       "type" => "Listen",
 101       "actor" => audio.data["actor"],
 102       "to" => audio.data["to"],
 103       "object" => audio.data,
 104       "published" => audio.data["published"]
 105     }
 106
 107     %Pleroma.Activity{
 108       data: data,
 109       actor: data["actor"],
 110       recipients: data["to"]
 111     }
 112   end
 113
 114   def direct_note_factory do
 115     user2 = insert(:user)
 116
 117     %Pleroma.Object{data: data} = note_factory()
 118     %Pleroma.Object{data: Map.merge(data, %{"to" => [user2.ap_id]})}
 119   end
 120
 121   def article_factory do
 122     note_factory()
 123     |> Map.put("type", "Article")
 124   end
 125
 126   def tombstone_factory do
 127     data = %{
 128       "type" => "Tombstone",
 129       "id" => Pleroma.Web.ActivityPub.Utils.generate_object_id(),
 130       "formerType" => "Note",
 131       "deleted" => DateTime.utc_now() |> DateTime.to_iso8601()
 132     }
 133
 134     %Pleroma.Object{
 135       data: data
 136     }
 137   end
 138
 139   def direct_note_activity_factory do
 140     dm = insert(:direct_note)
 141
 142     data = %{
 143       "id" => Pleroma.Web.ActivityPub.Utils.generate_activity_id(),
 144       "type" => "Create",
 145       "actor" => dm.data["actor"],
 146       "to" => dm.data["to"],
 147       "object" => dm.data,
 148       "published" => DateTime.utc_now() |> DateTime.to_iso8601(),
 149       "context" => dm.data["context"]
 150     }
 151
 152     %Pleroma.Activity{
 153       data: data,
 154       actor: data["actor"],
 155       recipients: data["to"]
 156     }
 157   end
 158
 159   def note_activity_factory(attrs \\ %{}) do
 160     user = attrs[:user] || insert(:user)
 161     note = attrs[:note] || insert(:note, user: user)
 162
 163     data_attrs = attrs[:data_attrs] || %{}
 164     attrs = Map.drop(attrs, [:user, :note, :data_attrs])
 165
 166     data =
 167       %{
 168         "id" => Pleroma.Web.ActivityPub.Utils.generate_activity_id(),
 169         "type" => "Create",
 170         "actor" => note.data["actor"],
 171         "to" => note.data["to"],
 172         "object" => note.data["id"],
 173         "published" => DateTime.utc_now() |> DateTime.to_iso8601(),
 174         "context" => note.data["context"]
 175       }
 176       |> Map.merge(data_attrs)
 177
 178     %Pleroma.Activity{
 179       data: data,
 180       actor: data["actor"],
 181       recipients: data["to"]
 182     }
 183     |> Map.merge(attrs)
 184   end
 185
 186   defp expiration_offset_by_minutes(attrs, minutes) do
 187     scheduled_at =
 188       NaiveDateTime.utc_now()
 189       |> NaiveDateTime.add(:timer.minutes(minutes), :millisecond)
 190       |> NaiveDateTime.truncate(:second)
 191
 192     %Pleroma.ActivityExpiration{}
 193     |> Map.merge(attrs)
 194     |> Map.put(:scheduled_at, scheduled_at)
 195   end
 196
 197   def expiration_in_the_past_factory(attrs \\ %{}) do
 198     expiration_offset_by_minutes(attrs, -60)
 199   end
 200
 201   def expiration_in_the_future_factory(attrs \\ %{}) do
 202     expiration_offset_by_minutes(attrs, 61)
 203   end
 204
 205   def article_activity_factory do
 206     article = insert(:article)
 207
 208     data = %{
 209       "id" => Pleroma.Web.ActivityPub.Utils.generate_activity_id(),
 210       "type" => "Create",
 211       "actor" => article.data["actor"],
 212       "to" => article.data["to"],
 213       "object" => article.data,
 214       "published" => DateTime.utc_now() |> DateTime.to_iso8601(),
 215       "context" => article.data["context"]
 216     }
 217
 218     %Pleroma.Activity{
 219       data: data,
 220       actor: data["actor"],
 221       recipients: data["to"]
 222     }
 223   end
 224
 225   def announce_activity_factory(attrs \\ %{}) do
 226     note_activity = attrs[:note_activity] || insert(:note_activity)
 227     user = attrs[:user] || insert(:user)
 228
 229     data = %{
 230       "type" => "Announce",
 231       "actor" => note_activity.actor,
 232       "object" => note_activity.data["id"],
 233       "to" => [user.follower_address, note_activity.data["actor"]],
 234       "cc" => ["https://www.w3.org/ns/activitystreams#Public"],
 235       "context" => note_activity.data["context"]
 236     }
 237
 238     %Pleroma.Activity{
 239       data: data,
 240       actor: user.ap_id,
 241       recipients: data["to"]
 242     }
 243   end
 244
 245   def like_activity_factory(attrs \\ %{}) do
 246     note_activity = attrs[:note_activity] || insert(:note_activity)
 247     object = Object.normalize(note_activity)
 248     user = insert(:user)
 249
 250     data =
 251       %{
 252         "id" => Pleroma.Web.ActivityPub.Utils.generate_activity_id(),
 253         "actor" => user.ap_id,
 254         "type" => "Like",
 255         "object" => object.data["id"],
 256         "published_at" => DateTime.utc_now() |> DateTime.to_iso8601()
 257       }
 258       |> Map.merge(attrs[:data_attrs] || %{})
 259
 260     %Pleroma.Activity{
 261       data: data
 262     }
 263   end
 264
 265   def follow_activity_factory do
 266     follower = insert(:user)
 267     followed = insert(:user)
 268
 269     data = %{
 270       "id" => Pleroma.Web.ActivityPub.Utils.generate_activity_id(),
 271       "actor" => follower.ap_id,
 272       "type" => "Follow",
 273       "object" => followed.ap_id,
 274       "published_at" => DateTime.utc_now() |> DateTime.to_iso8601()
 275     }
 276
 277     %Pleroma.Activity{
 278       data: data,
 279       actor: follower.ap_id
 280     }
 281   end
 282
 283   def oauth_app_factory do
 284     %Pleroma.Web.OAuth.App{
 285       client_name: "Some client",
 286       redirect_uris: "https://example.com/callback",
 287       scopes: ["read", "write", "follow", "push"],
 288       website: "https://example.com",
 289       client_id: Ecto.UUID.generate(),
 290       client_secret: "aaa;/&bbb"
 291     }
 292   end
 293
 294   def instance_factory do
 295     %Pleroma.Instances.Instance{
 296       host: "domain.com",
 297       unreachable_since: nil
 298     }
 299   end
 300
 301   def oauth_token_factory do
 302     oauth_app = insert(:oauth_app)
 303
 304     %Pleroma.Web.OAuth.Token{
 305       token: :crypto.strong_rand_bytes(32) |> Base.url_encode64(),
 306       scopes: ["read"],
 307       refresh_token: :crypto.strong_rand_bytes(32) |> Base.url_encode64(),
 308       user: build(:user),
 309       app_id: oauth_app.id,
 310       valid_until: NaiveDateTime.add(NaiveDateTime.utc_now(), 60 * 10)
 311     }
 312   end
 313
 314   def oauth_authorization_factory do
 315     %Pleroma.Web.OAuth.Authorization{
 316       token: :crypto.strong_rand_bytes(32) |> Base.url_encode64(padding: false),
 317       scopes: ["read", "write", "follow", "push"],
 318       valid_until: NaiveDateTime.add(NaiveDateTime.utc_now(), 60 * 10),
 319       user: build(:user),
 320       app: build(:oauth_app)
 321     }
 322   end
 323
 324   def push_subscription_factory do
 325     %Pleroma.Web.Push.Subscription{
 326       user: build(:user),
 327       token: build(:oauth_token),
 328       endpoint: "https://example.com/example/1234",
 329       key_auth: "8eDyX_uCN0XRhSbY5hs7Hg==",
 330       key_p256dh:
 331         "BCIWgsnyXDv1VkhqL2P7YRBvdeuDnlwAPT2guNhdIoW3IP7GmHh1SMKPLxRf7x8vJy6ZFK3ol2ohgn_-0yP7QQA=",
 332       data: %{}
 333     }
 334   end
 335
 336   def notification_factory do
 337     %Pleroma.Notification{
 338       user: build(:user)
 339     }
 340   end
 341
 342   def scheduled_activity_factory do
 343     %Pleroma.ScheduledActivity{
 344       user: build(:user),
 345       scheduled_at: NaiveDateTime.add(NaiveDateTime.utc_now(), :timer.minutes(60), :millisecond),
 346       params: build(:note) |> Map.from_struct() |> Map.get(:data)
 347     }
 348   end
 349
 350   def registration_factory do
 351     user = insert(:user)
 352
 353     %Pleroma.Registration{
 354       user: user,
 355       provider: "twitter",
 356       uid: "171799000",
 357       info: %{
 358         "name" => "John Doe",
 359         "email" => "john@doe.com",
 360         "nickname" => "johndoe",
 361         "description" => "My bio"
 362       }
 363     }
 364   end
 365
 366   def config_factory do
 367     %Pleroma.Web.AdminAPI.Config{
 368       key: sequence(:key, &"some_key_#{&1}"),
 369       group: "pleroma",
 370       value:
 371         sequence(
 372           :value,
 373           fn key ->
 374             :erlang.term_to_binary(%{another_key: "#{key}somevalue", another: "#{key}somevalue"})
 375           end
 376         )
 377     }
 378   end
 379
 380   def marker_factory do
 381     %Pleroma.Marker{
 382       user: build(:user),
 383       timeline: "notifications",
 384       lock_version: 0,
 385       last_read_id: "1"
 386     }
 387   end
 388 end