Use correct solr date format

MITLibraries · Jul 28, 2016 · 42fb222 · 42fb222
1 parent f180de3
commit 42fb222
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/kepler/tasks.py b/kepler/tasks.py
@@ -285,7 +285,7 @@ def _prep_solr_record(record):
             record[k] = list(v)
         else:
             try:
-                record[k] = v.isoformat()
+                record[k] = v.to('utc').format('YYYY-MM-DDTHH:mm:ss') + 'Z'
             except AttributeError:
                 pass
     return record

diff --git a/tests/test_tasks.py b/tests/test_tasks.py
@@ -4,14 +4,16 @@
 import os.path
 import re
 
+import arrow
 import pytest
 import requests_mock
 from mock import patch, DEFAULT
 
 from kepler.models import Job, Item
 from kepler.tasks import *
 from kepler.tasks import (_index_records, _load_marc_records,
-                          _upload_to_geoserver, _fgdc_to_mods)
+                          _prep_solr_record, _upload_to_geoserver,
+                          _fgdc_to_mods)
 
 
 pytestmark = pytest.mark.usefixtures('app')
@@ -281,3 +283,9 @@ def testFgdcToModsReturnsMods(bag_tif):
     fgdc = os.path.join(bag_tif, 'data/fgdc.xml')
     mods = _fgdc_to_mods(fgdc)
     assert u'<mods:title>Some land</mods:title>' in mods
+
+
+def test_prep_solr_record_converts_dates():
+    now = arrow.now()
+    rec = _prep_solr_record({'foo': now})
+    assert rec['foo'] == now.to('utc').format('YYYY-MM-DDTHH:mm:ss') + 'Z'