Fix tests

jdgsmallwood · Jul 10, 2024 · 17e2217 · 17e2217
1 parent 22b6286
commit 17e2217
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/test_lambda_function.py b/src/test_lambda_function.py
@@ -18,7 +18,7 @@ def test_handler_uv_above_threshold(mocker):
 
     handler({}, {})
 
-    mock_send_message.assert_called_once_with("UV is 4.0 > 3.0 - be sunsmart!")
+    mock_send_message.assert_called_once_with("UV is 4.0 > 3.0 - be sunsmart! UV observations courtesy of ARPANSA.")
 
 
 def test_handler_uv_below_threshold(mocker):
@@ -38,7 +38,7 @@ def test_handler_uv_below_threshold(mocker):
 
     handler({}, {})
 
-    mock_send_message.assert_called_once_with("UV is safe in Melbourne right now.")
+    mock_send_message.assert_called_once_with("UV is safe in Melbourne right now. UV observations courtesy of ARPANSA.")
 
 
 def test_handler_invalid_uv_value(mocker):