Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Move docs #35

Merged
merged 3 commits into from
Jan 14, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
31 changes: 0 additions & 31 deletions Makefile

This file was deleted.

197 changes: 180 additions & 17 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -8,17 +8,25 @@

The primary purpose of this library is to simplify the testing of SQL data models and queries by allowing users to mock input data and create tests for various scenarios. It provides a consistent and convenient way to test the execution of your query without the need to process a massive amount of data.

# Documentation

A full documentation can be found [on the documentation page](https://deeplcom.github.io/sql-mock/)

The library currently supports the following databases.
* BigQuery
* Clickhouse
* Redshift
* Snowflake

## Installation
## Documentation

* [Installation](#installation)
* [Quickstart](#quickstart)
* Basic usage
* [Create table mocks](/docs/defining_table_mocks.md)
* [Specifying default values](/docs/default_values.md)
* [Specifying the query to test](/docs/your_sql_query_to_test.md)
* [Result assertions](/docs/result_assertion.md)
* System specific usage
* [Use with BigQuery](/docs/bigquery.md)
* [Use with Clickhouse](/docs/clickhouse.md)
* [Use with Redshift](/docs/redshift.md)
* [Use with Snowflake](/docs/snowflake.md)
* [Use with dbt](/docs/dbt.md)

You can find some examples in the [examples folder](https://github.com/DeepLcom/sql-mock/tree/main/examples).

### Installation

The library can be installed from [PyPI](https://pypi.org/project/sql-mock/) using pip:

Expand All @@ -42,11 +50,105 @@ If you need to modify this source code, install the dependencies using poetry:
poetry install --all-extras
```

#### Recommended Setup for Pytest

If you are using pytest, make sure to add a `conftest.py` file to the root of your project.
In the file add the following lines:

```python
import pytest
pytest.register_assert_rewrite('sql_mock')
```

This allows you to get a rich comparison when using the `.assert_equal` method on the table mock instances.

We also recommend using [pytest-icdiff](https://github.com/hjwp/pytest-icdiff) for better visibility on diffs of failed tests.

### Quickstart

Before diving into specific database scenarios, let's start with a simplified example of how SQL Mock works behind the scenes.

1. You have an original SQL query, for instance:

```sql
-- path/to/query_for_result_table.sql
SELECT id FROM data.table1
```

2. Using SQL Mock, you define table mocks. You can use the built-in column types provided by SQL Mock. Available column types include `Int`, `String`, `Date`, and more. Each database type has their own column types. Define your tables by subclassing a mock table class that fits your database (e.g. `BigQueryTableMock`) and specifying the column types along with default values. In our example we use the `ClickHouseTableMock` class

```python
from sql_mock.clickhouse import column_mocks as col
from sql_mock.clickhouse.table_mocks import ClickHouseTableMock
from sql_mock.table_mocks import table_meta

@table_meta(table_ref='data.table1')
class Table(ClickHouseTableMock):
id = col.Int(default=1)
name = col.String(default='Peter')

@table_meta(table_ref='data.result_table', query_path='path/to/query_for_result_table.sql')
class ResultTable(ClickHouseTableMock):
id = col.Int(default=1)
```

3. **Creating mock data:** Define mock data for your tables using dictionaries. Each dictionary represents a row in the table, with keys corresponding to column names. Table column keys that don't get a value will use the default.

```python
user_data = [
{}, # This will use the defaults for both id and name
{'id': 2, 'name': 'Martin'},
{'id': 3}, # This will use defaults for the name
]

input_table_mock = Table.from_dicts(user_data)
```

4. **Getting results for a table mock:** Use the `from_mocks` method of the table mock object to generate mock query results based on your mock data.

```python
res = ResultTable.from_mocks(input_data=[input_table_mock])
```

5. Behind the scene SQL Mock replaces table references (e.g. `data.table1`) in your query with Common Table Expressions (CTEs) filled with dummy data. It can roughly be compared to something like this:

```sql
WITH data__table1 AS (
-- Mocked inputs
SELECT
cast('1' AS 'String') AS id,
cast('Peter' AS 'String') AS name
UNION ALL
SELECT
cast('2' AS 'String') AS id,
cast('Martin' AS 'String') AS name
UNION ALL
SELECT
cast('3' AS 'String') AS id,
cast('Peter' AS 'String') AS name
)

result AS (
-- Original query with replaced references
SELECT id FROM data__table1
)

SELECT
cast(id AS 'String') AS id
FROM result
```

6. Finally, you can compare your results to some expected results using the `assert_equal` method.

```python
expected = [{'id': '1'},{'id': '2'},{'id': '3'}]
res.assert_equal(expected)
```

## Pydantic V1 vs. V2

SQL Mock's published version supports Pydantic V2. You might run into issues when your code depends on Pydantic V1.
We have an alternative branch you can install from that supports Pydantic V1 in the meanwhile: https://github.com/DeepLcom/sql-mock/tree/pydantic-v1

We have an alternative branch you can install from that supports Pydantic V1 in the meanwhile: <https://github.com/DeepLcom/sql-mock/tree/pydantic-v1>

## Contributing

Expand All @@ -56,10 +158,10 @@ We welcome contributions to improve and enhance this open-source project. Whethe

If you encounter a bug, have a feature request, or face any issues with the project, we encourage you to report them using the project's issue tracker. When creating an issue, please include the following information:

- A clear and descriptive title.
- A detailed description of the problem or suggestion.
- Steps to reproduce the issue (if applicable).
- Any error messages or screenshots that help clarify the problem.
* A clear and descriptive title.
* A detailed description of the problem or suggestion.
* Steps to reproduce the issue (if applicable).
* Any error messages or screenshots that help clarify the problem.

### Feature Requests

Expand All @@ -79,3 +181,64 @@ The SQL Mock Buddy can be accessed here: [https://chat.openai.com/g/g-FIXNcqu1l-
SQL Mock Buddy should help you to get started quickly with SQL Mock.

It is still in beta mode and you should definitely double-check its output!

## FAQ

### My database system is not supported yet but I want to use SQL Mock. What should I do?

We are planning to add more and more supported database systems. However, if your system is not supported yet, you can still use SQL Mock. There are only 2 things you need to do:

#### Create your `TableMock` class

First, you need to create a `TableMock` class for your database system that inherits from `sql_mock.table_mocks.BaseTableMock`.

That class needs to implement the `_get_results` method which should make sure to fetch the results of a query (e.g. produced by `self._generate_query()`) and return it as list of dictionaries.

Look at one of the existing client libraries to see how this could work (e.g. [BigQueryTableMock](https://github.com/DeepLcom/sql-mock/blob/main/src/sql_mock/bigquery/table_mocks.py)).

You might want to create a settings class as well in case you need some specific connection settings to be available within the `_get_results` method.

#### Create your `ColumnMocks`

Your database system might support specific database types. In order to make them available as column types, you can use the `sql_mock.column_mocks.BaseColumnMock` class as a base and inherit your specific column types from it.
For most of your column mocks you might only need to specify the `dtype` that should be used to parse the inputs.

A good practise is to create a `BaseColumnMock` class that is specific to your database and inherit all your column types from it, e.g.:

```python
from sql_mock.column_mocks import BaseColumnMock

class MyFanceDatabaseColumnMock(BaseColumnMock):
# In case you need some specific logic that overwrites the default behavior, you can do so here
pass

class Int(MyFanceDatabaseColumnMock):
dtype = "Integer"

class String(MyFanceDatabaseColumnMock):
dtype = "String"
```

#### Contribute your database setup

There will definitely be folks in the community that are in the need of support for the database you just created all the setup for.
Feel free to create a PR on this repository that we can start supporting your database system!

### I am missing a specific BaseColumnMock type for my model fields

We implemented some basic column types but it could happen that you don't find the one you need.
Luckily, you can easily create those with the tools provided.
The only thing you need to do is to inherit from the `BaseColumnMock` that is specific to your database system (e.g. `BigQueryColumnMock`) and write classes for the column mocks you are missing. Usually you only need to set the correct `dtype`. This would later be used in the `cast(col to <dtype>)` expression.

```python
# Replace the import with the database system you are using
from sql_mock.bigquery.column_mock import BigQueryColumnMock

class MyFancyMissingColType(BigQueryColumnMock):
dtype = "FancyMissingColType"

# In case you need to implement additional logic for casting, you can do so here
...
```

**Don't forget to create a PR in case you feel that your column mock type could be useful for the community**!
4 changes: 0 additions & 4 deletions docs/.buildinfo

This file was deleted.

Empty file removed docs/.nojekyll
Empty file.
62 changes: 0 additions & 62 deletions docs/_sources/faq.md.txt

This file was deleted.

39 changes: 0 additions & 39 deletions docs/_sources/getting_started/installation.md.txt

This file was deleted.

Loading