add sort tests

Author: rharlev

tests/test_jsonbin.py

@@ -1,7 +1,8 @@
 import unittest
+import uuid
 from jsonbox import JsonBox
 
-TEST_BOX_ID = "test_gywvcu8ew7t7gascbascbuwd"
+TEST_BOX_ID = str(uuid.uuid4()).replace("-", "_")
 TEST_COLLECTION_ID = "collection_427453"
 TEST_RECORD_ID = "test_sjdgfygsf2347623564twfgyu"
 TEST_DATA_KEY_1 = "aaa"
@@ -14,6 +15,20 @@ class TestJsonBox(unittest.TestCase):
     def setUp(self):
         self.jb = JsonBox()
 
+    def test_get_record_id_list(self):
+        data = [{"aaa": "bbb", self.jb.RECORD_ID_KEY: 1}, {"aaa": "bbb", self.jb.RECORD_ID_KEY: 2}]
+        ids = self.jb.get_record_id(data)
+
+        self.assertIsNotNone(ids)
+        self.assertEqual(ids, [1, 2])
+
+    def test_get_record_id_single(self):
+        data = {"aaa": "bbb", self.jb.RECORD_ID_KEY: 1}
+        ids = self.jb.get_record_id(data)
+
+        self.assertIsNotNone(ids)
+        self.assertEqual(ids, 1,)
+
     def test_read_record(self):
         json_data = self.jb.read(TEST_BOX_ID, "5d80031fca4f06001791fb28")
         self.assertIsNotNone(json_data)
@@ -25,7 +40,20 @@ def test_read_box(self):
         self.assertIsNotNone(json_data)
         self.assertTrue(isinstance(json_data, list))
 
-        print(json_data)
+    def test_read_sort(self):
+        data = [{"name": "first", "age": "25"}, {"name": "second", "age": "19"}]
+        box_id = TEST_BOX_ID + "_sort"
+        result = self.jb.write(data, box_id)
+        record_ids = self.jb.get_record_id(result)
+
+        json_data = self.jb.read(box_id, sort_by="age")
+        self.assertIsNotNone(json_data)
+        self.assertTrue(isinstance(json_data, list))
+        self.assertEqual(json_data[0]["name"], "second")
+
+        # cleanup
+        for record_id in record_ids:
+            self.jb.delete(box_id, record_id)
 
     def test_write_box(self):
         data = {
@@ -35,8 +63,6 @@ def test_write_box(self):
         self.assertIsNotNone(json_data)
         self.assertEqual(json_data[TEST_DATA_KEY_1], TEST_DATA_VALUE_1)
 
-        print(json_data)
-
     def test_write_box_collection(self):
         data = {
             TEST_DATA_KEY_1: TEST_DATA_VALUE_1
@@ -45,8 +71,6 @@ def test_write_box_collection(self):
         self.assertIsNotNone(json_data)
         self.assertEqual(json_data[TEST_DATA_KEY_1], TEST_DATA_VALUE_1)
 
-        print(json_data)
-
     def test_update(self):
         data = {
             TEST_DATA_KEY_1: TEST_DATA_VALUE_1
@@ -70,8 +94,6 @@ def test_update(self):
         self.assertIsNotNone(json_data)
         self.assertEqual(json_data[TEST_DATA_KEY_2], TEST_DATA_VALUE_2)
 
-        print(json_data)
-
     def test_delete(self):
         data = {
             TEST_DATA_KEY_1: TEST_DATA_VALUE_1
@@ -85,10 +107,8 @@ def test_delete(self):
         json_data = self.jb.delete(TEST_BOX_ID, record_id)
         self.assertIsNotNone(json_data)
 
-        reuslt = self.jb.read(TEST_BOX_ID, record_id)
-        self.assertFalse(reuslt)
-
-        print(json_data)
+        result = self.jb.read(TEST_BOX_ID, record_id)
+        self.assertFalse(result)
 
 
 if __name__ == '__main__':