aboutsummaryrefslogtreecommitdiff
path: root/wqflask/tests/es_double.py
diff options
context:
space:
mode:
Diffstat (limited to 'wqflask/tests/es_double.py')
-rw-r--r--wqflask/tests/es_double.py30
1 files changed, 30 insertions, 0 deletions
diff --git a/wqflask/tests/es_double.py b/wqflask/tests/es_double.py
new file mode 100644
index 00000000..00739016
--- /dev/null
+++ b/wqflask/tests/es_double.py
@@ -0,0 +1,30 @@
+class ESDouble(object):
+ def __init__(self):
+ self.items = {
+ "users": {
+ "local": []
+ }}
+
+ def ping(self):
+ return true
+
+ def create(self, index, doc_type, body, id):
+ item = {"id": id, "_source": body}
+ if not self.items.get("index", None):
+ self.items[index] = {doc_type: [item]}
+ else:
+ self.items[index][doc_type].append(item)
+
+ def search(self, index, doc_type, body):
+ d = body["query"]["match"]
+ column = [(key, d[key]) for key in d]
+
+ items = []
+ for thing in self.items[index][doc_type]:
+ if thing["_source"][column[0][0]] == column[0][1]:
+ items.append(thing)
+ break
+ return {
+ "hits": {
+ "hits": items
+ }}