thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-22 23:52:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #55769 from AVMusorin/capn-proto-drop-cache

Browse Source 
Added test to check CapnProto cache
This commit is contained in:
Kruglov Pavel 2023-10-19 15:59:46 +02:00 committed by GitHub
commit 85e3c31f14
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 63 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
tests/integration/test_format_schema_on_server/test.py
View File

@ -112,3 +112,66 @@ message MessageTmp {
"SELECT * FROM test.simple FORMAT Protobuf SETTINGS format_schema='message_tmp:MessageTmp'"
)
assert "NO_COLUMNS_SERIALIZED_TO_PROTOBUF_FIELDS)" in str(exc.value)
def test_drop_capn_proto_format(started_cluster):
create_simple_table()
instance.query("INSERT INTO test.simple VALUES (1, 'abc'), (2, 'def')")
capn_proto_schema = """
@0x801f030c2b67bf19;
struct MessageTmp {
key @0 :UInt64;
value @1 :Text;
}
"""
capn_schema_path_name = "message_tmp.capnp"
database_path = os.path.abspath(os.path.join(instance.path, "database"))
format_schemas_path = os.path.join(database_path, "format_schemas")
with open(os.path.join(format_schemas_path, capn_schema_path_name), "w") as file:
file.write(capn_proto_schema)
assert instance.http_query(
"SELECT * FROM test.simple FORMAT CapnProto SETTINGS format_schema='message_tmp:MessageTmp'"
) == instance.query(
f"SELECT * FROM test.simple Format CapnProto SETTINGS format_schema='{format_schemas_path}/message_tmp:MessageTmp'"
)
new_schema = """
@0x801f030c2b67bf19;
struct MessageTmp {
key2 @0 :UInt64;
value2 @1 :Text;
}
"""
with open(os.path.join(format_schemas_path, capn_schema_path_name), "w") as file:
file.write(new_schema)
instance.query("DROP TABLE IF EXISTS test.new_simple")
instance.query(
"""
CREATE TABLE test.new_simple (key2 UInt64, value2 String)
ENGINE = MergeTree ORDER BY tuple();
"""
)
instance.query("INSERT INTO test.new_simple VALUES (1, 'abc'), (2, 'def')")
# instance.query("SYSTEM DROP FORMAT SCHEMA CACHE FOR CapnProto")
# Tets works with new scheme
assert instance.http_query(
"SELECT * FROM test.new_simple FORMAT CapnProto SETTINGS format_schema='message_tmp:MessageTmp'"
) == instance.query(
f"SELECT * FROM test.new_simple Format CapnProto SETTINGS format_schema='{format_schemas_path}/message_tmp:MessageTmp'"
)
# Tests that stop working with old scheme
with pytest.raises(Exception) as exc:
instance.http_query(
"SELECT * FROM test.simple FORMAT CapnProto SETTINGS format_schema='message_tmp:MessageTmp'"
)
assert (
"Capnproto schema doesn't contain field with name key. (THERE_IS_NO_COLUMN)"
in str(exc.value)
)