1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
#include "catch.hpp"
#include "test_helpers.hpp"
#include "duckdb/main/extension/extension_loader.hpp"
#include "duckdb/main/extension_manager.hpp"
#include "duckdb.hpp"
using namespace duckdb;
void SomeTableFunc(duckdb::ClientContext &, duckdb::TableFunctionInput &, duckdb::DataChunk &) {
}
TEST_CASE("Test catalog versioning", "[catalog]") {
DBConfig config;
config.SetOptionByName("allow_unsigned_extensions", true);
DuckDB db(nullptr, &config);
Connection con1(db);
con1.context->RunFunctionInTransaction([&]() {
auto &catalog = Catalog::GetCatalog(*con1.context, "");
REQUIRE(catalog.GetCatalogVersion(*con1.context) == 0);
});
REQUIRE_NO_FAIL(con1.Query("CREATE TABLE foo as SELECT 42"));
con1.context->RunFunctionInTransaction([&]() {
auto &catalog = Catalog::GetCatalog(*con1.context, "");
REQUIRE(catalog.GetCatalogVersion(*con1.context) == 1);
});
{
REQUIRE_NOTHROW(con1.BeginTransaction());
REQUIRE_NO_FAIL(con1.Query("CREATE TABLE foo2 as SELECT 42"));
auto &catalog = Catalog::GetCatalog(*con1.context, "");
auto catalog_version_1 = catalog.GetCatalogVersion(*con1.context);
REQUIRE(catalog_version_1.GetIndex() > TRANSACTION_ID_START);
REQUIRE_NO_FAIL(con1.Query("CREATE TABLE foo3 as SELECT 42"));
auto catalog_version_2 = catalog.GetCatalogVersion(*con1.context);
REQUIRE(catalog_version_2.GetIndex() > catalog_version_1.GetIndex());
REQUIRE_NOTHROW(con1.Commit());
}
con1.context->RunFunctionInTransaction([&]() {
auto &catalog = Catalog::GetCatalog(*con1.context, "");
REQUIRE(catalog.GetCatalogVersion(*con1.context) == 2);
});
// The following should not increment the version
REQUIRE_NO_FAIL(con1.Query("CREATE TABLE IF NOT EXISTS foo3 as SELECT 42"));
con1.context->RunFunctionInTransaction([&]() {
auto &catalog = Catalog::GetCatalog(*con1.context, "");
REQUIRE(catalog.GetCatalogVersion(*con1.context) == 2);
});
REQUIRE_NO_FAIL(con1.Query("CREATE SCHEMA IF NOT EXISTS my_schema"));
con1.context->RunFunctionInTransaction([&]() {
auto &catalog = Catalog::GetCatalog(*con1.context, "");
REQUIRE(catalog.GetCatalogVersion(*con1.context) == 3);
});
// The following should not increment the version
REQUIRE_NO_FAIL(con1.Query("CREATE SCHEMA IF NOT EXISTS my_schema"));
con1.context->RunFunctionInTransaction([&]() {
auto &catalog = Catalog::GetCatalog(*con1.context, "");
REQUIRE(catalog.GetCatalogVersion(*con1.context) == 3);
});
// check catalog version of system catalog
con1.context->RunFunctionInTransaction([&]() {
auto &catalog = Catalog::GetCatalog(*con1.context, "system");
REQUIRE(catalog.GetCatalogVersion(*con1.context) == 1);
});
REQUIRE_NO_FAIL(con1.Query("ATTACH ':memory:' as foo"));
con1.context->RunFunctionInTransaction([&]() {
auto &catalog = Catalog::GetCatalog(*con1.context, "system");
REQUIRE(catalog.GetCatalogVersion(*con1.context) == 2);
});
// check new version of attached DB starts at 0
con1.context->RunFunctionInTransaction([&]() {
auto &catalog = Catalog::GetCatalog(*con1.context, "foo");
REQUIRE(catalog.GetCatalogVersion(*con1.context) == 0);
});
// system transactions do not register catalog version changes :/
duckdb::TableFunction tf("some_new_table_function", {}, SomeTableFunc);
ExtensionInfo extension_info {};
ExtensionActiveLoad load_info {*db.instance, extension_info, "test_catalog_extension"};
ExtensionLoader loader {load_info};
loader.RegisterFunction(tf);
con1.context->RunFunctionInTransaction([&]() {
auto &catalog = Catalog::GetCatalog(*con1.context, "system");
REQUIRE(catalog.GetCatalogVersion(*con1.context) == 2);
});
}
|
