return nothing on schema get

getredash · Aug 31, 2023 · d600a4b · d600a4b
1 parent 126e7e5
commit d600a4b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/models/test_data_sources.py b/tests/models/test_data_sources.py
@@ -92,6 +92,7 @@ def test_expires_schema(self, mock_redis):
         expected_ttl = 606600
 
         with mock.patch("redash.query_runner.pg.PostgreSQL.get_schema") as patched_get_schema:
+            patched_get_schema.return_value = None
             data_source = self.factory.data_source.get_schema(refresh=True)
 
         mock_redis.assert_called_with(data_source._schema_key, ex=expected_ttl)