1 files changed, 34 insertions, 1 deletions

diff --git a/test/models/asset_test.rb b/test/models/asset_test.rb
index a1041e4..ab1cc5d 100644
--- a/test/models/asset_test.rb
+++ b/test/models/asset_test.rb
@@ -19,5 +19,38 @@ class AssetTest < ActiveSupport::TestCase
     assert_equal 0, draft.assets.length
     assert_equal 0, RelatedAsset.count
   end
-  
+
+  test "image? is true for supported image content types" do
+    assert Asset.new(:upload_content_type => "image/png").image?
+    assert Asset.new(:upload_content_type => "image/jpeg").image?
+  end
+
+  test "image? is false for non-image content types" do
+    assert_not Asset.new(:upload_content_type => "application/pdf").image?
+    assert_not Asset.new(:upload_content_type => nil).image?
+  end
+
+  test "license_key must be a known dictionary key" do
+    asset = Asset.new(:license_key => "not_a_real_license")
+    I18n.with_locale(:en) do
+      assert_not asset.valid?
+      assert_includes asset.errors[:license_key], "is not included in the list"
+    end
+  end
+
+  test "license_key may be blank" do
+    assert Asset.new(:license_key => nil).valid?
+  end  
+
+  test "pdf? is true only for application/pdf" do
+    assert Asset.new(:upload_content_type => "application/pdf").pdf?
+    assert_not Asset.new(:upload_content_type => "image/png").pdf?
+  end
+
+  test "show_credit? is false for a PDF even with every credit field present" do
+    asset = Asset.new(:name => "demo", :upload_content_type => "application/pdf",
+                       :creator => "Jane Doe", :source_url => "https://example.org", :license_key => "cc_by_4")
+    assert asset.has_credit?
+    assert_not asset.show_credit?
+  end
 end