counterweight/lolafect
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add tests for trino feature

Browse source
This commit is contained in:
Pablo Martin 2023-03-29 17:18:00 +02:00
parent c3507cccec
commit b51c810a39
1 changed files with 39 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

40
tests/test_integration/test_data_testing.py
View file

@ -1,7 +1,7 @@
from great_expectations.core.expectation_configuration import ExpectationConfiguration
from lolafect.lolaconfig import build_lolaconfig
from lolafect.data_testing import run_data_test_on_mysql
from lolafect.data_testing import run_data_test_on_mysql, run_data_test_on_trino
from lolafect.connections import open_ssh_tunnel_with_s3_pkey, close_ssh_tunnel
# __ __ _____ _ _ _____ _ _ _____ _
@ -110,3 +110,41 @@ def test_validation_on_mysql_fails():
data_test_failed = validation_result["success"] == False
assert data_test_failed
def test_validation_on_trino_succeeds():
validation_result = run_data_test_on_trino.run(
name="lolafect-testing-test_validation_on_mysql_fails",
trino_credentials={
"host": TEST_LOLACONFIG.TRINO_CREDENTIALS["host"],
"port": TEST_LOLACONFIG.TRINO_CREDENTIALS["port"],
"user": TEST_LOLACONFIG.TRINO_CREDENTIALS["user"],
"password": TEST_LOLACONFIG.TRINO_CREDENTIALS["password"],
"db": TEST_LOLACONFIG.TRINO_CREDENTIALS["default_db"],
},
query=TEST_QUERY,
expectation_configurations=TEST_EXPECTATIONS_THAT_FIT_DATA,
)
data_test_passed = validation_result["success"] == True
assert data_test_passed
def test_validation_on_trino_fails():
validation_result = run_data_test_on_trino.run(
name="lolafect-testing-test_validation_on_mysql_fails",
trino_credentials={
"host": TEST_LOLACONFIG.TRINO_CREDENTIALS["host"],
"port": TEST_LOLACONFIG.TRINO_CREDENTIALS["port"],
"user": TEST_LOLACONFIG.TRINO_CREDENTIALS["user"],
"password": TEST_LOLACONFIG.TRINO_CREDENTIALS["password"],
"db": TEST_LOLACONFIG.TRINO_CREDENTIALS["default_db"],
},
query=TEST_QUERY,
expectation_configurations=TEST_EXPECTATIONS_THAT_DONT_FIT_DATA,
)
data_test_failed = validation_result["success"] == False
assert data_test_failed