tests create_time attribute of archive

Author: aoberoi

opentok/archives.py

@@ -1,6 +1,9 @@
 from datetime import datetime, date
-from six import iteritems
+from six import iteritems, PY2, PY3
 import json
+import pytz
+if PY3:
+    from datetime import timezone
 
 dthandler = lambda obj: obj.isoformat() if isinstance(obj, datetime)  or isinstance(obj, date) else None
 
@@ -13,7 +16,10 @@ def __init__(self, sdk, values):
         self.status = values.get('status')
         self.session_id = values.get('sessionId')
         self.partner_id = values.get('partnerId')
-        self.created_at = datetime.fromtimestamp(values.get('createdAt') / 1000)
+        if PY2:
+            self.created_at = datetime.fromtimestamp(values.get('createdAt') / 1000, pytz.UTC)
+        if PY3:
+            self.created_at = datetime.fromtimestamp(values.get('createdAt') // 1000, timezone.utc)
         self.size = values.get('size')
         self.duration = values.get('duration')
         self.url = values.get('url')

tests/test_archive_api.py

@@ -5,6 +5,8 @@
 import httpretty
 import textwrap
 import json
+import datetime
+import pytz
 
 from opentok import OpenTok, Archive, __version__
 
@@ -13,9 +15,6 @@ def setUp(self):
         self.api_key = u('123456')
         self.api_secret = u('1234567890abcdef1234567890abcdef1234567890')
         self.session_id = u('SESSIONID')
-        # self.api_key = u('854511')
-        # self.api_secret = u('***REMOVED***')
-        # self.session_id = u('2_MX44NTQ1MTF-flR1ZSBOb3YgMTIgMDk6NDA6NTkgUFNUIDIwMTN-MC43NjU0Nzh-')
         self.opentok = OpenTok(self.api_key, self.api_secret)
 
     @httpretty.activate
@@ -56,7 +55,11 @@ def test_start_archive(self):
         expect(archive).to.have.property(u('status')).being.equal(u('started'))
         expect(archive).to.have.property(u('session_id')).being.equal(u('SESSIONID'))
         expect(archive).to.have.property(u('partner_id')).being.equal(123456)
-        #expect(archive).to.have.property(u('created_at')).being.equal(None)
+        if PY2:
+            created_at = datetime.datetime.fromtimestamp(1395183243, pytz.UTC)
+        if PY3:
+            created_at = datetime.datetime.fromtimestamp(1395183243, datetime.timezone.utc)
+        expect(archive).to.have.property(u('created_at')).being.equal(created_at)
         expect(archive).to.have.property(u('size')).being.equal(0)
         expect(archive).to.have.property(u('duration')).being.equal(0)
         expect(archive).to.have.property(u('url')).being.equal(None)