195 lines
4.5 KiB
Python
195 lines
4.5 KiB
Python
# test_with_pytest.py
|
|
import pytest
|
|
import tomllib
|
|
from mock import patch
|
|
from cnf.config import Config
|
|
|
|
from home.create_config_toml import create_config
|
|
from test_config import ConfigComplete, ConfigMinimum
|
|
|
|
|
|
class FakeBuffer:
|
|
rd = bytearray()
|
|
wr = str()
|
|
|
|
|
|
test_buffer = FakeBuffer
|
|
|
|
|
|
class FakeFile():
|
|
def __init__(self):
|
|
self.buf = test_buffer
|
|
|
|
def __enter__(self):
|
|
return self
|
|
|
|
def __exit__(self, exc_type, exc, tb):
|
|
pass
|
|
|
|
|
|
class FakeOptionsFile(FakeFile):
|
|
def read(self):
|
|
return self.buf.rd
|
|
|
|
|
|
class FakeConfigFile(FakeFile):
|
|
def write(self, data: str):
|
|
self.buf.wr += data
|
|
|
|
|
|
@pytest.fixture
|
|
def patch_open():
|
|
|
|
def new_open(file: str, OpenTextMode="r"):
|
|
if file == '/data/options.json':
|
|
return FakeOptionsFile()
|
|
elif file == '/home/proxy/config/config.toml':
|
|
# write_buffer += 'bla1'.encode('utf-8')
|
|
return FakeConfigFile()
|
|
|
|
raise TimeoutError
|
|
|
|
with patch('builtins.open', new_open) as conn:
|
|
yield conn
|
|
|
|
|
|
@pytest.fixture
|
|
def ConfigTomlEmpty():
|
|
return {
|
|
'gen3plus': {'at_acl': {'mqtt': {'allow': [], 'block': []},
|
|
'tsun': {'allow': [], 'block': []}}},
|
|
'ha': {'auto_conf_prefix': 'homeassistant',
|
|
'discovery_prefix': 'homeassistant',
|
|
'entity_prefix': 'tsun',
|
|
'proxy_node_id': 'proxy',
|
|
'proxy_unique_id': 'P170000000000001'},
|
|
'inverters': {
|
|
'allow_all': False
|
|
},
|
|
'mqtt': {'host': 'mqtt', 'passwd': '', 'port': 1883, 'user': ''},
|
|
'solarman': {
|
|
'enabled': True,
|
|
'host': 'iot.talent-monitoring.com',
|
|
'port': 10000,
|
|
},
|
|
'tsun': {
|
|
'enabled': True,
|
|
'host': 'logger.talent-monitoring.com',
|
|
'port': 5005,
|
|
},
|
|
}
|
|
|
|
|
|
def test_no_config(patch_open, ConfigTomlEmpty):
|
|
_ = patch_open
|
|
test_buffer.wr = ""
|
|
test_buffer.rd = "" # empty buffer, no json
|
|
create_config()
|
|
cnf = tomllib.loads(test_buffer.wr)
|
|
assert cnf == ConfigTomlEmpty
|
|
|
|
|
|
def test_empty_config(patch_open, ConfigTomlEmpty):
|
|
_ = patch_open
|
|
test_buffer.wr = ""
|
|
test_buffer.rd = "{}" # empty json
|
|
create_config()
|
|
cnf = tomllib.loads(test_buffer.wr)
|
|
assert cnf == ConfigTomlEmpty
|
|
|
|
|
|
def test_full_config(patch_open, ConfigComplete):
|
|
_ = patch_open
|
|
test_buffer.wr = ""
|
|
test_buffer.rd = """
|
|
{
|
|
"inverters": [
|
|
{
|
|
"serial": "R170000000000001",
|
|
"node_id": "PV-Garage",
|
|
"suggested_area": "Garage",
|
|
"modbus_polling": false,
|
|
"pv1_manufacturer": "man1",
|
|
"pv1_type": "type1",
|
|
"pv2_manufacturer": "man2",
|
|
"pv2_type": "type2"
|
|
},
|
|
{
|
|
"serial": "Y170000000000001",
|
|
"monitor_sn": 2000000000,
|
|
"node_id": "PV-Garage2",
|
|
"suggested_area": "Garage2",
|
|
"modbus_polling": true,
|
|
"client_mode_host": "InverterIP",
|
|
"client_mode_port": 1234,
|
|
"pv1_manufacturer": "man1",
|
|
"pv1_type": "type1",
|
|
"pv2_manufacturer": "man2",
|
|
"pv2_type": "type2",
|
|
"pv3_manufacturer": "man3",
|
|
"pv3_type": "type3",
|
|
"pv4_manufacturer": "man4",
|
|
"pv4_type": "type4"
|
|
}
|
|
],
|
|
"tsun.enabled": true,
|
|
"solarman.enabled": true,
|
|
"inverters.allow_all": false,
|
|
"gen3plus.at_acl.tsun.allow": [
|
|
"AT+Z",
|
|
"AT+UPURL",
|
|
"AT+SUPDATE"
|
|
],
|
|
"gen3plus.at_acl.tsun.block": [
|
|
"AT+SUPDATE"
|
|
],
|
|
"gen3plus.at_acl.mqtt.allow": [
|
|
"AT+"
|
|
],
|
|
"gen3plus.at_acl.mqtt.block": [
|
|
"AT+SUPDATE"
|
|
]
|
|
}
|
|
"""
|
|
create_config()
|
|
cnf = tomllib.loads(test_buffer.wr)
|
|
|
|
validated = Config.conf_schema.validate(cnf)
|
|
assert validated == ConfigComplete
|
|
|
|
|
|
def test_minimum_config(patch_open, ConfigMinimum):
|
|
_ = patch_open
|
|
test_buffer.wr = ""
|
|
test_buffer.rd = """
|
|
{
|
|
"inverters": [
|
|
{
|
|
"serial": "R170000000000001",
|
|
"monitor_sn": 0,
|
|
"node_id": "",
|
|
"suggested_area": "",
|
|
"modbus_polling": true,
|
|
"client_mode_host": "InverterIP",
|
|
"client_mode_port": 1234
|
|
}
|
|
],
|
|
"tsun.enabled": true,
|
|
"solarman.enabled": true,
|
|
"inverters.allow_all": true,
|
|
"gen3plus.at_acl.tsun.allow": [
|
|
"AT+Z",
|
|
"AT+UPURL",
|
|
"AT+SUPDATE"
|
|
],
|
|
"gen3plus.at_acl.mqtt.allow": [
|
|
"AT+"
|
|
]
|
|
}
|
|
"""
|
|
create_config()
|
|
cnf = tomllib.loads(test_buffer.wr)
|
|
|
|
validated = Config.conf_schema.validate(cnf)
|
|
assert validated == ConfigMinimum
|