// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/user_education/common/user_education_storage_service.h"

#include <memory>

#include "base/feature_list.h"
#include "components/user_education/common/user_education_data.h"
#include "components/user_education/test/test_user_education_storage_service.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace user_education {

namespace {
BASE_FEATURE(kTestIPHFeature,
             base::FEATURE_ENABLED_BY_DEFAULT);
BASE_FEATURE(kTestIPHFeature2,
             base::FEATURE_ENABLED_BY_DEFAULT);
}  // namespace

TEST(UserEducationStorageServiceTest, GetSnoozeCount) {
  test::TestUserEducationStorageService service;
  EXPECT_EQ(0, service.GetSnoozeCount(kTestIPHFeature));
  EXPECT_EQ(0, service.GetSnoozeCount(kTestIPHFeature2));
  FeaturePromoData data;
  data.snooze_count = 3;
  service.SavePromoData(kTestIPHFeature, data);
  EXPECT_EQ(3, service.GetSnoozeCount(kTestIPHFeature));
  EXPECT_EQ(0, service.GetSnoozeCount(kTestIPHFeature2));
  service.Reset(kTestIPHFeature);
  EXPECT_EQ(0, service.GetSnoozeCount(kTestIPHFeature));
  EXPECT_EQ(0, service.GetSnoozeCount(kTestIPHFeature2));
}

TEST(UserEducationStorageServiceTest, GetShownForKeys) {
  static constexpr char kAppName1[] = "App1";
  static constexpr char kAppName2[] = "App2";
  KeyedFeaturePromoData keyed_data1;
  keyed_data1.show_count = 1;
  keyed_data1.last_shown_time = base::Time::Now() - base::Minutes(5);
  KeyedFeaturePromoData keyed_data2;
  keyed_data2.show_count = 2;
  keyed_data2.last_shown_time = base::Time::Now() - base::Minutes(4);

  test::TestUserEducationStorageService service;
  EXPECT_THAT(service.GetKeyedPromoData(kTestIPHFeature), testing::IsEmpty());
  EXPECT_THAT(service.GetKeyedPromoData(kTestIPHFeature2), testing::IsEmpty());
  FeaturePromoData data;
  data.shown_for_keys.emplace(kAppName1, keyed_data1);
  data.shown_for_keys.emplace(kAppName2, keyed_data2);
  service.SavePromoData(kTestIPHFeature, data);
  EXPECT_THAT(
      service.GetKeyedPromoData(kTestIPHFeature),
      testing::UnorderedElementsAre(std::make_pair(kAppName1, keyed_data1),
                                    std::make_pair(kAppName2, keyed_data2)));
  EXPECT_THAT(service.GetKeyedPromoData(kTestIPHFeature2), testing::IsEmpty());
  service.Reset(kTestIPHFeature);
  EXPECT_THAT(service.GetKeyedPromoData(kTestIPHFeature), testing::IsEmpty());
  EXPECT_THAT(service.GetKeyedPromoData(kTestIPHFeature2), testing::IsEmpty());
}

}  // namespace user_education