| // Copyright 2018 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/system_data_updater_impl.h" |
| |
| #include <lib/fidl/cpp/binding_set.h> |
| #include <lib/inspect/cpp/inspect.h> |
| #include <lib/inspect/testing/cpp/inspect.h> |
| #include <lib/sys/cpp/component_context.h> |
| #include <lib/sys/cpp/testing/component_context_provider.h> |
| #include <lib/syslog/cpp/macros.h> |
| |
| #include "fuchsia/cobalt/cpp/fidl.h" |
| #include "src/lib/testing/loop_fixture/test_loop_fixture.h" |
| |
| namespace cobalt { |
| |
| using fuchsia::cobalt::SoftwareDistributionInfo; |
| using FuchsiaStatus = fuchsia::cobalt::Status; |
| using fuchsia::cobalt::SystemDataUpdaterPtr; |
| |
| class CobaltAppForTest { |
| public: |
| explicit CobaltAppForTest(std::unique_ptr<sys::ComponentContext> context) |
| : context_(std::move(context)) { |
| context_->outgoing()->AddPublicService( |
| system_data_updater_bindings_.GetHandler(&system_data_updater_impl_)); |
| } |
| |
| const inspect::Inspector& inspector() { return inspector_; } |
| |
| private: |
| inspect::Inspector inspector_; |
| |
| std::unique_ptr<sys::ComponentContext> context_; |
| |
| SystemDataUpdaterImpl system_data_updater_impl_; |
| fidl::BindingSet<fuchsia::cobalt::SystemDataUpdater> system_data_updater_bindings_; |
| }; |
| |
| class SystemDataUpdaterImplTests : public gtest::TestLoopFixture { |
| public: |
| void SetUp() override { |
| TestLoopFixture::SetUp(); |
| cobalt_app_.reset(new CobaltAppForTest(context_provider_.TakeContext())); |
| } |
| |
| void TearDown() override { |
| cobalt_app_.reset(); |
| TestLoopFixture::TearDown(); |
| } |
| |
| protected: |
| SystemDataUpdaterPtr GetSystemDataUpdater() { |
| SystemDataUpdaterPtr system_data_updater; |
| context_provider_.ConnectToPublicService(system_data_updater.NewRequest()); |
| return system_data_updater; |
| } |
| |
| private: |
| sys::testing::ComponentContextProvider context_provider_; |
| std::unique_ptr<CobaltAppForTest> cobalt_app_; |
| }; |
| |
| TEST_F(SystemDataUpdaterImplTests, ReturnsOk) { |
| SystemDataUpdaterPtr system_data_updater = GetSystemDataUpdater(); |
| SoftwareDistributionInfo info = SoftwareDistributionInfo(); |
| |
| bool called = false; |
| system_data_updater->SetSoftwareDistributionInfo(std::move(info), [&called](FuchsiaStatus s) { |
| called = true; |
| EXPECT_EQ(s, FuchsiaStatus::OK); |
| }); |
| |
| RunLoopUntilIdle(); |
| EXPECT_TRUE(called); |
| } |
| |
| } // namespace cobalt |