mirror of
https://github.com/RichieCahill/dotfiles.git
synced 2026-04-17 04:58:19 -04:00
fixed tests
This commit is contained in:
@@ -6,7 +6,9 @@ import json
|
|||||||
from unittest.mock import MagicMock
|
from unittest.mock import MagicMock
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
from sqlalchemy import create_engine
|
||||||
|
|
||||||
|
from python.orm.richie.base import RichieBase
|
||||||
from python.signal_bot.commands.inventory import (
|
from python.signal_bot.commands.inventory import (
|
||||||
_format_summary,
|
_format_summary,
|
||||||
parse_llm_response,
|
parse_llm_response,
|
||||||
@@ -75,8 +77,14 @@ class TestDeviceRegistry:
|
|||||||
return MagicMock(spec=SignalClient)
|
return MagicMock(spec=SignalClient)
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def registry(self, tmp_path, signal_mock):
|
def engine(self):
|
||||||
return DeviceRegistry(signal_mock, tmp_path / "devices.json")
|
engine = create_engine("sqlite://")
|
||||||
|
RichieBase.metadata.create_all(engine)
|
||||||
|
return engine
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def registry(self, signal_mock, engine):
|
||||||
|
return DeviceRegistry(signal_mock, engine)
|
||||||
|
|
||||||
def test_new_device_is_unverified(self, registry):
|
def test_new_device_is_unverified(self, registry):
|
||||||
registry.record_contact("+1234", "abc123")
|
registry.record_contact("+1234", "abc123")
|
||||||
@@ -92,8 +100,8 @@ class TestDeviceRegistry:
|
|||||||
|
|
||||||
def test_block_device(self, registry):
|
def test_block_device(self, registry):
|
||||||
registry.record_contact("+1234", "abc123")
|
registry.record_contact("+1234", "abc123")
|
||||||
registry.block("+1234")
|
assert registry.block("+1234")
|
||||||
assert registry.is_blocked("+1234")
|
assert not registry.is_verified("+1234")
|
||||||
|
|
||||||
def test_safety_number_change_resets_trust(self, registry):
|
def test_safety_number_change_resets_trust(self, registry):
|
||||||
registry.record_contact("+1234", "abc123")
|
registry.record_contact("+1234", "abc123")
|
||||||
@@ -102,13 +110,12 @@ class TestDeviceRegistry:
|
|||||||
registry.record_contact("+1234", "different_safety_number")
|
registry.record_contact("+1234", "different_safety_number")
|
||||||
assert not registry.is_verified("+1234")
|
assert not registry.is_verified("+1234")
|
||||||
|
|
||||||
def test_persistence(self, tmp_path, signal_mock):
|
def test_persistence(self, signal_mock, engine):
|
||||||
path = tmp_path / "devices.json"
|
reg1 = DeviceRegistry(signal_mock, engine)
|
||||||
reg1 = DeviceRegistry(signal_mock, path)
|
|
||||||
reg1.record_contact("+1234", "abc123")
|
reg1.record_contact("+1234", "abc123")
|
||||||
reg1.verify("+1234")
|
reg1.verify("+1234")
|
||||||
|
|
||||||
reg2 = DeviceRegistry(signal_mock, path)
|
reg2 = DeviceRegistry(signal_mock, engine)
|
||||||
assert reg2.is_verified("+1234")
|
assert reg2.is_verified("+1234")
|
||||||
|
|
||||||
def test_list_devices(self, registry):
|
def test_list_devices(self, registry):
|
||||||
@@ -138,6 +145,7 @@ class TestDispatch:
|
|||||||
return BotConfig(
|
return BotConfig(
|
||||||
signal_api_url="http://localhost:8080",
|
signal_api_url="http://localhost:8080",
|
||||||
phone_number="+1234567890",
|
phone_number="+1234567890",
|
||||||
|
inventory_api_url="http://localhost:9090",
|
||||||
)
|
)
|
||||||
|
|
||||||
def test_unverified_device_ignored(self, signal_mock, llm_mock, registry_mock, config):
|
def test_unverified_device_ignored(self, signal_mock, llm_mock, registry_mock, config):
|
||||||
|
|||||||
Reference in New Issue
Block a user