1 files changed, 22 insertions, 8 deletions
diff --git a/test/unit/model_test.rb b/test/unit/model_test.rb
index b7c5333c4d..bdfe88adc8 100644
--- a/test/unit/model_test.rb
+++ b/test/unit/model_test.rb
@@ -1,10 +1,24 @@
-require_relative '../test_helper'
-
-module ActionText
-  class ModelTest < ActiveSupport::TestCase
-    test "saving content" do
-      message = Message.create!(subject: "Greetings", content: "<h1>Hello world</h1>")
-      assert_equal "Hello world", message.content.body.to_plain_text
-    end
+require 'test_helper'
+
+class ActionText::ModelTest < ActiveSupport::TestCase
+  test "plain text conversion" do
+    message = Message.new(subject: "Greetings", content: "<h1>Hello world</h1>")
+    assert_equal "Hello world", message.content.body.to_plain_text
+  end
+
+  test "without content" do
+    message = Message.create!(subject: "Greetings")
+    assert message.content.body.nil?
+  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
+
+  test "saving content" do
+    message = Message.create!(subject: "Greetings", content: "<h1>Hello world</h1>")
+    assert_equal "Hello world", message.content.body.to_plain_text
   end
 end