Add test for value not found in ES backend

Mathieu Chataigner · Mathieu Chataigner · commit 6c098b5c3e8a · 2020-06-01T19:43:31.000+02:00
diff --git a/t/unit/backends/test_elasticsearch.py b/t/unit/backends/test_elasticsearch.py
@@ -60,6 +60,17 @@ def test_get_none(self):
             index=x.index,
         )
 
+    def test_get_task_not_found(self):
+        x = ElasticsearchBackend(app=self.app)
+        x._server = Mock()
+        x._server.get.side_effect = [
+            exceptions.NotFoundError(404, '{"_index":"celery","_type":"_doc","_id":"toto","found":false}',
+                                     {'_index': 'celery', '_type': '_doc', '_id': 'toto', 'found': False})
+        ]
+
+        res = x.get(sentinel.task_id)
+        assert res == None
+
     def test_delete(self):
         x = ElasticsearchBackend(app=self.app)
         x._server = Mock()
