Add tiff_url to Item model

MITLibraries · Oct 29, 2015 · 03b0bec · 03b0bec
1 parent 6b994ef
commit 03b0bec
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 2 deletions.
diff --git a/kepler/models.py b/kepler/models.py
@@ -40,6 +40,7 @@ class Item(db.Model):
     handle = db.Column(db.Unicode(255))
     access = db.Column(db.Enum(u'Public', u'Restricted'), default=u'Public')
     jobs = db.relationship('Job', backref='item', lazy='dynamic')
+    tiff_url = db.Column(db.Unicode(255))
 
     def __repr__(self):
         return '<Item #%d: %r>' % (self.id, self.uri)
diff --git a/kepler/tasks.py b/kepler/tasks.py
@@ -102,7 +102,7 @@ def get_geotiff_url_from_dspace(job):
     if len(tif_urls) != 1:
         raise Exception('Expected 1 tiff, found ' + str(len(tif_urls)))
     else:
-        job.item.geotiff_url = tif_urls[0]
+        job.item.tiff_url = tif_urls[0]
 
     db.session.commit()
 

diff --git a/tests/test_models.py b/tests/test_models.py
@@ -84,3 +84,7 @@ def testItemHasLayerId(self, db):
     def testItemHasHandle(self, db):
         item = Item(handle='http://hdl.handle.net/123456789/3')
         assert item.handle == 'http://hdl.handle.net/123456789/3'
+
+    def testItemHasTiffUrl(self, db):
+        item = Item(tiff_url='http://example.com/bitstream/handle/1234.5/67890/248077.tif?sequence=4')
+        assert item.tiff_url == 'http://example.com/bitstream/handle/1234.5/67890/248077.tif?sequence=4'
diff --git a/tests/test_tasks.py b/tests/test_tasks.py
@@ -63,7 +63,7 @@ def testGetGeotiffUrlFromDspaceAddsGeotiffUrlToItem(job, oai_ore):
               text=oai_ore)
         job.item.handle = 'http://hdl.handle.net/1234.5/67890'
         get_geotiff_url_from_dspace(job)
-        assert job.item.geotiff_url == 'http://example.com/bitstream/handle/1234.5/67890/248077.tif?sequence=4'
+        assert job.item.tiff_url == 'http://example.com/bitstream/handle/1234.5/67890/248077.tif?sequence=4'
 
 
 def testGetGeotiffUrlFromDspaceErrorsOnNoTiffs(job, oai_ore_no_tiffs):