mirror of
https://github.com/PostHog/posthog.git
synced 2024-11-28 00:46:45 +01:00
45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
|
import pytest
|
||
|
|
from infi.clickhouse_orm import Database
|
||
|
|
|
||
|
|
from ee.clickhouse.client import sync_execute
|
||
|
|
from ee.clickhouse.models.person import Person
|
||
|
|
from posthog.settings import (
|
||
|
|
CLICKHOUSE_DATABASE,
|
||
|
|
CLICKHOUSE_HTTP_URL,
|
||
|
|
CLICKHOUSE_PASSWORD,
|
||
|
|
CLICKHOUSE_USER,
|
||
|
|
CLICKHOUSE_VERIFY,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture(scope="package")
|
||
|
|
def django_db_setup(django_db_setup, django_db_keepdb):
|
||
|
|
database = Database(
|
||
|
|
CLICKHOUSE_DATABASE,
|
||
|
|
db_url=CLICKHOUSE_HTTP_URL,
|
||
|
|
username=CLICKHOUSE_USER,
|
||
|
|
password=CLICKHOUSE_PASSWORD,
|
||
|
|
verify_ssl_cert=CLICKHOUSE_VERIFY,
|
||
|
|
)
|
||
|
|
|
||
|
|
if not django_db_keepdb:
|
||
|
|
try:
|
||
|
|
database.drop_database()
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
|
||
|
|
if not django_db_keepdb or not database.db_exists:
|
||
|
|
database.create_database()
|
||
|
|
|
||
|
|
database.migrate("ee.clickhouse.migrations")
|
||
|
|
# Make DELETE / UPDATE synchronous to avoid flaky tests
|
||
|
|
sync_execute("SET mutations_sync = 1")
|
||
|
|
|
||
|
|
yield
|
||
|
|
|
||
|
|
if not django_db_keepdb:
|
||
|
|
try:
|
||
|
|
database.drop_database()
|
||
|
|
except:
|
||
|
|
pass
|