pyHeatpump/tests/cli/test_supervise.py

import os
import pytest
from click.testing import CliRunner
from unittest.mock import patch
from tempfile import mkstemp
import requests

from pyheatpump.cli import cli
from pyheatpump.config import config
from pyheatpump.db import DB
from pyheatpump.models import VariableValue

@pytest.fixture(scope='module')
def set_test_db():
    _, tmpdb = mkstemp(suffix='.db', dir=os.getcwd(), )
    print(f'Will store database in {tmpdb}')
    config['heatpump']['database'] = tmpdb
    if not DB.initialize(os.path.join(os.getcwd(), 'db/pyheatpump.test.sql')):
        sys.exit(-1)

    yield

    os.unlink(tmpdb)

@pytest.fixture
def runner():
    return CliRunner()

@pytest.fixture
def fetch(set_test_db, runner):
    try:
        runner.invoke(cli, 'fetch')
    except ValueError:
        pass

@pytest.fixture
def default_config():
    config['supervisor']['scheme'] = 'http'
    config['supervisor']['host'] = '127.0.0.1'
    config['supervisor']['port'] = '8080'
    config['supervisor']['post_path'] = '/post'
    config['supervisor']['get_path'] = '/get'
    config['heatpump']['mac_address'] = '00:11:22:33:44:55'

@pytest.mark.skip
@patch('pyheatpump.cli.requests')
def test_supervise(requestsMock, fetch, runner, default_config):
    data = {'macAddress':'001122334455'}
    data_analog = { **data, **{'Analog':{} } }
    data_integer = { **data, **{'Integer':{} } }
    data_digital = { **data, **{'Digital':{} } }

    try:
        r = runner.invoke(cli, 'supervise')
    except ValueError:
        pass

    requestsMock.post.assert_called()
    requestsMock.post.assert_called_with(
        url='http://127.0.0.1:8080/post',
        json=data_integer,
        verify=False
    )

    requestsMock.get.assert_called()
    requestsMock.get.assert_called_with(
        url=''.join('http://127.0.0.1:8080/get/001122334455'),
        verify=False)


@patch('pyheatpump.cli.requests')
def test_supervise_values(requestsMock, fetch, runner, default_config, set_db_values):
    data = {'macAddress':'001122334455'}
    data_analog = { **data, **{'Analog':{} } }
    data_integer = { **data, **{'Integer':{} } }
    data_digital = { **data, **{'Digital':{} } }

    try:
        r = runner.invoke(cli, 'supervise')
    except ValueError:
        pass

    requestsMock.post.assert_called()
    requestsMock.post.assert_called_with(
        url='http://127.0.0.1:8080/post',
        json=data_integer,
        verify=False
    )

    requestsMock.get.assert_called()
    requestsMock.get.assert_called_with(
        url=''.join('http://127.0.0.1:8080/get/001122334455'),
        verify=False)