Add test for fetch_many

redis_tasks/task.py

@@ -268,7 +268,6 @@ def fetch_many(cls, task_ids):
         return tasks
 
     def refresh(self, data=None):
-        key = self.key
         if not data:
             data = connection.hgetall(self.key)
         obj = {k.decode(): v for k, v in data.items()}

tests/test_task.py

@@ -308,3 +308,17 @@ def test_init_save_fetch_delete(connection, assert_atomic, stub):
     assert not connection.exists(t.key)
     with pytest.raises(TaskDoesNotExist):
         Task.fetch(t.id)
+
+
+def test_fetch_many(stub):
+    tasks_data = [{"args": []}, {"args": ["foo"]}, {"args": ["bar"]},
+                  {"args": ["foo", "bar"]}, {"args": []}]
+    tasks = [Task(stub, **d) for d in tasks_data]
+    for t in tasks:
+        t._save()
+
+    fetched = Task.fetch_many([t.id for t in tasks[1:4]])
+    assert len(fetched) == 3
+    for i, task in enumerate(fetched):
+        assert task.id == tasks[i+1].id
+        assert task.args == tasks_data[i+1]["args"]