bugfix: query results failed due to wrong using period

[releng.git] / utils / test / result_collection_api / tests / unit / fake_pymongo.py

diff --git a/utils/test/result_collection_api/tests/unit/fake_pymongo.py b/utils/test/result_collection_api/tests/unit/fake_pymongo.py
index 95c7371..bebb9e8 100644 (file)
--- a/utils/test/result_collection_api/tests/unit/fake_pymongo.py
+++ b/utils/test/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