60 lines
2 KiB
C++
60 lines
2 KiB
C++
|
/*
|
||
|
|
* Copyright (c) 2026, Shannon Booth <shannon@serenityos.org>
|
||
|
|
*
|
||
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <LibTest/TestCase.h>
|
||
|
|
#include <Libraries/LibDatabase/Database.h>
|
||
|
|
|
||
|
|
TEST_CASE(string_can_contain_null_bytes)
|
||
|
|
{
|
||
|
|
auto database = TRY_OR_FAIL(Database::Database::create_memory_backed());
|
||
|
|
|
||
|
|
database->execute_statement(TRY_OR_FAIL(database->prepare_statement(R"#(
|
||
|
|
CREATE TABLE WebStorage (
|
||
|
|
key TEXT PRIMARY KEY,
|
||
|
|
value TEXT
|
||
|
|
);
|
||
|
|
)#"sv)),
|
||
|
|
{});
|
||
|
|
|
||
|
|
auto get_item_statement = TRY_OR_FAIL(database->prepare_statement("SELECT value FROM WebStorage WHERE key = ?;"sv));
|
||
|
|
auto set_item_statement = TRY_OR_FAIL(database->prepare_statement("INSERT OR REPLACE INTO WebStorage VALUES (?, ?);"sv));
|
||
|
|
auto delete_item_statement = TRY_OR_FAIL(database->prepare_statement("DELETE FROM WebStorage WHERE key = ?;"sv));
|
||
|
|
|
||
|
|
auto get_item = [&](String const& key) {
|
||
|
|
Optional<String> result;
|
||
|
|
database->execute_statement(
|
||
|
|
get_item_statement,
|
||
|
|
[&](auto statement_id) {
|
||
|
|
result = database->result_column<String>(statement_id, 0);
|
||
|
|
},
|
||
|
|
key);
|
||
|
|
return result;
|
||
|
|
};
|
||
|
|
|
||
|
|
auto set_item = [&](String const& key, String const& value) {
|
||
|
|
database->execute_statement(
|
||
|
|
set_item_statement,
|
||
|
|
{},
|
||
|
|
key,
|
||
|
|
value);
|
||
|
|
};
|
||
|
|
|
||
|
|
auto remove_item = [&](String const& key) {
|
||
|
|
database->execute_statement(
|
||
|
|
delete_item_statement,
|
||
|
|
{},
|
||
|
|
key);
|
||
|
|
};
|
||
|
|
|
||
|
|
EXPECT_EQ(get_item("my_key"_string), Optional<String> {});
|
||
|
|
set_item("my_key"_string, "my_value"_string);
|
||
|
|
EXPECT_EQ(get_item("my_key"_string), Optional<String> { "my_value"_string });
|
||
|
|
set_item("my_key"_string, "my_value_with_\0_null"_string);
|
||
|
|
EXPECT_EQ(get_item("my_key"_string), Optional<String> { "my_value_with_\0_null"_string });
|
||
|
|
remove_item("my_key"_string);
|
||
|
|
EXPECT_EQ(get_item("my_key"_string), Optional<String> {});
|
||
|
|
}
|