add relevant consts to consts and use them in test_esutils

avihad · Jul 3, 2016 · 86b4401 · 86b4401
1 parent a8f1579
commit 86b4401
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/tests/test_esutils.py b/tests/test_esutils.py
@@ -47,5 +47,6 @@ def create_results(result_hits, failures=0):
                 }}
 
     def test_is_get_query_with_results_valid_response_returns_true(self):
-        es_get_response = {"_type": "type", "_index": "index", "fields": {}, "_version": 1, "found": True, "_id": "id"}
+        es_get_response = {EsConst._TYPE: "type", EsConst._INDEX: "index", EsConst.FIELDS: {},
+                           EsConst._VERSION: 1, EsConst.FOUND: True, EsConst._ID: "id"}
         self.assertTrue(EsUtils.is_get_query_with_results(es_get_response))
diff --git a/twistes/consts.py b/twistes/consts.py
@@ -30,6 +30,10 @@ class EsConst(object):
     TOTAL = 'total'
     FAILED = 'failed'
     SHARDS = '_shards'
+    _INDEX = "_index"
+    _TYPE = "_type"
+    _VERSION = "_version"
+    _ID = "_id"
     ALL_VALUES = '_all'
     FIELDS = 'fields'
     SCROLL_ID = 'scroll_id'