fix tests

Author: robswc

src/quickbase_json/qb_response.py

@@ -22,19 +22,20 @@ def __init__(self, response_type, **kwargs):
             self.update(kwargs.get('sample_data'))
         super().__init__()
 
-    def info(self):
+    def info(self, prt=True):
         """
         Prints information about the response.
+        :prt: Set to False to only grab the return as a string.
         """
         if self.response_type == 'records':
-
-            print(f'{Bcolors.OKBLUE}Sample Data:\n')
+            info = []
+            info.append(f'{Bcolors.OKBLUE}Sample Data:\n')
             try:
-                print('\t', self.get('data')[0], '\n')
+                info.append('\t' + str(self.get('data')[0]) + '\n\n')
             except KeyError as e:
-                print('\t', self.get('data'), '\n')
-            print(Bcolors.ENDC)
-            print(f'{Bcolors.OKGREEN}Fields:\n')
+                info.append('\t' + str(self.get('data')) + '\n')
+            info.append(Bcolors.ENDC)
+            info.append(f'{Bcolors.OKGREEN}Fields:\n')
 
             fields_info = []
             for field in self.get('fields'):
@@ -46,13 +47,20 @@ def info(self):
 
             # print field info
             for fi in fields_info:
-                print('\t', '{:<16s} {:<32s} {:<16s}'.format(fi[0], fi[1], fi[2]))
-            print(Bcolors.ENDC)
+                info.append('\t' + '{:<16s} {:<32s} {:<16s}\n'.format(fi[0], fi[1], fi[2]))
+            info.append(Bcolors.ENDC)
 
-            print(f'\n{Bcolors.OKCYAN}Metadata:\n')
+            info.append(f'\n{Bcolors.OKCYAN}Metadata:\n')
             for k, v in self.get('metadata').items():
-                print('\t{:<16s} {:<16s}'.format(k, str(v)))
-            print(Bcolors.ENDC)
+                info.append('\t{:<16s} {:<16s}\n'.format(k, str(v)))
+            info.append(Bcolors.ENDC)
+
+            info_str = ''.join(info)
+
+            if prt:
+                print(info_str)
+
+            return info_str
 
     def fields(self, prop):
         """

tests/sample_data.py

@@ -35,4 +35,29 @@
         "numFields": 3,
         "skip": 0
     }
-}
+}
+
+record_data_load = {'data': [{'6': {'value': 'Andre Harris'},
+                              '7': {'value': 10},
+                              '8': {'value': '2019-12-18T08:00:00.000Z'}}],
+                    'fields': [{'id': 6, 'label': 'Full Name', 'type': 'text'},
+                               {'id': 7, 'label': 'Amount', 'type': 'numeric'},
+                               {'id': 8, 'label': 'Date time', 'type': 'date time'}],
+                    'metadata': {'numFields': 3, 'numRecords': 1, 'skip': 0, 'totalRecords': 10}}
+
+record_data_get_info = """
+Sample Data:
+	{'6': {'value': 'Andre Harris'}, '7': {'value': 10}, '8': {'value': '2019-12-18T08:00:00.000Z'}}
+
+Fields:
+	id: 6            label: Full Name                 type: text      
+	id: 7            label: Amount                    type: numeric   
+	id: 8            label: Date time                 type: date time 
+
+Metadata:
+	totalRecords     10              
+	numRecords       1               
+	numFields        3               
+	skip             0               
+
+"""

tests/test_client.py renamed to tests/test_helpers.py

@@ -1,12 +1,7 @@
-
-
 import pytest
 
-
-
-
 # test where
-from quickbase_json.helpers import Where, IncorrectParameters
+from src.quickbase_json.helpers import Where, IncorrectParameters
 
 
 @pytest.mark.parametrize('fid, operator, value, expected', [
@@ -30,4 +25,3 @@ def test_where_join(fid, operator, value, expected):
 def test_invalid_params():
     with pytest.raises(IncorrectParameters):
         Where(3, 'EX', 12345).build(join='OR')
-

tests/test_response.py

@@ -0,0 +1,19 @@
+import pytest
+
+# test where
+from src.quickbase_json.qb_response import QBResponse
+from src.quickbase_json.helpers import Where, IncorrectParameters
+from tests import sample_data
+
+
+# test loading in data
+@pytest.mark.parametrize('data, expected', [
+    (sample_data.record_data, sample_data.record_data_load)
+])
+def test_load(data, expected):
+    assert QBResponse('records', sample_data=data) == expected
+
+
+# test getting info, basic tests, does not test formatting
+def test_get_info():
+    assert "{'6': {'value': 'Andre Harris'}, '7': {'value': 10}, '8': {'value': '2019-12-18T08:00:00.000Z'}}" in QBResponse('records', sample_data=sample_data.record_data).info(prt=False)