#17 TestSort + add entry timestamp sort

pytion/query.py

@@ -118,10 +118,13 @@ def __repr__(self):
 class Sort(object):
     directions = ["ascending", "descending"]
 
-    def __init__(self, property_name: str, direction: str):
+    def __init__(self, property_name: str, direction: str = "ascending"):
         if direction not in self.directions:
             raise ValueError(f"Allowed types {self.directions} ({direction} is provided)")
-        self.sorts = [{"property": property_name, "direction": direction}]
+        if property_name in ("created_time", "last_edited_time"):
+            self.sorts = [{"timestamp": property_name, "direction": direction}]
+        else:
+            self.sorts = [{"property": property_name, "direction": direction}]
 
     def add(self, property_name: str, direction: str):
         if direction not in self.directions:

tests/test_query.py

@@ -200,5 +200,12 @@ def test_query__invalid_prop(self, little_database):
         with pytest.raises(ValidationError):
             little_database.db_query(sorts=s)
 
-    def test_query__timestamp(self):
-        pass
+    def test_query__timestamp(self, little_database):
+        s = Sort("last_edited_time")
+        r = little_database.db_query(sorts=s)
+        assert isinstance(r.obj[0], Page)
+        assert len(r.obj) == 4
+        assert str(r.obj[0]) == "wait, what?"
+        assert str(r.obj[2]) == "Parent testing page"
+        assert "friends" in str(r.obj[1])
+        assert bool(r.obj[3].title) is False