Prefer the ASt API

3 files changed, 13 insertions, 0 deletions

diff --git a/test/fixtures/files/racecar.jpg b/test/fixtures/files/racecar.jpg
new file mode 100644
index 0000000000..934b4caa22
--- /dev/null
+++ b/test/fixtures/files/racecar.jpg
diff --git a/test/test_helper.rb b/test/test_helper.rb
index 94fec6b8be..79fde51549 100644
--- a/test/test_helper.rb
+++ b/test/test_helper.rb
@@ -19,3 +19,10 @@ if ActiveSupport::TestCase.respond_to?(:fixture_path=)
   ActiveSupport::TestCase.file_fixture_path = ActiveSupport::TestCase.fixture_path + "/files"
   ActiveSupport::TestCase.fixtures :all
 end
+
+class ActiveSupport::TestCase
+  private
+    def create_file_blob(filename:, content_type:, metadata: nil)
+      ActiveStorage::Blob.create_after_upload! io: file_fixture(filename).open, filename: filename, content_type: content_type, metadata: metadata
+    end
+end
diff --git a/test/unit/content_test.rb b/test/unit/content_test.rb
index b1a06e2cfa..f1acf054db 100644
--- a/test/unit/content_test.rb
+++ b/test/unit/content_test.rb
@@ -6,5 +6,11 @@ module ActionText
       message = Message.new(subject: "Greetings", content: "<h1>Hello world</h1>")
       assert_equal "Hello world", message.content.body.to_plain_text
     end
+
+    test "embed extraction" do
+      blob = create_file_blob(filename: "racecar.jpg", content_type: "image/jpg")
+      message = Message.create!(subject: "Greetings", content: ActionText::Content.new("Hello world").append_attachables(blob))
+      assert_equal "racecar.jpg", message.content.embeds.first.filename.to_s
+    end
   end
 end