| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575 |
- """Unit tests for NotificationService.
- Tests event-based notifications and toggle behavior.
- """
- import json
- from unittest.mock import AsyncMock, MagicMock, patch
- import pytest
- from backend.app.services.notification_service import NotificationService
- class TestNotificationService:
- """Tests for NotificationService class."""
- @pytest.fixture
- def service(self):
- """Create a fresh NotificationService instance."""
- return NotificationService()
- @pytest.fixture
- def mock_provider(self):
- """Create a mock notification provider."""
- provider = MagicMock()
- provider.id = 1
- provider.name = "Test Provider"
- provider.provider_type = "webhook"
- provider.enabled = True
- provider.config = json.dumps({"webhook_url": "http://test.local/webhook"})
- provider.on_print_start = True
- provider.on_print_complete = True
- provider.on_print_failed = True
- provider.on_print_stopped = False
- provider.on_print_progress = False
- provider.on_printer_offline = False
- provider.on_printer_error = False
- provider.on_filament_low = False
- provider.on_maintenance_due = False
- provider.on_ams_humidity_high = False
- provider.on_ams_temperature_high = False
- provider.quiet_hours_enabled = False
- provider.quiet_hours_start = None
- provider.quiet_hours_end = None
- provider.daily_digest_enabled = False
- provider.daily_digest_time = None
- provider.printer_id = None
- return provider
- @pytest.fixture
- def mock_db(self):
- """Create a mock database session."""
- db = AsyncMock()
- db.commit = AsyncMock()
- return db
- # ========================================================================
- # Tests for on_print_start
- # ========================================================================
- @pytest.mark.asyncio
- async def test_on_print_start_sends_notification(self, service, mock_provider, mock_db):
- """Verify notification is sent when print starts."""
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock) as mock_send,
- patch.object(service, "_build_message_from_template", new_callable=AsyncMock) as mock_build,
- ):
- mock_get.return_value = [mock_provider]
- mock_build.return_value = ("Print Started", "Test Printer: test.3mf")
- await service.on_print_start(
- printer_id=1,
- printer_name="Test Printer",
- data={"filename": "test.3mf", "subtask_name": "test"},
- db=mock_db,
- )
- mock_get.assert_called_once()
- mock_send.assert_called_once()
- @pytest.mark.asyncio
- async def test_on_print_start_skipped_when_no_providers(self, service, mock_db):
- """Verify no error when no providers are configured for event."""
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock) as mock_send,
- ):
- mock_get.return_value = []
- await service.on_print_start(
- printer_id=1,
- printer_name="Test Printer",
- data={},
- db=mock_db,
- )
- mock_send.assert_not_called()
- # ========================================================================
- # Tests for on_print_complete (status routing)
- # ========================================================================
- @pytest.mark.asyncio
- async def test_on_print_complete_routes_completed_status(self, service, mock_provider, mock_db):
- """Verify completed status uses on_print_complete field."""
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", new_callable=AsyncMock) as mock_build,
- ):
- mock_get.return_value = [mock_provider]
- mock_build.return_value = ("Test", "Test")
- await service.on_print_complete(
- printer_id=1,
- printer_name="Test",
- status="completed",
- data={},
- db=mock_db,
- )
- # Verify the correct event field was queried
- call_args = mock_get.call_args
- assert call_args[0][1] == "on_print_complete"
- @pytest.mark.asyncio
- async def test_on_print_complete_routes_failed_status(self, service, mock_provider, mock_db):
- """Verify failed status uses on_print_failed field."""
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", new_callable=AsyncMock) as mock_build,
- ):
- mock_get.return_value = [mock_provider]
- mock_build.return_value = ("Test", "Test")
- await service.on_print_complete(
- printer_id=1,
- printer_name="Test",
- status="failed",
- data={},
- db=mock_db,
- )
- call_args = mock_get.call_args
- assert call_args[0][1] == "on_print_failed"
- @pytest.mark.asyncio
- async def test_on_print_complete_routes_stopped_status(self, service, mock_provider, mock_db):
- """Verify stopped status uses on_print_stopped field."""
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", new_callable=AsyncMock) as mock_build,
- ):
- mock_get.return_value = [mock_provider]
- mock_build.return_value = ("Test", "Test")
- await service.on_print_complete(
- printer_id=1,
- printer_name="Test",
- status="stopped",
- data={},
- db=mock_db,
- )
- call_args = mock_get.call_args
- assert call_args[0][1] == "on_print_stopped"
- @pytest.mark.asyncio
- async def test_on_print_complete_routes_aborted_status(self, service, mock_provider, mock_db):
- """Verify aborted status uses on_print_stopped field."""
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", new_callable=AsyncMock) as mock_build,
- ):
- mock_get.return_value = [mock_provider]
- mock_build.return_value = ("Test", "Test")
- await service.on_print_complete(
- printer_id=1,
- printer_name="Test",
- status="aborted",
- data={},
- db=mock_db,
- )
- call_args = mock_get.call_args
- assert call_args[0][1] == "on_print_stopped"
- # ========================================================================
- # Tests for provider filtering
- # ========================================================================
- @pytest.mark.asyncio
- async def test_disabled_provider_not_returned(self, service, mock_provider, mock_db):
- """CRITICAL: Verify disabled providers don't receive notifications."""
- mock_provider.enabled = False
- # The actual filtering happens in _get_providers_for_event
- # which queries only enabled providers
- with patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get:
- # Simulate the query filtering out disabled providers
- mock_get.return_value = []
- result = await service._get_providers_for_event(mock_db, "on_print_start", printer_id=1)
- assert len(result) == 0
- @pytest.mark.asyncio
- async def test_provider_filtered_by_printer_id(self, service, mock_provider, mock_db):
- """Verify providers can be filtered by specific printer."""
- mock_provider.printer_id = 2 # Linked to printer 2
- with patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get:
- # When querying for printer 1, provider linked to printer 2 is excluded
- mock_get.return_value = []
- result = await service._get_providers_for_event(mock_db, "on_print_start", printer_id=1)
- assert len(result) == 0
- # ========================================================================
- # Tests for quiet hours
- # ========================================================================
- def test_is_in_quiet_hours_during_quiet_period(self, service, mock_provider):
- """Verify notifications are blocked during quiet hours."""
- mock_provider.quiet_hours_enabled = True
- mock_provider.quiet_hours_start = "22:00"
- mock_provider.quiet_hours_end = "07:00"
- with patch("backend.app.services.notification_service.datetime") as mock_datetime:
- # Test during quiet hours (23:00)
- mock_now = MagicMock()
- mock_now.hour = 23
- mock_now.minute = 0
- mock_datetime.now.return_value = mock_now
- result = service._is_in_quiet_hours(mock_provider)
- assert result is True
- def test_is_in_quiet_hours_outside_quiet_period(self, service, mock_provider):
- """Verify notifications are allowed outside quiet hours."""
- mock_provider.quiet_hours_enabled = True
- mock_provider.quiet_hours_start = "22:00"
- mock_provider.quiet_hours_end = "07:00"
- with patch("backend.app.services.notification_service.datetime") as mock_datetime:
- # Test outside quiet hours (12:00)
- mock_now = MagicMock()
- mock_now.hour = 12
- mock_now.minute = 0
- mock_datetime.now.return_value = mock_now
- result = service._is_in_quiet_hours(mock_provider)
- assert result is False
- def test_is_in_quiet_hours_disabled(self, service, mock_provider):
- """Verify quiet hours check returns False when disabled."""
- mock_provider.quiet_hours_enabled = False
- result = service._is_in_quiet_hours(mock_provider)
- assert result is False
- def test_is_in_quiet_hours_early_morning(self, service, mock_provider):
- """Verify quiet hours work across midnight (early morning)."""
- mock_provider.quiet_hours_enabled = True
- mock_provider.quiet_hours_start = "22:00"
- mock_provider.quiet_hours_end = "07:00"
- with patch("backend.app.services.notification_service.datetime") as mock_datetime:
- # Test early morning (03:00) - should be in quiet hours
- mock_now = MagicMock()
- mock_now.hour = 3
- mock_now.minute = 0
- mock_datetime.now.return_value = mock_now
- result = service._is_in_quiet_hours(mock_provider)
- assert result is True
- # ========================================================================
- # Tests for AMS alarms
- # ========================================================================
- @pytest.mark.asyncio
- async def test_on_ams_humidity_high_sends_notification(self, service, mock_provider, mock_db):
- """Verify AMS humidity alarm sends notification."""
- mock_provider.on_ams_humidity_high = True
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock) as mock_send,
- patch.object(service, "_build_message_from_template", new_callable=AsyncMock) as mock_build,
- ):
- mock_get.return_value = [mock_provider]
- mock_build.return_value = ("AMS Humidity Alert", "High humidity detected")
- await service.on_ams_humidity_high(
- printer_id=1,
- printer_name="Test Printer",
- ams_label="AMS-A",
- humidity=75.0,
- threshold=60.0,
- db=mock_db,
- )
- mock_send.assert_called_once()
- # Verify force_immediate is True for alarms
- call_kwargs = mock_send.call_args[1]
- assert call_kwargs.get("force_immediate") is True
- @pytest.mark.asyncio
- async def test_on_ams_temperature_high_sends_notification(self, service, mock_provider, mock_db):
- """Verify AMS temperature alarm sends notification."""
- mock_provider.on_ams_temperature_high = True
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock) as mock_send,
- patch.object(service, "_build_message_from_template", new_callable=AsyncMock) as mock_build,
- ):
- mock_get.return_value = [mock_provider]
- mock_build.return_value = ("AMS Temperature Alert", "High temp detected")
- await service.on_ams_temperature_high(
- printer_id=1,
- printer_name="Test Printer",
- ams_label="AMS-A",
- temperature=40.0,
- threshold=35.0,
- db=mock_db,
- )
- mock_send.assert_called_once()
- # Verify force_immediate is True for alarms
- call_kwargs = mock_send.call_args[1]
- assert call_kwargs.get("force_immediate") is True
- @pytest.mark.asyncio
- async def test_ams_alarm_skipped_when_toggle_disabled(self, service, mock_provider, mock_db):
- """CRITICAL: Verify AMS alarms respect toggle setting."""
- mock_provider.on_ams_humidity_high = False
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock) as mock_send,
- ):
- # Provider with toggle disabled won't be returned
- mock_get.return_value = []
- await service.on_ams_humidity_high(
- printer_id=1,
- printer_name="Test",
- ams_label="AMS-A",
- humidity=75.0,
- threshold=60.0,
- db=mock_db,
- )
- mock_send.assert_not_called()
- # ========================================================================
- # Tests for daily digest
- # ========================================================================
- @pytest.mark.asyncio
- async def test_daily_digest_queues_notification(self, service, mock_provider, mock_db):
- """Verify notifications are queued when digest mode is enabled."""
- mock_provider.daily_digest_enabled = True
- mock_provider.daily_digest_time = "09:00"
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock) as mock_send,
- patch.object(service, "_build_message_from_template", new_callable=AsyncMock) as mock_build,
- ):
- mock_get.return_value = [mock_provider]
- mock_build.return_value = ("Test", "Test")
- await service.on_print_complete(
- printer_id=1,
- printer_name="Test",
- status="completed",
- data={},
- db=mock_db,
- )
- # When digest is enabled, _send_to_providers should still be called
- # but internally it will queue instead of send immediately
- mock_send.assert_called_once()
- @pytest.mark.asyncio
- async def test_force_immediate_bypasses_digest(self, service, mock_provider, mock_db):
- """Verify force_immediate=True bypasses digest mode."""
- mock_provider.daily_digest_enabled = True
- mock_provider.on_ams_humidity_high = True
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock) as mock_send,
- patch.object(service, "_build_message_from_template", new_callable=AsyncMock) as mock_build,
- ):
- mock_get.return_value = [mock_provider]
- mock_build.return_value = ("Alert", "Alert message")
- await service.on_ams_humidity_high(
- printer_id=1,
- printer_name="Test",
- ams_label="AMS-A",
- humidity=75.0,
- threshold=60.0,
- db=mock_db,
- )
- # Verify force_immediate is passed
- call_kwargs = mock_send.call_args[1]
- assert call_kwargs.get("force_immediate") is True
- class TestDigestModeAlwaysSendsImmediately:
- """CRITICAL: Tests that notifications always send immediately regardless of digest setting."""
- @pytest.fixture
- def service(self):
- return NotificationService()
- @pytest.mark.asyncio
- async def test_notification_sends_immediately_even_with_digest_enabled(self, service):
- """CRITICAL: All notifications must be sent immediately, digest is just a summary."""
- # Create a mock provider with digest enabled
- mock_provider = MagicMock()
- mock_provider.id = 1
- mock_provider.name = "Test Provider"
- mock_provider.provider_type = "ntfy"
- mock_provider.enabled = True
- mock_provider.daily_digest_enabled = True # Digest enabled
- mock_provider.daily_digest_time = "23:59"
- mock_provider.config = '{"server": "https://ntfy.sh", "topic": "test"}'
- mock_db = AsyncMock()
- # Mock the _send_to_provider method
- with (
- patch.object(service, "_send_to_provider", new_callable=AsyncMock) as mock_send,
- patch.object(service, "_queue_for_digest", new_callable=AsyncMock) as mock_queue,
- patch.object(service, "_update_provider_status", new_callable=AsyncMock),
- patch.object(service, "_log_notification", new_callable=AsyncMock),
- ):
- mock_send.return_value = (True, None)
- await service._send_to_providers(
- providers=[mock_provider],
- title="Print Started",
- message="Your print has started",
- db=mock_db,
- event_type="print_start",
- )
- # CRITICAL: _send_to_provider MUST be called (immediate send)
- mock_send.assert_called_once()
- # Digest queue should also be called (for daily summary)
- mock_queue.assert_called_once()
- @pytest.mark.asyncio
- async def test_notification_sends_without_digest_queue_when_disabled(self, service):
- """When digest is disabled, notification sends but no digest queue."""
- mock_provider = MagicMock()
- mock_provider.id = 1
- mock_provider.name = "Test Provider"
- mock_provider.provider_type = "ntfy"
- mock_provider.enabled = True
- mock_provider.daily_digest_enabled = False # Digest disabled
- mock_provider.daily_digest_time = None
- mock_provider.config = '{"server": "https://ntfy.sh", "topic": "test"}'
- mock_db = AsyncMock()
- with (
- patch.object(service, "_send_to_provider", new_callable=AsyncMock) as mock_send,
- patch.object(service, "_queue_for_digest", new_callable=AsyncMock) as mock_queue,
- patch.object(service, "_update_provider_status", new_callable=AsyncMock),
- patch.object(service, "_log_notification", new_callable=AsyncMock),
- ):
- mock_send.return_value = (True, None)
- await service._send_to_providers(
- providers=[mock_provider],
- title="Print Started",
- message="Your print has started",
- db=mock_db,
- event_type="print_start",
- )
- # Notification must still be sent immediately
- mock_send.assert_called_once()
- # Digest queue should NOT be called when digest is disabled
- mock_queue.assert_not_called()
- class TestNotificationProviderTypes:
- """Tests for different notification provider types."""
- @pytest.fixture
- def service(self):
- return NotificationService()
- @pytest.mark.asyncio
- async def test_webhook_provider_sends_request(self, service):
- """Verify webhook provider sends HTTP request."""
- config = {
- "webhook_url": "http://test.local/webhook",
- "field_title": "title",
- "field_message": "message",
- }
- # Create a mock response
- mock_response = MagicMock()
- mock_response.status_code = 200
- # Mock the _get_client method
- mock_client = AsyncMock()
- mock_client.post = AsyncMock(return_value=mock_response)
- with patch.object(service, "_get_client", new_callable=AsyncMock) as mock_get_client:
- mock_get_client.return_value = mock_client
- success, message = await service._send_webhook(config, "Test Title", "Test Message")
- assert success is True
- mock_client.post.assert_called_once()
- @pytest.mark.asyncio
- async def test_webhook_handles_failure(self, service):
- """Verify webhook gracefully handles HTTP errors."""
- config = {
- "webhook_url": "http://test.local/webhook",
- }
- with patch("httpx.AsyncClient") as mock_client_class:
- mock_instance = AsyncMock()
- mock_instance.post.side_effect = Exception("Connection failed")
- mock_client_class.return_value.__aenter__ = AsyncMock(return_value=mock_instance)
- mock_client_class.return_value.__aexit__ = AsyncMock()
- success, message = await service._send_webhook(config, "Test", "Test")
- assert success is False
- assert "Connection failed" in message or "error" in message.lower()
- @pytest.mark.asyncio
- async def test_webhook_slack_format_sends_text_only(self, service):
- """Verify Slack/Mattermost format sends only text field."""
- config = {
- "webhook_url": "http://mattermost.local/hooks/abc123",
- "payload_format": "slack",
- }
- mock_response = MagicMock()
- mock_response.status_code = 200
- mock_client = AsyncMock()
- mock_client.post = AsyncMock(return_value=mock_response)
- with patch.object(service, "_get_client", new_callable=AsyncMock) as mock_get_client:
- mock_get_client.return_value = mock_client
- success, message = await service._send_webhook(config, "Test Title", "Test Message")
- assert success is True
- mock_client.post.assert_called_once()
- # Verify payload format is Slack-compatible
- call_args = mock_client.post.call_args
- payload = call_args.kwargs.get("json") or call_args[1].get("json")
- assert "text" in payload
- assert "*Test Title*" in payload["text"]
- assert "Test Message" in payload["text"]
- # Should NOT have generic fields
- assert "timestamp" not in payload
- assert "source" not in payload
- class TestHomeAssistantProvider:
- """Tests for Home Assistant notification provider."""
- @pytest.fixture
- def service(self):
- return NotificationService()
- @pytest.mark.asyncio
- async def test_send_homeassistant_success(self, service):
- """Verify HA provider sends persistent notification to correct endpoint."""
- mock_response = MagicMock()
- mock_response.status_code = 200
- mock_client = AsyncMock()
- mock_client.post = AsyncMock(return_value=mock_response)
- mock_db = AsyncMock()
- with (
- patch.object(service, "_get_client", new_callable=AsyncMock) as mock_get_client,
- patch(
- "backend.app.api.routes.settings.get_homeassistant_settings",
- new_callable=AsyncMock,
- ) as mock_ha_settings,
- ):
- mock_get_client.return_value = mock_client
- mock_ha_settings.return_value = {
- "ha_url": "http://ha.local:8123",
- "ha_token": "test-token-123",
- "ha_enabled": True,
- }
- success, message = await service._send_homeassistant({}, "Test Title", "Test Message", db=mock_db)
- assert success is True
- mock_client.post.assert_called_once()
- call_args = mock_client.post.call_args
- assert call_args[0][0] == "http://ha.local:8123/api/services/persistent_notification/create"
- payload = call_args.kwargs.get("json") or call_args[1].get("json")
- assert payload["title"] == "Test Title"
- assert payload["message"] == "Test Message"
- @pytest.mark.asyncio
- async def test_send_homeassistant_no_db_no_env(self, service):
- """Verify HA provider fails gracefully without DB or env vars."""
- with patch.dict("os.environ", {}, clear=True):
- success, message = await service._send_homeassistant({}, "Test", "Test", db=None)
- assert success is False
- assert "not configured" in message.lower()
- @pytest.mark.asyncio
- async def test_send_homeassistant_auth_failure(self, service):
- """Verify HA provider reports auth failure."""
- mock_response = MagicMock()
- mock_response.status_code = 401
- mock_client = AsyncMock()
- mock_client.post = AsyncMock(return_value=mock_response)
- mock_db = AsyncMock()
- with (
- patch.object(service, "_get_client", new_callable=AsyncMock) as mock_get_client,
- patch(
- "backend.app.api.routes.settings.get_homeassistant_settings",
- new_callable=AsyncMock,
- ) as mock_ha_settings,
- ):
- mock_get_client.return_value = mock_client
- mock_ha_settings.return_value = {
- "ha_url": "http://ha.local:8123",
- "ha_token": "bad-token",
- "ha_enabled": True,
- }
- success, message = await service._send_homeassistant({}, "Test", "Test", db=mock_db)
- assert success is False
- assert "authentication" in message.lower()
- @pytest.mark.asyncio
- async def test_send_homeassistant_env_fallback(self, service):
- """Verify HA provider falls back to env vars when no DB session."""
- mock_response = MagicMock()
- mock_response.status_code = 200
- mock_client = AsyncMock()
- mock_client.post = AsyncMock(return_value=mock_response)
- with (
- patch.object(service, "_get_client", new_callable=AsyncMock) as mock_get_client,
- patch.dict("os.environ", {"HA_URL": "http://env-ha:8123", "HA_TOKEN": "env-token"}),
- ):
- mock_get_client.return_value = mock_client
- success, message = await service._send_homeassistant({}, "Test", "Test", db=None)
- assert success is True
- call_args = mock_client.post.call_args
- assert "env-ha:8123" in call_args[0][0]
- @pytest.mark.asyncio
- async def test_send_homeassistant_empty_config_accepted(self, service):
- """Verify HA provider works with empty config dict (no fields needed)."""
- mock_response = MagicMock()
- mock_response.status_code = 200
- mock_client = AsyncMock()
- mock_client.post = AsyncMock(return_value=mock_response)
- mock_db = AsyncMock()
- with (
- patch.object(service, "_get_client", new_callable=AsyncMock) as mock_get_client,
- patch(
- "backend.app.api.routes.settings.get_homeassistant_settings",
- new_callable=AsyncMock,
- ) as mock_ha_settings,
- ):
- mock_get_client.return_value = mock_client
- mock_ha_settings.return_value = {
- "ha_url": "http://ha.local:8123",
- "ha_token": "token",
- "ha_enabled": True,
- }
- success, _ = await service._send_homeassistant({}, "Title", "Body", db=mock_db)
- assert success is True
- @pytest.mark.asyncio
- async def test_send_to_provider_dispatches_homeassistant(self, service):
- """Verify _send_to_provider dispatches to _send_homeassistant."""
- provider = MagicMock()
- provider.provider_type = "homeassistant"
- provider.config = "{}"
- provider.quiet_hours_enabled = False
- with patch.object(service, "_send_homeassistant", new_callable=AsyncMock) as mock_send:
- mock_send.return_value = (True, "OK")
- success, _ = await service._send_to_provider(provider, "Title", "Message", db=AsyncMock())
- assert success is True
- mock_send.assert_called_once()
- class TestNotificationVariableFallbacks:
- """Tests for notification variable fallback values."""
- @pytest.fixture
- def service(self):
- return NotificationService()
- def test_format_duration_with_valid_seconds(self, service):
- """Verify duration formats correctly with valid input."""
- result = service._format_duration(3661) # 1h 1m 1s
- assert "1h" in result
- def test_format_duration_with_none_returns_unknown(self, service):
- """CRITICAL: Verify None duration returns 'Unknown' fallback."""
- result = service._format_duration(None)
- assert result == "Unknown"
- def test_format_duration_with_zero(self, service):
- """Verify zero duration formats correctly."""
- result = service._format_duration(0)
- # Should return some valid string, not "Unknown"
- assert result is not None
- assert isinstance(result, str)
- def test_format_duration_hours_and_minutes(self, service):
- """Verify duration formats hours and minutes."""
- result = service._format_duration(5400) # 1h 30m
- assert "1h" in result
- assert "30m" in result
- def test_format_duration_minutes_only(self, service):
- """Verify duration formats minutes only when < 1 hour."""
- result = service._format_duration(1800) # 30m
- assert "30m" in result or "30" in result
- @pytest.mark.asyncio
- async def test_print_complete_fallback_values(self, service):
- """CRITICAL: Verify fallback values when archive_data is missing."""
- mock_db = AsyncMock()
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", new_callable=AsyncMock) as mock_build,
- ):
- mock_get.return_value = [] # No providers, just testing variable setup
- mock_build.return_value = ("Test", "Test")
- await service.on_print_complete(
- printer_id=1,
- printer_name="Test",
- status="completed",
- data={"subtask_name": "test_print"},
- db=mock_db,
- archive_data=None, # No archive data - should use fallbacks
- )
- # Test passes if no exception is raised with missing archive_data
- @pytest.mark.asyncio
- async def test_print_complete_with_archive_data(self, service):
- """Verify archive data values are used when provided."""
- mock_db = AsyncMock()
- captured_variables = {}
- async def capture_build(db, event_type, variables):
- captured_variables.update(variables)
- return ("Test", "Test")
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", side_effect=capture_build),
- ):
- mock_get.return_value = []
- await service.on_print_complete(
- printer_id=1,
- printer_name="Test",
- status="completed",
- data={"subtask_name": "test_print"},
- db=mock_db,
- archive_data={
- "print_time_seconds": 3600,
- "actual_filament_grams": 50.5,
- },
- )
- # When archive data is provided, duration should not be "Unknown"
- if captured_variables.get("duration"):
- assert captured_variables["duration"] != "Unknown"
- @pytest.mark.asyncio
- async def test_print_complete_with_finish_photo_url(self, service):
- """Verify finish_photo_url is passed through from archive_data."""
- mock_db = AsyncMock()
- mock_provider = MagicMock()
- mock_provider.id = 1
- captured_variables = {}
- async def capture_build(db, event_type, variables):
- captured_variables.update(variables)
- return ("Test", "Test")
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", side_effect=capture_build),
- ):
- mock_get.return_value = [mock_provider]
- await service.on_print_complete(
- printer_id=1,
- printer_name="Test",
- status="completed",
- data={"subtask_name": "test_print"},
- db=mock_db,
- archive_data={
- "print_time_seconds": 3600,
- "actual_filament_grams": 50.5,
- "finish_photo_url": "http://localhost:8000/api/v1/archives/1/photos/finish_test.jpg",
- },
- )
- # finish_photo_url should be passed through to template variables
- assert (
- captured_variables.get("finish_photo_url")
- == "http://localhost:8000/api/v1/archives/1/photos/finish_test.jpg"
- )
- @pytest.mark.asyncio
- async def test_print_start_estimated_time_fallback(self, service):
- """Verify estimated time shows 'Unknown' when not available."""
- mock_db = AsyncMock()
- mock_provider = MagicMock()
- mock_provider.id = 1
- captured_variables = {}
- async def capture_build(db, event_type, variables):
- captured_variables.update(variables)
- return ("Test", "Test")
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", side_effect=capture_build),
- patch("backend.app.api.routes.settings.get_setting", new_callable=AsyncMock, return_value=None),
- ):
- # Need at least one provider to trigger message building
- mock_get.return_value = [mock_provider]
- await service.on_print_start(
- printer_id=1,
- printer_name="Test",
- data={
- "subtask_name": "test",
- # No estimated_time or mc_remaining_time
- },
- db=mock_db,
- )
- # When no time data, should show "Unknown"
- assert captured_variables.get("estimated_time") == "Unknown"
- @pytest.mark.asyncio
- async def test_print_progress_remaining_time_fallback(self, service):
- """Verify remaining time shows 'Unknown' when not available."""
- mock_db = AsyncMock()
- mock_provider = MagicMock()
- mock_provider.id = 1
- captured_variables = {}
- async def capture_build(db, event_type, variables):
- captured_variables.update(variables)
- return ("Test", "Test")
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", side_effect=capture_build),
- patch("backend.app.api.routes.settings.get_setting", new_callable=AsyncMock, return_value=None),
- ):
- # Need at least one provider to trigger message building
- mock_get.return_value = [mock_provider]
- await service.on_print_progress(
- printer_id=1,
- printer_name="Test",
- progress=50,
- remaining_time=None, # No remaining time
- filename="test.3mf",
- db=mock_db,
- )
- # When no remaining time, should show "Unknown"
- assert captured_variables.get("remaining_time") == "Unknown"
- @pytest.mark.asyncio
- async def test_filename_fallback_to_unknown(self, service):
- """Verify filename defaults to 'Unknown' when not provided."""
- mock_db = AsyncMock()
- mock_provider = MagicMock()
- mock_provider.id = 1
- captured_variables = {}
- async def capture_build(db, event_type, variables):
- captured_variables.update(variables)
- return ("Test", "Test")
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", side_effect=capture_build),
- ):
- # Need at least one provider to trigger message building
- mock_get.return_value = [mock_provider]
- await service.on_print_complete(
- printer_id=1,
- printer_name="Test",
- status="completed",
- data={}, # No subtask_name or filename
- db=mock_db,
- )
- # Filename should default to something (either "Unknown" or cleaned empty)
- assert "filename" in captured_variables
- @pytest.mark.asyncio
- async def test_print_start_uses_archive_print_time_seconds(self, service):
- """Verify print_time_seconds from archive_data is used for estimated_time."""
- mock_db = AsyncMock()
- mock_provider = MagicMock()
- mock_provider.id = 1
- captured_variables = {}
- async def capture_build(db, event_type, variables):
- captured_variables.update(variables)
- return ("Test", "Test")
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", side_effect=capture_build),
- patch("backend.app.api.routes.settings.get_setting", new_callable=AsyncMock, return_value=None),
- ):
- mock_get.return_value = [mock_provider]
- # Pass archive_data with print_time_seconds (7200 seconds = 2 hours)
- await service.on_print_start(
- printer_id=1,
- printer_name="Test",
- data={"subtask_name": "test"},
- db=mock_db,
- archive_data={"print_time_seconds": 7200},
- )
- # Should use archive's print_time_seconds: 7200 seconds = 2h 0m
- assert captured_variables.get("estimated_time") == "2h 0m"
- @pytest.mark.asyncio
- async def test_print_start_archive_data_overrides_mqtt(self, service):
- """Verify archive_data takes priority over MQTT remaining_time."""
- mock_db = AsyncMock()
- mock_provider = MagicMock()
- mock_provider.id = 1
- captured_variables = {}
- async def capture_build(db, event_type, variables):
- captured_variables.update(variables)
- return ("Test", "Test")
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", side_effect=capture_build),
- patch("backend.app.api.routes.settings.get_setting", new_callable=AsyncMock, return_value=None),
- ):
- mock_get.return_value = [mock_provider]
- # Both archive_data and MQTT remaining_time provided
- # Archive says 2 hours, MQTT says 30 minutes (wrong at start)
- await service.on_print_start(
- printer_id=1,
- printer_name="Test",
- data={
- "subtask_name": "test",
- "remaining_time": 1800, # 30 minutes from MQTT
- },
- db=mock_db,
- archive_data={"print_time_seconds": 7200}, # 2 hours from 3MF
- )
- # Should use archive's print_time_seconds (more reliable)
- assert captured_variables.get("estimated_time") == "2h 0m"
- @pytest.mark.asyncio
- async def test_print_start_falls_back_to_mqtt_when_no_archive(self, service):
- """Verify MQTT remaining_time is used when archive_data not provided."""
- mock_db = AsyncMock()
- mock_provider = MagicMock()
- mock_provider.id = 1
- captured_variables = {}
- async def capture_build(db, event_type, variables):
- captured_variables.update(variables)
- return ("Test", "Test")
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", side_effect=capture_build),
- patch("backend.app.api.routes.settings.get_setting", new_callable=AsyncMock, return_value=None),
- ):
- mock_get.return_value = [mock_provider]
- # Only MQTT remaining_time provided (1800 seconds = 30 minutes)
- await service.on_print_start(
- printer_id=1,
- printer_name="Test",
- data={
- "subtask_name": "test",
- "remaining_time": 1800,
- },
- db=mock_db,
- # No archive_data
- )
- # Should use MQTT remaining_time
- assert captured_variables.get("estimated_time") == "30m"
- @pytest.mark.asyncio
- async def test_print_start_eta_calculated_from_estimated_time(self, service):
- """Verify ETA is calculated as wall-clock time from estimated_time."""
- mock_db = AsyncMock()
- mock_provider = MagicMock()
- mock_provider.id = 1
- captured_variables = {}
- async def capture_build(db, event_type, variables):
- captured_variables.update(variables)
- return ("Test", "Test")
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", side_effect=capture_build),
- patch("backend.app.api.routes.settings.get_setting", new_callable=AsyncMock, return_value=None),
- ):
- mock_get.return_value = [mock_provider]
- await service.on_print_start(
- printer_id=1,
- printer_name="Test",
- data={"subtask_name": "test"},
- db=mock_db,
- archive_data={"print_time_seconds": 3600}, # 1 hour
- )
- # ETA should be a time string in HH:MM format
- eta = captured_variables.get("eta")
- assert eta is not None
- assert eta != "Unknown"
- assert ":" in eta # HH:MM format
- @pytest.mark.asyncio
- async def test_print_start_eta_unknown_when_no_time(self, service):
- """Verify ETA shows 'Unknown' when no time data available."""
- mock_db = AsyncMock()
- mock_provider = MagicMock()
- mock_provider.id = 1
- captured_variables = {}
- async def capture_build(db, event_type, variables):
- captured_variables.update(variables)
- return ("Test", "Test")
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", side_effect=capture_build),
- patch("backend.app.api.routes.settings.get_setting", new_callable=AsyncMock, return_value=None),
- ):
- mock_get.return_value = [mock_provider]
- await service.on_print_start(
- printer_id=1,
- printer_name="Test",
- data={"subtask_name": "test"},
- db=mock_db,
- )
- assert captured_variables.get("eta") == "Unknown"
- @pytest.mark.asyncio
- async def test_print_start_eta_respects_12h_format(self, service):
- """Verify ETA uses 12-hour format when time_format is '12h'."""
- mock_db = AsyncMock()
- mock_provider = MagicMock()
- mock_provider.id = 1
- captured_variables = {}
- async def capture_build(db, event_type, variables):
- captured_variables.update(variables)
- return ("Test", "Test")
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", side_effect=capture_build),
- patch("backend.app.api.routes.settings.get_setting", new_callable=AsyncMock, return_value="12h"),
- ):
- mock_get.return_value = [mock_provider]
- await service.on_print_start(
- printer_id=1,
- printer_name="Test",
- data={"subtask_name": "test"},
- db=mock_db,
- archive_data={"print_time_seconds": 3600},
- )
- eta = captured_variables.get("eta")
- assert eta is not None
- # 12h format should contain AM or PM
- assert "AM" in eta or "PM" in eta
- class TestNotificationTemplates:
- """Tests for notification message template rendering."""
- @pytest.fixture
- def service(self):
- return NotificationService()
- @pytest.mark.asyncio
- async def test_template_renders_variables(self, service):
- """Verify template variables are replaced correctly."""
- template_title = "Print {progress}% Complete"
- template_body = "{printer}: {filename}\nRemaining: {remaining_time}"
- variables = {
- "printer": "Test Printer",
- "filename": "test.3mf",
- "progress": "50",
- "remaining_time": "1h 30m",
- }
- title = template_title.format(**variables)
- body = template_body.format(**variables)
- assert title == "Print 50% Complete"
- assert "Test Printer" in body
- assert "test.3mf" in body
- assert "1h 30m" in body
- @pytest.mark.asyncio
- async def test_template_handles_missing_variables(self, service):
- """Verify missing template variables don't cause crashes."""
- template = "{printer}: {unknown_var}"
- variables = {"printer": "Test"}
- # Should handle gracefully - either leave placeholder or skip
- try:
- result = template.format_map({**variables, "unknown_var": "{unknown_var}"})
- assert "Test" in result
- except KeyError:
- pytest.fail("Template should handle missing variables gracefully")
- class TestPrinterErrorNotifications:
- """Tests for HMS error (printer error) notifications."""
- @pytest.fixture
- def service(self):
- return NotificationService()
- @pytest.fixture
- def mock_provider(self):
- """Create a mock notification provider with error notifications enabled."""
- provider = MagicMock()
- provider.id = 1
- provider.name = "Test Provider"
- provider.provider_type = "webhook"
- provider.enabled = True
- provider.config = json.dumps({"webhook_url": "http://test.local/webhook"})
- provider.on_printer_error = True # Enable error notifications
- provider.quiet_hours_enabled = False
- provider.daily_digest_enabled = False
- provider.printer_id = None
- return provider
- @pytest.fixture
- def mock_db(self):
- """Create a mock database session."""
- db = AsyncMock()
- db.commit = AsyncMock()
- return db
- @pytest.mark.asyncio
- async def test_on_printer_error_sends_notification(self, service, mock_provider, mock_db):
- """Verify HMS error notification is sent when triggered."""
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock) as mock_send,
- patch.object(service, "_build_message_from_template", new_callable=AsyncMock) as mock_build,
- ):
- mock_get.return_value = [mock_provider]
- mock_build.return_value = ("Printer Error", "AMS/Filament Error: 0700_8010")
- await service.on_printer_error(
- printer_id=1,
- printer_name="Test Printer",
- error_type="AMS/Filament Error",
- db=mock_db,
- error_detail="Error code: 0700_8010",
- )
- mock_get.assert_called_once()
- mock_send.assert_called_once()
- @pytest.mark.asyncio
- async def test_on_printer_error_skipped_when_disabled(self, service, mock_provider, mock_db):
- """CRITICAL: Verify error notifications respect toggle setting."""
- mock_provider.on_printer_error = False
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock) as mock_send,
- ):
- # Provider with toggle disabled won't be returned
- mock_get.return_value = []
- await service.on_printer_error(
- printer_id=1,
- printer_name="Test",
- error_type="AMS Error",
- db=mock_db,
- error_detail="Test error",
- )
- mock_send.assert_not_called()
- @pytest.mark.asyncio
- async def test_on_printer_error_includes_error_detail(self, service, mock_provider, mock_db):
- """Verify error details are passed to template variables."""
- captured_variables = {}
- async def capture_build(db, event_type, variables):
- captured_variables.update(variables)
- return ("Test", "Test")
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", side_effect=capture_build),
- ):
- mock_get.return_value = [mock_provider]
- await service.on_printer_error(
- printer_id=1,
- printer_name="X1 Carbon",
- error_type="AMS/Filament Error",
- db=mock_db,
- error_detail="Error code: 0700_8010",
- )
- assert captured_variables["printer"] == "X1 Carbon"
- assert captured_variables["error_type"] == "AMS/Filament Error"
- assert captured_variables["error_detail"] == "Error code: 0700_8010"
- @pytest.mark.asyncio
- async def test_on_printer_error_fallback_when_no_detail(self, service, mock_provider, mock_db):
- """Verify fallback message when error_detail is None."""
- captured_variables = {}
- async def capture_build(db, event_type, variables):
- captured_variables.update(variables)
- return ("Test", "Test")
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", side_effect=capture_build),
- ):
- mock_get.return_value = [mock_provider]
- await service.on_printer_error(
- printer_id=1,
- printer_name="Test Printer",
- error_type="Unknown Error",
- db=mock_db,
- error_detail=None, # No detail provided
- )
- assert captured_variables["error_detail"] == "No details available"
- class TestPlateNotEmptyNotifications:
- """Tests for plate not empty (build plate detection) notifications."""
- @pytest.fixture
- def service(self):
- return NotificationService()
- @pytest.fixture
- def mock_provider(self):
- """Create a mock notification provider with plate detection enabled."""
- provider = MagicMock()
- provider.id = 1
- provider.name = "Test Provider"
- provider.provider_type = "webhook"
- provider.enabled = True
- provider.config = json.dumps({"webhook_url": "http://test.local/webhook"})
- provider.on_plate_not_empty = True
- provider.quiet_hours_enabled = False
- provider.daily_digest_enabled = False
- provider.printer_id = None
- return provider
- @pytest.fixture
- def mock_db(self):
- """Create a mock database session."""
- db = AsyncMock()
- db.commit = AsyncMock()
- return db
- @pytest.mark.asyncio
- async def test_on_plate_not_empty_sends_notification(self, service, mock_provider, mock_db):
- """Verify plate not empty notification is sent when triggered."""
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock) as mock_send,
- patch.object(service, "_build_message_from_template", new_callable=AsyncMock) as mock_build,
- ):
- mock_get.return_value = [mock_provider]
- mock_build.return_value = ("Plate Not Empty", "Objects detected on build plate")
- await service.on_plate_not_empty(
- printer_id=1,
- printer_name="Test Printer",
- db=mock_db,
- difference_percent=5.2,
- )
- mock_get.assert_called_once()
- mock_send.assert_called_once()
- # Verify force_immediate is True (critical alert)
- call_kwargs = mock_send.call_args[1]
- assert call_kwargs.get("force_immediate") is True
- @pytest.mark.asyncio
- async def test_on_plate_not_empty_skipped_when_disabled(self, service, mock_provider, mock_db):
- """Verify notification is skipped when toggle is disabled."""
- mock_provider.on_plate_not_empty = False
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock) as mock_send,
- ):
- mock_get.return_value = []
- await service.on_plate_not_empty(
- printer_id=1,
- printer_name="Test",
- db=mock_db,
- )
- mock_send.assert_not_called()
- @pytest.mark.asyncio
- async def test_on_plate_not_empty_includes_difference_percent(self, service, mock_provider, mock_db):
- """Verify difference percentage is passed to template variables."""
- captured_variables = {}
- async def capture_build(db, event_type, variables):
- captured_variables.update(variables)
- return ("Test", "Test")
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", side_effect=capture_build),
- ):
- mock_get.return_value = [mock_provider]
- await service.on_plate_not_empty(
- printer_id=1,
- printer_name="X1 Carbon",
- db=mock_db,
- difference_percent=3.5,
- )
- assert captured_variables["printer"] == "X1 Carbon"
- assert captured_variables["difference_percent"] == "3.5"
- class TestBedCooledNotifications:
- """Tests for bed cooled (after print) notifications."""
- @pytest.fixture
- def service(self):
- return NotificationService()
- @pytest.fixture
- def mock_provider(self):
- """Create a mock notification provider with bed cooled enabled."""
- provider = MagicMock()
- provider.id = 1
- provider.name = "Test Provider"
- provider.provider_type = "webhook"
- provider.enabled = True
- provider.config = json.dumps({"webhook_url": "http://test.local/webhook"})
- provider.on_bed_cooled = True
- provider.quiet_hours_enabled = False
- provider.daily_digest_enabled = False
- provider.printer_id = None
- return provider
- @pytest.fixture
- def mock_db(self):
- """Create a mock database session."""
- db = AsyncMock()
- db.commit = AsyncMock()
- return db
- @pytest.mark.asyncio
- async def test_on_bed_cooled_sends_notification(self, service, mock_provider, mock_db):
- """Verify bed cooled notification is sent when triggered."""
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock) as mock_send,
- patch.object(service, "_build_message_from_template", new_callable=AsyncMock) as mock_build,
- ):
- mock_get.return_value = [mock_provider]
- mock_build.return_value = ("Bed Cooled", "Test Printer: Bed cooled to 30°C")
- await service.on_bed_cooled(
- printer_id=1,
- printer_name="Test Printer",
- bed_temp=30.0,
- threshold=35.0,
- filename="benchy.3mf",
- db=mock_db,
- )
- mock_get.assert_called_once()
- mock_send.assert_called_once()
- @pytest.mark.asyncio
- async def test_on_bed_cooled_skipped_when_no_providers(self, service, mock_db):
- """Verify notification is skipped when no providers have bed cooled enabled."""
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock) as mock_send,
- ):
- mock_get.return_value = []
- await service.on_bed_cooled(
- printer_id=1,
- printer_name="Test Printer",
- bed_temp=30.0,
- threshold=35.0,
- filename="benchy.3mf",
- db=mock_db,
- )
- mock_send.assert_not_called()
- @pytest.mark.asyncio
- async def test_on_bed_cooled_includes_correct_variables(self, service, mock_provider, mock_db):
- """Verify bed temp, threshold, and filename are passed to template variables."""
- captured_variables = {}
- async def capture_build(db, event_type, variables):
- captured_variables.update(variables)
- return ("Test", "Test")
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", side_effect=capture_build),
- ):
- mock_get.return_value = [mock_provider]
- await service.on_bed_cooled(
- printer_id=1,
- printer_name="X1 Carbon",
- bed_temp=28.7,
- threshold=35.0,
- filename="benchy.gcode.3mf",
- db=mock_db,
- )
- assert captured_variables["printer"] == "X1 Carbon"
- assert captured_variables["bed_temp"] == "29"
- assert captured_variables["threshold"] == "35"
- assert captured_variables["filename"] == "benchy"
- @pytest.mark.asyncio
- async def test_on_bed_cooled_handles_none_filename(self, service, mock_provider, mock_db):
- """Verify None filename is handled gracefully."""
- captured_variables = {}
- async def capture_build(db, event_type, variables):
- captured_variables.update(variables)
- return ("Test", "Test")
- with (
- patch.object(service, "_get_providers_for_event", new_callable=AsyncMock) as mock_get,
- patch.object(service, "_send_to_providers", new_callable=AsyncMock),
- patch.object(service, "_build_message_from_template", side_effect=capture_build),
- ):
- mock_get.return_value = [mock_provider]
- await service.on_bed_cooled(
- printer_id=1,
- printer_name="Test Printer",
- bed_temp=30.0,
- threshold=35.0,
- filename=None,
- db=mock_db,
- )
- assert captured_variables["filename"] == "Unknown"
|