fix args

astro-datalab · Dec 13, 2024 · d951ab7 · d951ab7
1 parent 377817b
commit d951ab7
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/dlairflow/test/test_postgresql.py b/dlairflow/test/test_postgresql.py
@@ -27,6 +27,20 @@ def mock_connection(connection):
 
     monkeypatch.setattr(BaseHook, "get_connection", mock_connection)
 
-    test_operator = pg_dump_schema("login,password,host,schema")
+    test_operator = pg_dump_schema("login,password,host,schema", "dump_schema")
+
+    print(dir(test_operator))
+
+
+def test_pg_dump_schema_alt_dir(monkeypatch):
+    """Test pg_dump task with alternate directory.
+    """
+    def mock_connection(connection):
+        conn = MockConnection(connection)
+        return conn
+
+    monkeypatch.setattr(BaseHook, "get_connection", mock_connection)
+
+    test_operator = pg_dump_schema("login,password,host,schema", "dump_schema", "dump_dir")
 
     print(dir(test_operator))