[conf][tests] added mac_address retrieval

pyheatpump/config.py

@@ -8,11 +8,14 @@ from pprint import pprint
 import uvicorn
 import json
 
+from pyheatpump.logger import logger_init
+logger = logger_init()
+
 default_config = {
     'heatpump': {
         'serial_port': '/dev/ttyUSB0',
         'baudrate': 19200,
-        'mac_address': '00:00:00:00:00:00',
+        'mac_address': 'None',
         'ip_address': 'dhcp',
         'read_only': 'False',
         'database': '/var/run/pyheatpump/pyheatpump.sqlite3'
@@ -47,6 +50,27 @@ config.read(filenames=CONFIG_FILES)
 API_HOST=config.get('api', 'host')
 API_PORT=config.get('api', 'port')
 
+def save_config():
+    with open(CONFIG_FILES[0], 'w') as conf_file:
+        config.write(conf_file)
+
+def mac_address_init():
+    from netifaces import gateways, ifaddresses, AF_INET, AF_LINK
+    interface = gateways()['default'][AF_INET][1]
+    if len(ifaddresses(interface)) == 0:
+        logger.error("Can't find interface")
+        sys.exit(1)
+
+    if len(ifaddresses(interface)[AF_LINK]) == 0:
+        logger.error("Can't find interface")
+        sys.exit(1)
+
+    addr = ifaddresses(interface)[AF_LINK][0]['addr']
+    config.set('heatpump', 'mac_address', addr)
+    save_config()
+
+    return addr
+
 async def get_config(request):
     items = dict([ (
         section, dict([ (key, config.get(section, key)) for key in items ])
@@ -54,7 +78,6 @@ async def get_config(request):
     return JSONResponse(items)
 
 async def set_config(request):
-
     body = json.loads(await request.json())
 
     for sect in body.keys():
@@ -64,9 +87,7 @@ async def set_config(request):
             else:
                 config.set(sect, item, value=body[sect][item])
 
-    with open(CONFIG_FILES[0], 'w') as conf_file:
-        config.write(conf_file)
-
+    save_config()
     return PlainTextResponse('OK')
 
 async def config_route(request, *args, **kwargs):

tests/conftest.py

@@ -0,0 +1 @@
+pytest_plugins = "pytester"

tests/test_config.py

@@ -2,7 +2,7 @@
 import pytest
 from starlette.authentication import UnauthenticatedUser
 from starlette.testclient import TestClient
-from pyheatpump.config import app, config, default_config, CONFIG_FILES, get_config, set_config, config_route, ROUTES
+from pyheatpump.config import app, config, default_config, CONFIG_FILES, get_config, set_config, config_route, ROUTES, mac_address_init, save_config
 from unittest.mock import patch, MagicMock
 from pprint import pprint
 import json
@@ -10,16 +10,33 @@ from tempfile import mkstemp
 from configparser import ConfigParser
 import os
 
+@pytest.fixture
+def tmpconf(testdir):
+    tmp_conf_filepath = os.path.join(testdir.tmpdir, 'pyheatpump.ini')
+    config = ConfigParser(
+        default_section='heatpump')
+    config.read_dict(default_config)
+
+
+    with open(tmp_conf_filepath, 'w') as f:
+        f.write("[heatpump]\nmac_address = 'None'")
+        CONFIG_FILES = [tmp_conf_filepath]
+        config.read(filenames=CONFIG_FILES)
+
+    return config
+
 def test_get_():
     c = TestClient(app)
     r = c.get('/')
     assert r.status_code == 200
 
+
 class RequestMock(MagicMock):
     def __get__(self, key):
         if key == 'method':
             return 'GET'
 
+
 @pytest.mark.asyncio
 async def test_get_config():
     resp = await get_config(RequestMock())
@@ -30,6 +47,7 @@ async def test_get_config():
         for item in default_config[sect].keys():
             assert item in d_resp[sect].keys()
 
+
 def test_set_config():
     c = TestClient(app)
     _, tmpconf = mkstemp()
@@ -47,3 +65,13 @@ def test_set_config():
     assert config.get('supervisor', 'scheme') == 'test'
 
     os.unlink(tmpconf)
+
+
+def test_mac_address(testdir, tmpconf):
+    try:
+        assert tmpconf.get('heatpump', 'mac_address') is 'None'
+        mac_address = mac_address_init()
+        config_mac_address = tmpconf.get('heatpump', 'mac_address')
+        assert config_mac_address == mac_address
+    except Exception as e:
+        raise e