def test_col_split_conditions(
id_column: str, project_column: str, timestamp_column: str, query, expected_result
) -> None:
dataset = get_dataset("events")
query = parse_query(query, dataset)
splitter = ColumnSplitQueryStrategy(id_column, project_column, timestamp_column)
request = Request("a", query, HTTPRequestSettings(), {}, "r")
entity = get_entity(query.get_from_clause().key)
plan = entity.get_query_plan_builder().build_plan(request)
def do_query(
query: ClickhouseQuery, request_settings: RequestSettings,
) -> QueryResult:
return QueryResult(
{
"data": [
{
id_column: "asd123",
project_column: 123,
timestamp_column: "2019-10-01 22:33:42",
}
]
},
{},
)
assert (
splitter.execute(plan.query, HTTPRequestSettings(), do_query) is not None
) == expected_result
def test_col_split_conditions(id_column: str, project_column: str,
timestamp_column: str, query,
expected_result) -> None:
dataset = get_dataset("events")
query = parse_query(query, dataset)
splitter = ColumnSplitQueryStrategy(id_column, project_column,
timestamp_column)
def do_query(query: ClickhouseQuery,
request_settings: RequestSettings = None) -> QueryResult:
return QueryResult(
{
"data": [{
id_column: "asd123",
project_column: 123,
timestamp_column: "2019-10-01 22:33:42",
}]
},
{},
)
assert (splitter.execute(query, HTTPRequestSettings(), do_query)
is not None) == expected_result