1 // 2 // Copyright (C) 2014 The Android Open Source Project 3 // 4 // Licensed under the Apache License, Version 2.0 (the "License"); 5 // you may not use this file except in compliance with the License. 6 // You may obtain a copy of the License at 7 // 8 // http://www.apache.org/licenses/LICENSE-2.0 9 // 10 // Unless required by applicable law or agreed to in writing, software 11 // distributed under the License is distributed on an "AS IS" BASIS, 12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 // See the License for the specific language governing permissions and 14 // limitations under the License. 15 // 16 17 #include "update_engine/update_manager/real_time_provider.h" 18 19 #include <memory> 20 21 #include <base/logging.h> 22 #include <base/time/time.h> 23 #include <gtest/gtest.h> 24 25 #include "update_engine/common/fake_clock.h" 26 #include "update_engine/update_manager/umtest_utils.h" 27 28 using base::Time; 29 using chromeos_update_engine::FakeClock; 30 using std::unique_ptr; 31 32 namespace chromeos_update_manager { 33 34 class UmRealTimeProviderTest : public ::testing::Test { 35 protected: 36 void SetUp() override { 37 // The provider initializes correctly. 38 provider_.reset(new RealTimeProvider(&fake_clock_)); 39 ASSERT_NE(nullptr, provider_.get()); 40 ASSERT_TRUE(provider_->Init()); 41 } 42 43 // Generates a fixed timestamp for use in faking the current time. 44 Time CurrTime() { 45 Time::Exploded now_exp; 46 now_exp.year = 2014; 47 now_exp.month = 3; 48 now_exp.day_of_week = 2; 49 now_exp.day_of_month = 18; 50 now_exp.hour = 8; 51 now_exp.minute = 5; 52 now_exp.second = 33; 53 now_exp.millisecond = 675; 54 Time time; 55 ignore_result(Time::FromLocalExploded(now_exp, &time)); 56 return time; 57 } 58 59 FakeClock fake_clock_; 60 unique_ptr<RealTimeProvider> provider_; 61 }; 62 63 TEST_F(UmRealTimeProviderTest, CurrDateValid) { 64 const Time now = CurrTime(); 65 Time::Exploded exploded; 66 now.LocalExplode(&exploded); 67 exploded.hour = 0; 68 exploded.minute = 0; 69 exploded.second = 0; 70 exploded.millisecond = 0; 71 Time expected; 72 ignore_result(Time::FromLocalExploded(exploded, &expected)); 73 74 fake_clock_.SetWallclockTime(now); 75 UmTestUtils::ExpectVariableHasValue(expected, provider_->var_curr_date()); 76 } 77 78 TEST_F(UmRealTimeProviderTest, CurrHourValid) { 79 const Time now = CurrTime(); 80 Time::Exploded expected; 81 now.LocalExplode(&expected); 82 fake_clock_.SetWallclockTime(now); 83 UmTestUtils::ExpectVariableHasValue(expected.hour, 84 provider_->var_curr_hour()); 85 } 86 87 } // namespace chromeos_update_manager 88
