fix tests

guardrails-ai · Feb 21, 2024 · 56ba026 · 56ba026
1 parent 439b2bc
commit 56ba026
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/tests/blueprints/test_guards.py b/tests/blueprints/test_guards.py
@@ -13,6 +13,7 @@ def test_guards_get(mocker):
     mocker.patch("flask.Blueprint", new=MockBlueprint)
     mocker.patch("flask.request", mock_request)
     mocker.patch("src.clients.guard_client.GuardClient", new=MockGuardClient)
+    mocker.patch("opentelemetry.trace.get_tracer_provider")
     from src.blueprints.guards import guards, guards_bp
 
     response = guards()

diff --git a/tests/blueprints/test_root.py b/tests/blueprints/test_root.py
@@ -40,10 +40,8 @@ def __init__(self):
     assert root_bp.routes == ["/", "/health-check"]
     text.assert_called_once_with("SELECT count(datid) FROM pg_stat_activity;")
     assert mock_pg.db.session.queries == ["SELECT count(datid) FROM pg_stat_activity;"]
-
-    assert info_spy.call_count == 1
-
-    info_spy.assert_called_once_with("response: ", [(1,)])
+
+    info_spy.assert_called_once_with("response: %s", [(1,)])
     assert response == { "status": 200, "message": "Ok" }
 
     mocker.resetall()