Fix unit test to pass

tests/unit_tests/test_lobbyconnection.py

@@ -161,22 +161,28 @@ async def test_bad_command_calls_abort(lobbyconnection):
 
 
 async def test_database_outage_error_responds_cleanly(lobbyconnection):
-    lobbyconnection.send = mock.AsyncMock()
     lobbyconnection.abort = mock.AsyncMock()
+    lobbyconnection.check_policy_conformity = mock.AsyncMock(return_value=True)
+    lobbyconnection.send = mock.AsyncMock()
+
+    def mock_ensure_authenticated(cmd):
+        raise OperationalError(statement='', params=[], orig=None)
+
+    lobbyconnection.ensure_authenticated = mock_ensure_authenticated
 
-    with pytest.raises(OperationalError):
-        await lobbyconnection.on_message_received({
-            "command": "hello",
-            "login": "test",
-            "password": sha256(b"test_password").hexdigest(),
-            "unique_id": "blah"
-        })
+    await lobbyconnection.on_message_received({
+        "command": "hello",
+        "login": "test",
+        "password": sha256(b"test_password").hexdigest(),
+        "unique_id": "blah"
+    })
 
     lobbyconnection.send.assert_called_once_with({
         "command": "notice",
         "style": "error",
         "text": "Unable to connect to database. Please try again later."
     })
+
     lobbyconnection.abort.assert_called_once_with("Error connecting to database")