Support for multiple pushes from last_push

Author: otobrglez

databox test/test_push.py

@@ -49,6 +49,10 @@ def test_last_push(self):
 
         assert self.client.last_push()['err'] == []
 
+    def test_last_push_with_number(self):
+        self.client._push_json = lambda data=None, path='/': path
+        assert self.client.last_push(3) == '/lastpushes/3'
+
     def test_short(self):
         Client._push_json = mock_push_json
 

databox/__init__.py

@@ -68,8 +68,8 @@ def insert_all(self, rows):
             'data': [self.process_kpi(**row) for row in rows]
         })['status'] == 'ok'
 
-    def last_push(self):
-        return self._push_json(path='/lastpushes/1')
+    def last_push(self, number=1):
+        return self._push_json(path='/lastpushes/{n}'.format(**{'n': number}))
 
 
 def push(key, value, date=None, token=None):

example.py

@@ -37,5 +37,5 @@
     }},
 ])
 
-print client.last_push()
+print client.last_push(3)