offline_event_logger_unittest.cc 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. // Copyright 2016 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "components/offline_pages/core/offline_event_logger.h"
  5. #include "testing/gtest/include/gtest/gtest.h"
  6. namespace offline_pages {
  7. namespace {
  8. const char kMessage[] = "Message is ";
  9. const int kTimeLength = 21;
  10. class EventLoggerTestClient : public OfflineEventLogger::Client {
  11. public:
  12. void CustomLog(const std::string& message) override {
  13. last_log_message_ = message;
  14. }
  15. const std::string& last_log_message() { return last_log_message_; }
  16. private:
  17. std::string last_log_message_;
  18. };
  19. } // namespace
  20. TEST(OfflineEventLoggerTest, SettingClientEnableLogging) {
  21. OfflineEventLogger logger;
  22. EventLoggerTestClient client;
  23. logger.SetClient(&client);
  24. EXPECT_TRUE(logger.GetIsLogging());
  25. }
  26. TEST(OfflineEventLoggerTest, SettingClientAndLog) {
  27. OfflineEventLogger logger;
  28. EventLoggerTestClient client;
  29. logger.SetClient(&client);
  30. logger.SetIsLogging(true);
  31. for (size_t i = 0; i < kMaxLogCount + 1; ++i)
  32. logger.RecordActivity(kMessage + std::to_string(i));
  33. std::vector<std::string> log;
  34. logger.GetLogs(&log);
  35. EXPECT_EQ(kMaxLogCount, log.size());
  36. EXPECT_EQ(client.last_log_message(), log[0].substr(kTimeLength));
  37. EXPECT_EQ(std::string(kMessage) + std::to_string(kMaxLogCount),
  38. client.last_log_message());
  39. }
  40. } // namespace offline_pages