From da342687570cd53f2bfb0e1d69db77b1ba664801 Mon Sep 17 00:00:00 2001 From: SerenaFeng Date: Tue, 31 May 2016 23:18:46 +0800 Subject: bugfix: query results failed due to wrong using period JIRA: FUNCTEST-281 Change-Id: I3d0bdc302afc45593ad3feaec21b89d5c6eeaa0b Signed-off-by: SerenaFeng --- result_collection_api/tests/unit/fake_pymongo.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) (limited to 'result_collection_api/tests/unit/fake_pymongo.py') diff --git a/result_collection_api/tests/unit/fake_pymongo.py b/result_collection_api/tests/unit/fake_pymongo.py index 95c7371..bebb9e8 100644 --- a/result_collection_api/tests/unit/fake_pymongo.py +++ b/result_collection_api/tests/unit/fake_pymongo.py @@ -73,11 +73,24 @@ class MemDb(object): def insert(self, doc_or_docs, check_keys=True): return thread_execute(self._insert, doc_or_docs, check_keys) + @staticmethod + def _compare_date(spec, value): + for k, v in spec.iteritems(): + if k == '$gte' and value >= v: + return True + return False + @staticmethod def _in(content, *args): for arg in args: for k, v in arg.iteritems(): - if k != 'creation_date' and content.get(k, None) != v: + if k == 'start_date': + if not MemDb._compare_date(v, content.get(k)): + return False + elif k == 'trust_indicator': + if float(content.get(k)) != float(v): + return False + elif content.get(k, None) != v: return False return True -- cgit 1.2.3-korg