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