Some tests for media_data, yeah!

One in the submissions, that posts a jpg with gps data.
One in the modelmethods to test some behaviours.

The later test fails.
Fixes coming up.

1 files changed, 17 insertions, 0 deletions

diff --git a/mediagoblin/tests/test_modelmethods.py b/mediagoblin/tests/test_modelmethods.py
index c1064d3a..7719bd97 100644
--- a/mediagoblin/tests/test_modelmethods.py
+++ b/mediagoblin/tests/test_modelmethods.py
@@ -17,7 +17,9 @@
 # Maybe not every model needs a test, but some models have special
 # methods, and so it makes sense to test them here.
 
+from nose.tools import assert_equal
 
+from mediagoblin.db.base import Session
 from mediagoblin.db.models import MediaEntry
 
 from mediagoblin.tests.tools import get_app, \
@@ -128,3 +130,18 @@ class TestMediaEntrySlugs(object):
             u"@!#?@!", save=False)
         qbert_entry.generate_slug()
         assert qbert_entry.slug is None
+
+
+def test_media_data_init():
+    Session.rollback()
+    Session.remove()
+    media = MediaEntry()
+    media.media_type = u"mediagoblin.media_types.image"
+    assert media.media_data is None
+    media.media_data_init()
+    assert media.media_data is not None
+    obj_in_session = 0
+    for obj in Session():
+        obj_in_session += 1
+        print repr(obj)
+    assert_equal(obj_in_session, 0)