mirror of
https://github.com/loot/libloot.git
synced 2026-07-27 14:16:01 -07:00
Add Database::set_user_known_bash_tags()
This commit is contained in:
@@ -70,8 +70,15 @@ std::vector<std::string> Database::GetKnownBashTags() const {
|
||||
|
||||
std::vector<std::string> Database::GetUserKnownBashTags() const {
|
||||
try {
|
||||
return convert<std::string>(
|
||||
database_->user_known_bash_tags());
|
||||
return convert<std::string>(database_->user_known_bash_tags());
|
||||
} catch (const ::rust::Error& e) {
|
||||
std::rethrow_exception(mapError(e));
|
||||
}
|
||||
}
|
||||
|
||||
void Database::SetUserKnownBashTags(const std::vector<std::string>& bashTags) {
|
||||
try {
|
||||
return database_->set_user_known_bash_tags(convert(bashTags));
|
||||
} catch (const ::rust::Error& e) {
|
||||
std::rethrow_exception(mapError(e));
|
||||
}
|
||||
|
||||
@@ -32,6 +32,8 @@ public:
|
||||
|
||||
std::vector<std::string> GetUserKnownBashTags() const override;
|
||||
|
||||
void SetUserKnownBashTags(const std::vector<std::string>& bashTags) override;
|
||||
|
||||
std::vector<Message> GetGeneralMessages(
|
||||
bool evaluateConditions = false) const override;
|
||||
|
||||
|
||||
@@ -120,6 +120,15 @@ impl Database {
|
||||
.to_vec())
|
||||
}
|
||||
|
||||
pub fn set_user_known_bash_tags(&self, bash_tags: Vec<String>) -> Result<(), VerboseError> {
|
||||
self.0
|
||||
.write()
|
||||
.map_err(DatabaseLockPoisonError::from)?
|
||||
.set_user_known_bash_tags(bash_tags);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub fn general_messages(
|
||||
&self,
|
||||
evaluate_conditions: bool,
|
||||
|
||||
@@ -272,6 +272,8 @@ mod ffi {
|
||||
|
||||
pub fn user_known_bash_tags(&self) -> Result<Vec<String>>;
|
||||
|
||||
pub fn set_user_known_bash_tags(&self, bash_tags: Vec<String>) -> Result<()>;
|
||||
|
||||
pub fn general_messages(&self, evaluate_conditions: bool) -> Result<Vec<Message>>;
|
||||
|
||||
pub fn groups(&self, include_user_metadata: bool) -> Result<Vec<Group>>;
|
||||
|
||||
@@ -502,6 +502,23 @@ TEST_P(DatabaseInterfaceTest,
|
||||
EXPECT_EQ(expectedTags, tags);
|
||||
}
|
||||
|
||||
TEST_P(DatabaseInterfaceTest,
|
||||
setUserKnownBashTagsShouldReplaceAllKnownBashTagsInTheUserlist) {
|
||||
ASSERT_NO_THROW(GenerateMasterlist());
|
||||
ASSERT_NO_THROW(GenerateUserlist());
|
||||
|
||||
ASSERT_NO_THROW(handle_->GetDatabase().LoadMasterlist(masterlistPath));
|
||||
ASSERT_NO_THROW(handle_->GetDatabase().LoadUserlist(userlistPath_));
|
||||
|
||||
std::vector<std::string> tags({
|
||||
"Filter",
|
||||
});
|
||||
|
||||
handle_->GetDatabase().SetUserKnownBashTags({"Filter"});
|
||||
|
||||
EXPECT_EQ(tags, handle_->GetDatabase().GetUserKnownBashTags());
|
||||
}
|
||||
|
||||
TEST_P(DatabaseInterfaceTest,
|
||||
getGeneralMessagesShouldGetGeneralMessagesFromTheMasterlistAndUserlist) {
|
||||
ASSERT_NO_THROW(GenerateMasterlist());
|
||||
|
||||
Reference in New Issue
Block a user