add support for expireat

mockredis/redis.py

@@ -125,6 +125,14 @@ def expire(self, key, seconds, currenttime=datetime.now()):
             return 1
         return 0
 
+    def expireat(self, key, when):
+        """Emulate expireat"""
+        expire_time = datetime.fromtimestamp(when)
+        if key in self.redis:
+            self.timeouts[key] = expire_time
+            return 1
+        return 0
+
     def ttl(self, key, currenttime=datetime.now()):
         """
         Emulate ttl

mockredis/tests/test_redis.py

@@ -120,6 +120,22 @@ def test_ttl_no_timeout(self):
         self.redis.set('key', 'key')
         self.assertEqual(self.redis.ttl('key'), None)
 
+    def test_expireat_calculates_time(self):
+        """
+        test whether expireat sets the correct ttl, setting a timestamp 30s in the future
+        """
+        import time
+        from datetime import datetime
+
+        self.redis.set('key', 'key')
+        self.redis.expireat('key', time.mktime(datetime.utcnow().timetuple()) + 30)
+
+        result = self.redis.ttl('key')
+        # should be an int
+        self.assertTrue(isinstance(result, int))
+        # should be less than the timeout originally set
+        self.assertTrue(result <= 30)
+
     def test_push_pop_returns_str(self):
         key = 'l'
         values = ['5', 5, [], {}]