blob: 3010d9430aacd8794b5bcaaedd2267d678e41ec5 [file] [log] [blame]
// Copyright 2020 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "src/storage/volume_image/address_descriptor.h"
#include <string>
#include <string_view>
#include <vector>
#include <gmock/gmock.h>
#include <gtest/gtest.h>
#include "rapidjson/document.h"
#include "rapidjson/schema.h"
#include "rapidjson/writer.h"
#include "src/lib/json_parser/json_parser.h"
#include "src/lib/json_parser/rapidjson_validation.h"
#include "src/storage/volume_image/serialization/schema.h"
namespace storage::volume_image {
namespace {
std::string GetSerializedJson(fit::function<void(rapidjson::Document*)> mutator = nullptr) {
constexpr std::string_view kSerializedAddressDescriptor = R"(
{
"magic": 12526821592682033285,
"mappings": [
{
"source": 20,
"target": 120,
"count": 10,
"options": {}
},
{
"source": 250,
"target": 160,
"count": 10
},
{
"source": 2900,
"target": 170,
"count": 10,
"size": 20
}
]
})";
json_parser::JSONParser parser;
rapidjson::Document parsed_document = parser.ParseFromString(
kSerializedAddressDescriptor.data(), "serialized_address_descriptor.json");
ZX_ASSERT_MSG(!parser.HasError(), "%s\n", parser.error_str().c_str());
if (mutator != nullptr) {
mutator(&parsed_document);
}
rapidjson::StringBuffer buffer;
rapidjson::Writer<rapidjson::StringBuffer> writer(buffer);
parsed_document.Accept(writer);
return buffer.GetString();
}
TEST(AddressDescriptorTest, SerializeReturnsSchemaValidData) {
AddressDescriptor descriptor = {};
descriptor.mappings = std::vector<AddressMap>({
{.source = 10,
.target = 20,
.count = 10,
.options = {{"random_option_1", 32}, {"random_option_2", 33}}},
{.source = 20, .target = 30, .count = 10},
});
auto schema_json = GetSchema(Schema::kAddressDescriptor);
ASSERT_FALSE(schema_json.empty());
json_parser::JSONParser parser;
auto result = descriptor.Serialize();
ASSERT_TRUE(result.is_ok()) << result.error();
auto value = result.take_value();
auto document =
parser.ParseFromString(std::string(value.begin(), value.end()), "serialized.json");
ASSERT_FALSE(parser.HasError()) << parser.error_str();
auto schema_result = json_parser::InitSchema(schema_json);
ASSERT_TRUE(schema_result.is_ok()) << schema_result.error_value().ToString();
auto validation_result = json_parser::ValidateSchema(document, schema_result.value());
EXPECT_TRUE(validation_result.is_ok()) << validation_result.error_value();
}
MATCHER(AddressMapEq, "") {
auto [a, b] = arg;
return a.source == b.source && a.target == b.target && a.count == b.count && a.size == b.size;
}
TEST(AddressDescriptorTest, DeserializeSerializedDataIsOk) {
const auto deserialized_result = AddressDescriptor::Deserialize(GetSerializedJson());
ASSERT_TRUE(deserialized_result.is_ok());
const auto serialized_result = deserialized_result.value().Serialize();
ASSERT_TRUE(serialized_result.is_ok());
const auto redeserialized_result = AddressDescriptor::Deserialize(serialized_result.value());
ASSERT_TRUE(redeserialized_result.is_ok());
const auto& deserialized_1 = deserialized_result.value();
const auto& deserialized_2 = deserialized_result.value();
ASSERT_THAT(deserialized_1.mappings,
testing::UnorderedPointwise(AddressMapEq(), deserialized_2.mappings));
}
TEST(AddressDescriptorTest, DeserializeFromValidDataReturnsAddressDescritptor) {
const auto deserialized_result = AddressDescriptor::Deserialize(GetSerializedJson());
ASSERT_TRUE(deserialized_result.is_ok()) << deserialized_result.error();
ASSERT_THAT(
deserialized_result.value().mappings,
testing::UnorderedPointwise(
AddressMapEq(), std::vector<AddressMap>({
{.source = 250, .target = 160, .count = 10, .size = std::nullopt},
{.source = 20, .target = 120, .count = 10, .size = std::nullopt},
{.source = 2900, .target = 170, .count = 10, .size = 20},
})));
}
TEST(AddressMapTest, DebugStringIsOk) {
AddressMap map = {
.source = 100,
.target = 200,
.count = 50,
};
EXPECT_THAT(map.DebugString(),
testing::AllOf(testing::ContainsRegex("source:[ ]+100"),
testing::ContainsRegex("target:[ ]+200"),
testing::ContainsRegex("count:[ ]+50"),
testing::ContainsRegex("size:[ ]+std::nullopt"),
testing::ContainsRegex("options:[ ]+\\{[ \n]*\\}")));
map.size = 150;
EXPECT_THAT(
map.DebugString(),
testing::AllOf(testing::ContainsRegex("source:[ ]+100"),
testing::ContainsRegex("target:[ ]+200"),
testing::ContainsRegex("count:[ ]+50"), testing::ContainsRegex("size:[ ]+150"),
testing::ContainsRegex("options:[ ]+\\{[ \n]*\\}")));
map.options["option_name"] = 1234;
map.options["option_name_2"] = 12345;
EXPECT_THAT(
map.DebugString(),
testing::AllOf(
testing::ContainsRegex("source:[ ]+100"), testing::ContainsRegex("target:[ ]+200"),
testing::ContainsRegex("count:[ ]+50"), testing::ContainsRegex("size:[ ]+150"),
testing::ContainsRegex(
"options:[ ]+\\{\n[ ]+option_name:[ ]+1234[ \n]+option_name_2:[ ]+12345[ \n]+\\}")));
}
TEST(AddressDescriptorTest, DeserializeWithBadMagicIsError) {
ASSERT_TRUE(AddressDescriptor::Deserialize(GetSerializedJson([](auto* document) {
(*document)["magic"] = AddressDescriptor::kMagic - 1;
})).is_error());
}
TEST(AddressDescriptorTest, DeserializeWithEmptyMappingsIsError) {
ASSERT_TRUE(AddressDescriptor::Deserialize(GetSerializedJson([](auto* document) {
rapidjson::Value value;
value.SetArray();
(*document)["mappings"] = value;
})).is_error());
}
} // namespace
} // namespace storage::volume_image