rs-mock 0.1.0__tar.gz

rs_mock-0.1.0/PKG-INFO

@@ -0,0 +1,49 @@
+Metadata-Version: 2.3
+Name: rs-mock
+Version: 0.1.0
+Summary: Mock Redshift locally by transpiling Redshift SQL to DuckDB and running it.
+Author: Tyler Riccio
+Author-email: Tyler Riccio <tylerriccio8@gmail.com>
+Requires-Dist: duckdb>=1.5.4
+Requires-Dist: sqlglot>=30.12.0
+Requires-Python: >=3.14
+Description-Content-Type: text/markdown
+
+# rs-mock
+
+A super lightweight, in-process **Redshift mocker** for test suites.
+
+rs-mock transpiles Redshift SQL to duckdb SQL with [sqlglot](https://github.com/tobymao/sqlglot)
+and runs it against an in-memory [duckdb](https://duckdb.org/) database. No cluster,
+no network, no Docker — just fast, in-process SQL.
+
+## Usage
+
+```python
+from rs_mock import RedshiftMock
+
+rs = RedshiftMock()
+
+# DDL / DML — state persists across calls for the instance's lifetime
+rs.execute("CREATE TABLE users (id INT, name VARCHAR)")
+rs.execute("INSERT INTO users VALUES (1, 'alice'), (2, 'bob')")
+
+# SELECTs, JOINs, CTEs — execute() returns the duckdb cursor
+rows = rs.execute("SELECT id, name FROM users ORDER BY id").fetchall()
+# [(1, 'alice'), (2, 'bob')]
+
+# Need duckdb power features? Grab the cursor or the connection.
+df = rs.execute("SELECT * FROM users").df()
+rs.connection  # the underlying duckdb connection
+```
+
+Supported: regular selects, joins, CTEs, and DDL/DML. Redshift-specific syntax
+(e.g. `GETDATE()`) is rewritten to its duckdb equivalent automatically.
+
+## Development
+
+```bash
+make test   # run tests
+make lint   # ruff + pyrefly
+make prek   # pre-commit hooks
+```

rs_mock-0.1.0/README.md

@@ -0,0 +1,38 @@
+# rs-mock
+
+A super lightweight, in-process **Redshift mocker** for test suites.
+
+rs-mock transpiles Redshift SQL to duckdb SQL with [sqlglot](https://github.com/tobymao/sqlglot)
+and runs it against an in-memory [duckdb](https://duckdb.org/) database. No cluster,
+no network, no Docker — just fast, in-process SQL.
+
+## Usage
+
+```python
+from rs_mock import RedshiftMock
+
+rs = RedshiftMock()
+
+# DDL / DML — state persists across calls for the instance's lifetime
+rs.execute("CREATE TABLE users (id INT, name VARCHAR)")
+rs.execute("INSERT INTO users VALUES (1, 'alice'), (2, 'bob')")
+
+# SELECTs, JOINs, CTEs — execute() returns the duckdb cursor
+rows = rs.execute("SELECT id, name FROM users ORDER BY id").fetchall()
+# [(1, 'alice'), (2, 'bob')]
+
+# Need duckdb power features? Grab the cursor or the connection.
+df = rs.execute("SELECT * FROM users").df()
+rs.connection  # the underlying duckdb connection
+```
+
+Supported: regular selects, joins, CTEs, and DDL/DML. Redshift-specific syntax
+(e.g. `GETDATE()`) is rewritten to its duckdb equivalent automatically.
+
+## Development
+
+```bash
+make test   # run tests
+make lint   # ruff + pyrefly
+make prek   # pre-commit hooks
+```

rs_mock-0.1.0/pyproject.toml

@@ -0,0 +1,30 @@
+[project]
+name = "rs-mock"
+version = "0.1.0"
+description = "Mock Redshift locally by transpiling Redshift SQL to DuckDB and running it."
+readme = "README.md"
+authors = [
+    { name = "Tyler Riccio", email = "tylerriccio8@gmail.com" }
+]
+requires-python = ">=3.14"
+dependencies = [
+    "duckdb>=1.5.4",
+    "sqlglot>=30.12.0",
+]
+
+[build-system]
+requires = ["uv_build>=0.11.18,<0.12.0"]
+build-backend = "uv_build"
+
+[dependency-groups]
+dev = [
+    "hypothesis>=6.156.2",
+    "prek>=0.4.8",
+    "pyrefly>=1.1.1",
+    "pytest>=9.1.1",
+    "pytest-cov>=7.1.0",
+    "pytest-randomly>=4.1.0",
+    "pytest-testmap>=0.3.0",
+    "pytest-xdist>=3.8.0",
+    "ruff>=0.15.20",
+]

rs_mock-0.1.0/src/rs_mock/__init__.py

@@ -0,0 +1,5 @@
+"""rs-mock: a lightweight in-process Redshift mocker built on duckdb + sqlglot."""
+
+from rs_mock.mock import RedshiftMock, UnimplementedPostgresFeature
+
+__all__ = ["RedshiftMock", "UnimplementedPostgresFeature"]

rs_mock-0.1.0/src/rs_mock/mock.py

@@ -0,0 +1,57 @@
+"""The RedshiftMock: run Redshift SQL against an in-memory duckdb database."""
+
+from __future__ import annotations
+
+import duckdb
+import sqlglot
+
+
+class UnimplementedPostgresFeature(Exception):
+    """Raised when SQL uses a PostgreSQL construct that Redshift does not support.
+
+    Redshift excludes many PostgreSQL functions, data types, and features (see the
+    AWS "Unsupported PostgreSQL ..." docs). Since this mock stands in for Redshift,
+    such SQL must fail loudly here rather than silently succeed on duckdb.
+    """
+
+
+class RedshiftMock:
+    """An in-process stand-in for Redshift.
+
+    Redshift SQL is transpiled to duckdb via sqlglot, then executed against a
+    persistent in-memory duckdb connection so state (tables, inserted rows)
+    survives across `execute` calls for the lifetime of the instance.
+    """
+
+    def __init__(self) -> None:
+        self._conn = duckdb.connect(":memory:")
+
+    @property
+    def connection(self) -> duckdb.DuckDBPyConnection:
+        """The underlying duckdb connection, for power users."""
+        return self._conn
+
+    def execute(self, sql: str) -> duckdb.DuckDBPyConnection:
+        """Transpile Redshift `sql` to duckdb and execute it.
+
+        A single string may hold multiple statements; each is transpiled and
+        run in order. Returns the duckdb cursor of the final statement, so
+        callers can `.fetchall()`, `.df()`, `.arrow()`, etc.
+        """
+        # `transpile` splits multi-statement input and parses with the redshift
+        # dialect; blank input transpiles to a single empty string, so filter
+        # those out. Executing nothing is a user error.
+        statements = [
+            s for s in sqlglot.transpile(sql, read="redshift", write="duckdb") if s
+        ]
+        if not statements:
+            raise ValueError("no SQL statement to execute")
+
+        result = self._conn
+        for statement in statements:
+            result = self._conn.execute(statement)
+        return result
+
+    def close(self) -> None:
+        """Close the underlying duckdb connection."""
+        self._conn.close()