tursom/memgraph
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Trigger restore after query modules load (#183)

Browse Source
This commit is contained in:
antonio2368
2021-06-30 10:01:51 +02:00
committed by GitHub
parent 15911b64dc
commit e016c74e4b
6 changed files with 47 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
tests/unit/query_trigger.cpp
View File

@@ -830,10 +830,15 @@ class TriggerStoreTest : public ::testing::Test {
std::optional<storage::Storage::Accessor> storage_accessor;
};
TEST_F(TriggerStoreTest, Load) {
TEST_F(TriggerStoreTest, Restore) {
std::optional<query::TriggerStore> store;
store.emplace(testing_directory, &ast_cache, &*dba, &antlr_lock);
const auto reset_store = [&] {
store.emplace(testing_directory);
store->RestoreTriggers(&ast_cache, &*dba, &antlr_lock);
};
reset_store();
const auto check_empty = [&] {
ASSERT_EQ(store->GetTriggerInfo().size(), 0);
@@ -879,7 +884,7 @@ TEST_F(TriggerStoreTest, Load) {
check_triggers();
// recreate trigger store, this should reload everything from the disk
store.emplace(testing_directory, &ast_cache, &*dba, &antlr_lock);
reset_store();
check_triggers();
ASSERT_NO_THROW(store->DropTrigger(trigger_name_after));
@@ -887,13 +892,13 @@ TEST_F(TriggerStoreTest, Load) {
check_empty();
store.emplace(testing_directory, &ast_cache, &*dba, &antlr_lock);
reset_store();
check_empty();
}
TEST_F(TriggerStoreTest, AddTrigger) {
query::TriggerStore store{testing_directory, &ast_cache, &*dba, &antlr_lock};
query::TriggerStore store{testing_directory};
// Invalid query in statements
ASSERT_THROW(store.AddTrigger("trigger", "RETUR 1", {}, query::TriggerEventType::VERTEX_CREATE,
@@ -926,7 +931,7 @@ TEST_F(TriggerStoreTest, AddTrigger) {
}
TEST_F(TriggerStoreTest, DropTrigger) {
query::TriggerStore store{testing_directory, &ast_cache, &*dba, &antlr_lock};
query::TriggerStore store{testing_directory};
ASSERT_THROW(store.DropTrigger("Unknown"), utils::BasicException);
@@ -940,7 +945,7 @@ TEST_F(TriggerStoreTest, DropTrigger) {
}
TEST_F(TriggerStoreTest, TriggerInfo) {
query::TriggerStore store{testing_directory, &ast_cache, &*dba, &antlr_lock};
query::TriggerStore store{testing_directory};
std::vector<query::TriggerStore::TriggerInfo> expected_info;
store.AddTrigger("trigger", "RETURN 1", {}, query::TriggerEventType::VERTEX_CREATE,
@@ -985,7 +990,7 @@ TEST_F(TriggerStoreTest, TriggerInfo) {
}
TEST_F(TriggerStoreTest, AnyTriggerAllKeywords) {
query::TriggerStore store{testing_directory, &ast_cache, &*dba, &antlr_lock};
query::TriggerStore store{testing_directory};
using namespace std::literals;