chore: import upstream snapshot with attribution

python/tests/resource/postgres/__init__.py

@@ -0,0 +1,13 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========

python/tests/resource/postgres/test_accessor.py

@@ -0,0 +1,23 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+
+from mirage.accessor.postgres import PostgresAccessor
+from mirage.resource.postgres.config import PostgresConfig
+
+
+def test_accessor_holds_config():
+    cfg = PostgresConfig(dsn="postgres://localhost/db")
+    a = PostgresAccessor(cfg)
+    assert a.config is cfg
+    assert a._pool is None

python/tests/resource/postgres/test_config.py

@@ -0,0 +1,31 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+
+from mirage.resource.postgres.config import PostgresConfig
+from mirage.resource.secrets import reveal_secret
+
+
+def test_defaults():
+    cfg = PostgresConfig(dsn="postgres://localhost/db")
+    assert reveal_secret(cfg.dsn) == "postgres://localhost/db"
+    assert cfg.schemas is None
+    assert cfg.default_row_limit == 1000
+    assert cfg.max_read_rows == 10_000
+    assert cfg.max_read_bytes == 10 * 1024 * 1024
+    assert cfg.default_search_limit == 100
+
+
+def test_schema_filter():
+    cfg = PostgresConfig(dsn="postgres://localhost/db", schemas=["public"])
+    assert cfg.schemas == ["public"]

python/tests/resource/postgres/test_resource.py

@@ -0,0 +1,58 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+
+from mirage.resource.postgres import PostgresConfig, PostgresResource
+
+
+def test_resource_name():
+    res = PostgresResource(PostgresConfig(dsn="postgres://localhost/db"))
+    assert res.name == "postgres"
+    assert res.caches_reads is False
+
+
+def test_resource_registers_three_ops():
+    res = PostgresResource(PostgresConfig(dsn="postgres://localhost/db"))
+    op_names = {ro.name for ro in res.ops_list()}
+    assert {"read", "readdir", "stat"} <= op_names
+
+
+def test_resource_registers_commands():
+    res = PostgresResource(PostgresConfig(dsn="postgres://localhost/db"))
+    cmd_names = {rc.name for rc in res.commands()}
+    expected = {
+        "cat", "find", "head", "jq", "ls", "stat", "tail", "tree", "wc",
+        "grep", "rg"
+    }
+    assert expected <= cmd_names
+
+
+def test_resource_in_registry():
+    from mirage.resource.registry import REGISTRY, build_resource
+
+    assert "postgres" in REGISTRY
+    res = build_resource("postgres", config={"dsn": "postgres://localhost/db"})
+    assert res.name == "postgres"
+
+
+def test_resource_get_state_redacts_dsn():
+    res = PostgresResource(PostgresConfig(dsn="postgres://user:pw@host/db"))
+    state = res.get_state()
+    assert state["type"] == "postgres"
+    assert state["config"]["dsn"] == "<REDACTED>"
+    assert "redacted_fields" not in state
+
+
+def test_resource_load_state_noop():
+    res = PostgresResource(PostgresConfig(dsn="postgres://localhost/db"))
+    res.load_state({"type": "postgres"})