[tests] tests de lib/shift_response et de heatpump/dict

tests/conftest.py

@@ -1,5 +1,6 @@
 import os
 import pytest
+import shutil
 import sys
 from tempfile import mkstemp
 
@@ -25,6 +26,14 @@ def set_test_db():
     yield
 
 
+@pytest.fixture(scope="module")
+def set_db_values():
+    _, tmpdb = mkstemp(suffix='.db', dir=os.getcwd(), )
+    db_file = "./db/prod-b827ebbcab83-20201206.db"
+    shutil.copy(db_file, tmpdb)
+    config['heatpump']['database'] = tmpdb
+    yield
+
 
 @pytest.fixture
 def var_types():

tests/test_heatpump.py

@@ -6,13 +6,39 @@ import json
 from pyheatpump.heatpump import app
 
 
+@pytest.mark.skip
 def test_get_(set_test_db):
     c = TestClient(app)
     r = c.get('/')
     assert r.status_code == 200
 
     resp = r.content.decode()
-    assert type(resp) == str
+    assert isinstance(resp, str)
 
     d_resp = json.loads(resp)
-    assert type(d_resp) == dict
+    assert isinstance(d_resp, dict)
+
+
+def test_dict(set_db_values):
+    c = TestClient(app)
+    r = c.get('/')
+    assert r.status_code == 200
+
+    resp = r.content.decode()
+    assert isinstance(resp, str)
+
+    d_resp = json.loads(resp)
+    assert isinstance(d_resp, dict)
+
+    assert 'macAddress' in d_resp.keys()
+    assert isinstance(d_resp['macAddress'], str)
+    assert 'Analog' in d_resp.keys()
+    assert isinstance(d_resp['Analog'], dict)
+    assert 'Digital' in d_resp.keys()
+    assert isinstance(d_resp['Digital'], dict)
+    assert 'Integer' in d_resp.keys()
+    assert isinstance(d_resp['Integer'], dict)
+    int_keys = list(map(int, d_resp['Integer'].keys()))
+    assert min(int_keys) == 1
+    assert max(int_keys) == 1250
+

tests/test_lib.py

@@ -0,0 +1,39 @@
+#!/usr/bin/env python3
+import pytest
+
+from pyheatpump.lib import shift_response
+
+def test_shift_response():
+    heatpump_dict = {}
+    heatpump_dict['Analog'] = {
+        key: 0
+        for key in range(1, 1251)
+    }
+    heatpump_dict['Integer'] = {
+        key: 0
+        for key in range(5002, 6252 )
+    }
+    heatpump_dict['Digital'] = {
+        key: 0
+        for key in range(1, 1001)
+    }
+
+    assert isinstance(heatpump_dict, dict)
+    
+    shifted = shift_response(heatpump_dict)
+    assert isinstance(shifted, dict) 
+
+    assert 'Analog' in shifted.keys()
+    assert isinstance(shifted['Analog'], dict)
+
+    assert 'Digital' in shifted.keys()
+    assert isinstance(shifted['Digital'], dict)
+
+    assert 'Integer' in shifted.keys()
+    assert isinstance(shifted['Integer'], dict)
+    for k in shifted['Integer'].keys():
+        assert isinstance(k, str)
+
+    int_keys = list(map(int, shifted['Integer'].keys()))
+    assert min(int_keys) == 1
+    assert max(int_keys) == 1250