91 lines
2.6 KiB
Python
91 lines
2.6 KiB
Python
# test_with_pytest.py
|
|
import pytest
|
|
import asyncio
|
|
import aiomqtt
|
|
import logging
|
|
|
|
from mock import patch, Mock
|
|
from app.src.singleton import Singleton
|
|
from app.src.inverter import Inverter
|
|
from app.src.mqtt import Mqtt
|
|
from app.src.gen3plus.solarman_v5 import SolarmanV5
|
|
from app.src.config import Config
|
|
|
|
|
|
pytest_plugins = ('pytest_asyncio',)
|
|
|
|
|
|
@pytest.fixture(scope="module", autouse=True)
|
|
def module_init():
|
|
def new_init(cls, cb_mqtt_is_up):
|
|
cb_mqtt_is_up()
|
|
|
|
Singleton._instances.clear()
|
|
with patch.object(Mqtt, '__init__', new_init):
|
|
yield
|
|
|
|
@pytest.fixture(scope="module")
|
|
def test_port():
|
|
return 1883
|
|
|
|
@pytest.fixture(scope="module")
|
|
def test_hostname():
|
|
# if getenv("GITHUB_ACTIONS") == "true":
|
|
# return 'mqtt'
|
|
# else:
|
|
return 'test.mosquitto.org'
|
|
|
|
@pytest.fixture
|
|
def config_conn(test_hostname, test_port):
|
|
Config.act_config = {
|
|
'mqtt':{
|
|
'host': test_hostname,
|
|
'port': test_port,
|
|
'user': '',
|
|
'passwd': ''
|
|
},
|
|
'ha':{
|
|
'auto_conf_prefix': 'homeassistant',
|
|
'discovery_prefix': 'homeassistant',
|
|
'entity_prefix': 'tsun',
|
|
'proxy_node_id': 'test_1',
|
|
'proxy_unique_id': ''
|
|
},
|
|
'inverters': {
|
|
'allow_all': True,
|
|
"R170000000000001":{
|
|
'node_id': 'inv_1'
|
|
}
|
|
}
|
|
}
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_inverter_cb(config_conn):
|
|
_ = config_conn
|
|
|
|
with patch.object(Inverter, '_cb_mqtt_is_up', wraps=Inverter._cb_mqtt_is_up) as spy:
|
|
print('call Inverter.class_init')
|
|
Inverter.class_init()
|
|
assert 'homeassistant/' == Inverter.discovery_prfx
|
|
assert 'tsun/' == Inverter.entity_prfx
|
|
assert 'test_1/' == Inverter.proxy_node_id
|
|
spy.assert_called_once()
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_mqtt_is_up(config_conn):
|
|
_ = config_conn
|
|
|
|
with patch.object(Mqtt, 'publish') as spy:
|
|
Inverter.class_init()
|
|
await Inverter._cb_mqtt_is_up()
|
|
spy.assert_called()
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_mqtt_proxy_statt_invalid(config_conn):
|
|
_ = config_conn
|
|
|
|
with patch.object(Mqtt, 'publish') as spy:
|
|
Inverter.class_init()
|
|
await Inverter._async_publ_mqtt_proxy_stat('InValId_kEy')
|
|
spy.assert_not_called()
|