| // Copyright 2021 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/cobalt/bin/app/diagnostics_impl.h" |
| |
| #include <lib/inspect/cpp/inspect.h> |
| #include <lib/inspect/testing/cpp/inspect.h> |
| #include <lib/syslog/cpp/macros.h> |
| |
| #include <cstdio> |
| |
| #include <gtest/gtest.h> |
| |
| #include "src/lib/testing/loop_fixture/test_loop_fixture.h" |
| |
| namespace cobalt { |
| |
| using inspect::testing::ChildrenMatch; |
| using inspect::testing::IntIs; |
| using inspect::testing::NameMatches; |
| using inspect::testing::NodeMatches; |
| using inspect::testing::PropertyList; |
| using inspect::testing::StringIs; |
| using ::testing::IsSupersetOf; |
| using ::testing::UnorderedElementsAre; |
| |
| class DiagnosticsTest : public gtest::TestLoopFixture { |
| public: |
| void SetUp() override { |
| listener_ = std::make_unique<DiagnosticsImpl>(inspector_.GetRoot().CreateChild("core")); |
| } |
| |
| inspect::Hierarchy InspectHierarchy() { |
| fpromise::result<inspect::Hierarchy> result = inspect::ReadFromVmo(inspector_.DuplicateVmo()); |
| return result.take_value(); |
| } |
| |
| protected: |
| std::unique_ptr<DiagnosticsImpl> listener_; |
| inspect::Inspector inspector_; |
| }; |
| |
| TEST_F(DiagnosticsTest, SentObservationResultSuccess) { |
| listener_->SentObservationResult(Status::OK); |
| EXPECT_THAT(InspectHierarchy(), |
| AllOf(NodeMatches(NameMatches("root")), |
| ChildrenMatch(UnorderedElementsAre(AllOf( |
| NodeMatches(NameMatches("core")), |
| ChildrenMatch(Contains(AllOf(NodeMatches(AllOf( |
| NameMatches("sending_observations"), |
| PropertyList(IsSupersetOf({IntIs("successes", 1), IntIs("errors", 0), |
| IntIs("last_success_time", testing::Gt(0)), |
| IntIs("last_error_time", 0)})))))))))))); |
| } |
| |
| TEST_F(DiagnosticsTest, SentObservationResultError) { |
| listener_->SentObservationResult( |
| Status(StatusCode::DEADLINE_EXCEEDED, "error_message", "error_details")); |
| EXPECT_THAT( |
| InspectHierarchy(), |
| AllOf( |
| NodeMatches(NameMatches("root")), |
| ChildrenMatch(UnorderedElementsAre(AllOf( |
| NodeMatches(NameMatches("core")), |
| ChildrenMatch(Contains(AllOf(NodeMatches(AllOf( |
| NameMatches("sending_observations"), |
| PropertyList(UnorderedElementsAre( |
| IntIs("successes", 0), IntIs("errors", 1), IntIs("last_success_time", 0), |
| IntIs("last_error_time", testing::Gt(0)), |
| IntIs("last_error_code", static_cast<int64_t>(StatusCode::DEADLINE_EXCEEDED)), |
| StringIs("last_error_message", "error_message"), |
| StringIs("last_error_details", "error_details"))))))))))))); |
| } |
| |
| TEST_F(DiagnosticsTest, ObservationStoreUpdatedOnce) { |
| std::map<lib::ReportSpec, uint64_t> num_obs_per_report; |
| num_obs_per_report[{.customer_id = 1, .project_id = 2, .metric_id = 3, .report_id = 4}] = 5; |
| num_obs_per_report[{.customer_id = 6, .project_id = 7, .metric_id = 8, .report_id = 9}] = 10; |
| listener_->ObservationStoreUpdated(num_obs_per_report, 1024, 2048); |
| EXPECT_THAT(InspectHierarchy(), |
| AllOf(NodeMatches(NameMatches("root")), |
| ChildrenMatch(UnorderedElementsAre( |
| AllOf(NodeMatches(NameMatches("core")), |
| ChildrenMatch(Contains(AllOf(NodeMatches(AllOf( |
| NameMatches("observations_stored"), |
| PropertyList(UnorderedElementsAre( |
| IntIs("byte_count", 1024), IntIs("byte_count_limit", 2048), |
| IntIs("report_1-2-3-4", 5), IntIs("report_6-7-8-9", 10), |
| IntIs("total", 15))))))))))))); |
| } |
| |
| TEST_F(DiagnosticsTest, ObservationStoreUpdatedMultipleTimes) { |
| std::map<lib::ReportSpec, uint64_t> num_obs_per_report; |
| num_obs_per_report[{.customer_id = 1, .project_id = 2, .metric_id = 3, .report_id = 4}] = 5; |
| listener_->ObservationStoreUpdated(num_obs_per_report, 1024, 2048); |
| num_obs_per_report[{.customer_id = 1, .project_id = 2, .metric_id = 3, .report_id = 4}] = 12; |
| listener_->ObservationStoreUpdated(num_obs_per_report, 4096, 8192); |
| EXPECT_THAT(InspectHierarchy(), |
| AllOf(NodeMatches(NameMatches("root")), |
| ChildrenMatch(UnorderedElementsAre( |
| AllOf(NodeMatches(NameMatches("core")), |
| ChildrenMatch(Contains(AllOf(NodeMatches(AllOf( |
| NameMatches("observations_stored"), |
| PropertyList(UnorderedElementsAre( |
| IntIs("byte_count", 4096), IntIs("byte_count_limit", 8192), |
| IntIs("report_1-2-3-4", 12), IntIs("total", 12))))))))))))); |
| } |
| |
| TEST_F(DiagnosticsTest, LoggerCalled) { |
| listener_->LoggerCalled(1, "fuchsia/cobalt"); |
| EXPECT_THAT( |
| InspectHierarchy(), |
| AllOf(NodeMatches(NameMatches("root")), |
| ChildrenMatch(UnorderedElementsAre(AllOf( |
| NodeMatches(NameMatches("core")), |
| ChildrenMatch(Contains(AllOf( |
| NodeMatches(NameMatches("internal_metrics")), |
| ChildrenMatch(Contains(AllOf( |
| NodeMatches(AllOf(NameMatches("logger_calls"), |
| PropertyList(UnorderedElementsAre( |
| IntIs("total", 1), |
| IntIs("last_successful_time", testing::Gt(0)))))), |
| ChildrenMatch(UnorderedElementsAre( |
| AllOf(NodeMatches(NameMatches("per_project")), |
| ChildrenMatch(UnorderedElementsAre(NodeMatches(AllOf( |
| NameMatches("fuchsia/cobalt"), |
| PropertyList(UnorderedElementsAre( |
| IntIs("num_calls", 1), |
| IntIs("last_successful_time", testing::Gt(0))))))))), |
| AllOf(NodeMatches(NameMatches("per_method")), |
| ChildrenMatch(UnorderedElementsAre(NodeMatches( |
| AllOf(NameMatches("method_1"), |
| PropertyList(UnorderedElementsAre( |
| IntIs("num_calls", 1), |
| IntIs("last_successful_time", |
| testing::Gt(0)))))))))))))))))))))); |
| listener_->LoggerCalled(2, "fuchsia/memory"); |
| EXPECT_THAT( |
| InspectHierarchy(), |
| AllOf( |
| NodeMatches(NameMatches("root")), |
| ChildrenMatch(UnorderedElementsAre(AllOf( |
| NodeMatches(NameMatches("core")), |
| ChildrenMatch(Contains(AllOf( |
| NodeMatches(NameMatches("internal_metrics")), |
| ChildrenMatch(Contains(AllOf( |
| NodeMatches(AllOf( |
| NameMatches("logger_calls"), |
| PropertyList(UnorderedElementsAre( |
| IntIs("total", 2), IntIs("last_successful_time", testing::Gt(0)))))), |
| ChildrenMatch(UnorderedElementsAre( |
| AllOf(NodeMatches(NameMatches("per_project")), |
| ChildrenMatch(UnorderedElementsAre( |
| NodeMatches( |
| AllOf(NameMatches("fuchsia/cobalt"), |
| PropertyList(UnorderedElementsAre( |
| IntIs("num_calls", 1), |
| IntIs("last_successful_time", testing::Gt(0)))))), |
| NodeMatches(AllOf( |
| NameMatches("fuchsia/memory"), |
| PropertyList(UnorderedElementsAre( |
| IntIs("num_calls", 1), |
| IntIs("last_successful_time", testing::Gt(0))))))))), |
| AllOf(NodeMatches(NameMatches("per_method")), |
| ChildrenMatch(UnorderedElementsAre( |
| NodeMatches( |
| AllOf(NameMatches("method_1"), |
| PropertyList(UnorderedElementsAre( |
| IntIs("num_calls", 1), |
| IntIs("last_successful_time", testing::Gt(0)))))), |
| NodeMatches(AllOf(NameMatches("method_2"), |
| PropertyList(UnorderedElementsAre( |
| IntIs("num_calls", 1), |
| IntIs("last_successful_time", |
| testing::Gt(0)))))))))))))))))))))); |
| listener_->LoggerCalled(2, "fuchsia/cobalt"); |
| EXPECT_THAT( |
| InspectHierarchy(), |
| AllOf( |
| NodeMatches(NameMatches("root")), |
| ChildrenMatch(UnorderedElementsAre(AllOf( |
| NodeMatches(NameMatches("core")), |
| ChildrenMatch(Contains(AllOf( |
| NodeMatches(NameMatches("internal_metrics")), |
| ChildrenMatch(Contains(AllOf( |
| NodeMatches(AllOf( |
| NameMatches("logger_calls"), |
| PropertyList(UnorderedElementsAre( |
| IntIs("total", 3), IntIs("last_successful_time", testing::Gt(0)))))), |
| ChildrenMatch(UnorderedElementsAre( |
| AllOf(NodeMatches(NameMatches("per_project")), |
| ChildrenMatch(UnorderedElementsAre( |
| NodeMatches( |
| AllOf(NameMatches("fuchsia/cobalt"), |
| PropertyList(UnorderedElementsAre( |
| IntIs("num_calls", 2), |
| IntIs("last_successful_time", testing::Gt(0)))))), |
| NodeMatches(AllOf( |
| NameMatches("fuchsia/memory"), |
| PropertyList(UnorderedElementsAre( |
| IntIs("num_calls", 1), |
| IntIs("last_successful_time", testing::Gt(0))))))))), |
| AllOf(NodeMatches(NameMatches("per_method")), |
| ChildrenMatch(UnorderedElementsAre( |
| NodeMatches( |
| AllOf(NameMatches("method_1"), |
| PropertyList(UnorderedElementsAre( |
| IntIs("num_calls", 1), |
| IntIs("last_successful_time", testing::Gt(0)))))), |
| NodeMatches(AllOf(NameMatches("method_2"), |
| PropertyList(UnorderedElementsAre( |
| IntIs("num_calls", 2), |
| IntIs("last_successful_time", |
| testing::Gt(0)))))))))))))))))))))); |
| } |
| |
| TEST_F(DiagnosticsTest, TrackDiskUsage) { |
| listener_->TrackDiskUsage(1, 256, 1024); |
| EXPECT_THAT(InspectHierarchy(), |
| AllOf(NodeMatches(NameMatches("root")), |
| ChildrenMatch(UnorderedElementsAre(AllOf( |
| NodeMatches(NameMatches("core")), |
| ChildrenMatch(Contains(AllOf( |
| NodeMatches(NameMatches("internal_metrics")), |
| ChildrenMatch(Contains(AllOf( |
| NodeMatches(NameMatches("disk_usage")), |
| ChildrenMatch(UnorderedElementsAre(AllOf( |
| NodeMatches(NameMatches("per_storage_class")), |
| ChildrenMatch(UnorderedElementsAre(NodeMatches(AllOf( |
| NameMatches("storage_class_1"), |
| PropertyList(UnorderedElementsAre( |
| IntIs("current_bytes", 256), IntIs("max_bytes", 256), |
| IntIs("byte_limit", 1024))))))))))))))))))))); |
| listener_->TrackDiskUsage(2, 124, -1); |
| EXPECT_THAT( |
| InspectHierarchy(), |
| AllOf(NodeMatches(NameMatches("root")), |
| ChildrenMatch(UnorderedElementsAre(AllOf( |
| NodeMatches(NameMatches("core")), |
| ChildrenMatch(Contains(AllOf( |
| NodeMatches(NameMatches("internal_metrics")), |
| ChildrenMatch(Contains(AllOf( |
| NodeMatches(NameMatches("disk_usage")), |
| ChildrenMatch(UnorderedElementsAre(AllOf( |
| NodeMatches(NameMatches("per_storage_class")), |
| ChildrenMatch(UnorderedElementsAre( |
| NodeMatches(AllOf( |
| NameMatches("storage_class_1"), |
| PropertyList(UnorderedElementsAre(IntIs("current_bytes", 256), |
| IntIs("max_bytes", 256), |
| IntIs("byte_limit", 1024))))), |
| NodeMatches( |
| AllOf(NameMatches("storage_class_2"), |
| PropertyList(UnorderedElementsAre( |
| IntIs("current_bytes", 124), IntIs("max_bytes", 124), |
| IntIs("byte_limit", 0))))))))))))))))))))); |
| listener_->TrackDiskUsage(1, 100, 1024); |
| EXPECT_THAT( |
| InspectHierarchy(), |
| AllOf(NodeMatches(NameMatches("root")), |
| ChildrenMatch(UnorderedElementsAre(AllOf( |
| NodeMatches(NameMatches("core")), |
| ChildrenMatch(Contains(AllOf( |
| NodeMatches(NameMatches("internal_metrics")), |
| ChildrenMatch(Contains(AllOf( |
| NodeMatches(NameMatches("disk_usage")), |
| ChildrenMatch(UnorderedElementsAre(AllOf( |
| NodeMatches(NameMatches("per_storage_class")), |
| ChildrenMatch(UnorderedElementsAre( |
| NodeMatches(AllOf( |
| NameMatches("storage_class_1"), |
| PropertyList(UnorderedElementsAre(IntIs("current_bytes", 100), |
| IntIs("max_bytes", 256), |
| IntIs("byte_limit", 1024))))), |
| NodeMatches( |
| AllOf(NameMatches("storage_class_2"), |
| PropertyList(UnorderedElementsAre( |
| IntIs("current_bytes", 124), IntIs("max_bytes", 124), |
| IntIs("byte_limit", 0))))))))))))))))))))); |
| } |
| |
| } // namespace cobalt |
