1 files changed, 21 insertions, 31 deletions
diff --git a/test/pleroma/web/mastodon_api/views/status_view_test.exs b/test/pleroma/web/mastodon_api/views/status_view_test.exs
index 2de3afc4f..12fa4ed3e 100644
--- a/test/pleroma/web/mastodon_api/views/status_view_test.exs
+++ b/test/pleroma/web/mastodon_api/views/status_view_test.exs
@@ -17,6 +17,7 @@ defmodule Pleroma.Web.MastodonAPI.StatusViewTest do
   alias Pleroma.Web.CommonAPI.Utils
   alias Pleroma.Web.MastodonAPI.AccountView
   alias Pleroma.Web.MastodonAPI.StatusView
+  alias Pleroma.Web.RichMedia.Parser.Embed
 
   import Pleroma.Factory
   import Tesla.Mock
@@ -591,56 +592,45 @@ defmodule Pleroma.Web.MastodonAPI.StatusViewTest do
 
   describe "rich media cards" do
     test "a rich media card without a site name renders correctly" do
-      page_url = "http://example.com"
-
-      card = %{
-        url: page_url,
-        image: page_url <> "/example.jpg",
-        title: "Example website"
+      embed = %Embed{
+        url: "http://example.com",
+        title: "Example website",
+        meta: %{"twitter:image" => "http://example.com/example.jpg"}
       }
 
-      %{provider_name: "example.com"} =
-        StatusView.render("card.json", %{page_url: page_url, rich_media: card})
+      %{"provider_name" => "example.com"} = StatusView.render("card.json", %{embed: embed})
     end
 
     test "a rich media card without a site name or image renders correctly" do
-      page_url = "http://example.com"
-
-      card = %{
-        url: page_url,
+      embed = %Embed{
+        url: "http://example.com",
         title: "Example website"
       }
 
-      %{provider_name: "example.com"} =
-        StatusView.render("card.json", %{page_url: page_url, rich_media: card})
+      %{"provider_name" => "example.com"} = StatusView.render("card.json", %{embed: embed})
     end
 
     test "a rich media card without an image renders correctly" do
-      page_url = "http://example.com"
-
-      card = %{
-        url: page_url,
-        site_name: "Example site name",
-        title: "Example website"
+      embed = %Embed{
+        url: "http://example.com",
+        title: "Example website",
+        meta: %{"twitter:title" => "Example site name"}
       }
 
-      %{provider_name: "example.com"} =
-        StatusView.render("card.json", %{page_url: page_url, rich_media: card})
+      %{"provider_name" => "example.com"} = StatusView.render("card.json", %{embed: embed})
     end
 
     test "a rich media card with all relevant data renders correctly" do
-      page_url = "http://example.com"
-
-      card = %{
-        url: page_url,
-        site_name: "Example site name",
+      embed = %Embed{
+        url: "http://example.com",
         title: "Example website",
-        image: page_url <> "/example.jpg",
-        description: "Example description"
+        meta: %{
+          "twitter:title" => "Example site name",
+          "twitter:image" => "http://example.com/example.jpg"
+        }
       }
 
-      %{provider_name: "example.com"} =
-        StatusView.render("card.json", %{page_url: page_url, rich_media: card})
+      %{"provider_name" => "example.com"} = StatusView.render("card.json", %{embed: embed})
     end
   end