Win 10 Inactive toast experiment metrics and storage modifications.

chrome/installer/util/experiment_unittest.cc

Index: chrome/installer/util/experiment_unittest.cc
diff --git a/chrome/installer/util/experiment_unittest.cc b/chrome/installer/util/experiment_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..e82c2b8ff2cb43f3dfd3df5b9e85212920d1d0ba
--- /dev/null
+++ b/chrome/installer/util/experiment_unittest.cc
@@ -0,0 +1,220 @@
+// Copyright 2017 The Chromium 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 "chrome/installer/util/experiment.h"
+
+#include <cmath>
+
+#include "base/time/time.h"
+#include "chrome/installer/util/experiment_metrics.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace installer {
+
+TEST(ExperimentTest, TestInitializeFromMetrics) {
+  Experiment experiment;
+  ExperimentMetrics metrics_with_group;
+  metrics_with_group.state = ExperimentMetrics::kGroupAssigned;
+  metrics_with_group.group = 5;
+  experiment.InitializeFromMetrics(metrics_with_group);
+  EXPECT_EQ(ExperimentMetrics::kGroupAssigned, experiment.state());
+  EXPECT_EQ(5, experiment.group());
+  experiment.SetInactiveDays(2);
+  // Test initializing from empty ExperimentMetrics.
+  experiment.InitializeFromMetrics(ExperimentMetrics());
+  EXPECT_EQ(ExperimentMetrics::kUninitialized, experiment.state());
+  EXPECT_EQ(0, experiment.group());
+  EXPECT_EQ(0, experiment.inactive_days());
+}
+
+TEST(ExperimentTest, TestSetState) {
+  Experiment experiment;
+  EXPECT_EQ(ExperimentMetrics::kUninitialized, experiment.state());
+  EXPECT_EQ(ExperimentMetrics::kUninitialized, experiment.metrics().state);
+  experiment.SetState(ExperimentMetrics::kGroupAssigned);
+  EXPECT_EQ(ExperimentMetrics::kGroupAssigned, experiment.metrics().state);
+  EXPECT_EQ(ExperimentMetrics::kGroupAssigned, experiment.state());
+}
+
+TEST(ExperimentTest, TestAssignGroup) {
+  Experiment experiment;
+  experiment.AssignGroup(5);
+  EXPECT_EQ(5, experiment.metrics().group);
+  EXPECT_EQ(5, experiment.group());
+  EXPECT_EQ(ExperimentMetrics::kGroupAssigned, experiment.metrics().state);
+  EXPECT_EQ(ExperimentMetrics::kGroupAssigned, experiment.state());
+}
+
+TEST(ExperimentTest, TestSetInactiveDays) {
+  Experiment experiment;
+  experiment.AssignGroup(5);
+
+  experiment.SetInactiveDays(0);
+  EXPECT_EQ(0, experiment.metrics().last_used_bucket);
+  EXPECT_EQ(0, experiment.inactive_days());
+
+  experiment.SetInactiveDays(28);
+  EXPECT_EQ(57, experiment.metrics().last_used_bucket);
+  EXPECT_EQ(28, experiment.inactive_days());
+
+  experiment.SetInactiveDays(35);
+  EXPECT_EQ(61, experiment.metrics().last_used_bucket);
+  EXPECT_EQ(35, experiment.inactive_days());
+
+  experiment.SetInactiveDays(60);
+  EXPECT_EQ(70, experiment.metrics().last_used_bucket);
+  EXPECT_EQ(60, experiment.inactive_days());
+
+  experiment.SetInactiveDays(ExperimentMetrics::kMaxLastUsed);
+  EXPECT_EQ(127, experiment.metrics().last_used_bucket);
+  EXPECT_EQ(ExperimentMetrics::kMaxLastUsed, experiment.inactive_days());
+
+  experiment.SetInactiveDays(2 * ExperimentMetrics::kMaxLastUsed);
+  EXPECT_EQ(127, experiment.metrics().last_used_bucket);
+  EXPECT_EQ(2 * ExperimentMetrics::kMaxLastUsed, experiment.inactive_days());
+}
+
+TEST(ExperimentTest, TestSetDisplayTime) {
+  Experiment experiment;
+  experiment.AssignGroup(5);
+  base::Time zero_day =
+      base::Time::FromDoubleT(ExperimentMetrics::kExperimentStartSeconds);
+  experiment.SetDisplayTime(zero_day);
+  EXPECT_EQ(0, experiment.metrics().first_toast_offset_days);
+  EXPECT_EQ(zero_day, experiment.first_display_time());
+  EXPECT_EQ(zero_day, experiment.latest_display_time());
+
+  base::Time one_day = zero_day + base::TimeDelta::FromDays(1);
+  experiment.SetDisplayTime(one_day);
+  EXPECT_EQ(1, experiment.metrics().first_toast_offset_days);
+  EXPECT_EQ(one_day, experiment.first_display_time());
+  EXPECT_EQ(one_day, experiment.latest_display_time());
+
+  // Test that calling SetDisplayTime again will not reset
+  // first_toast_offset_days.
+  base::Time two_day = zero_day + base::TimeDelta::FromDays(2);
+  experiment.SetDisplayTime(two_day);
+  EXPECT_EQ(1, experiment.metrics().first_toast_offset_days);
+  EXPECT_EQ(one_day, experiment.first_display_time());
+  EXPECT_EQ(two_day, experiment.latest_display_time());
+
+  // Test for maximum value.
+  Experiment new_experiment;
+  new_experiment.AssignGroup(5);
+  base::Time max_day =
+      zero_day +
+      base::TimeDelta::FromDays(ExperimentMetrics::kMaxFirstToastOffsetDays);
+  new_experiment.SetDisplayTime(max_day);
+  EXPECT_EQ(ExperimentMetrics::kMaxFirstToastOffsetDays,
+            new_experiment.metrics().first_toast_offset_days);
+  EXPECT_EQ(max_day, new_experiment.first_display_time());
+  EXPECT_EQ(max_day, new_experiment.latest_display_time());
+
+  // Test setting toast_hour. Since it depends on local time it is
+  // tested by setting toast hour for two consecutive hours and verifying the
+  // difference is 1.
+  base::Time two_day_one_hour = two_day + base::TimeDelta::FromHours(1);
+  Experiment hour_experiment;
+  hour_experiment.AssignGroup(5);
+  hour_experiment.SetDisplayTime(two_day_one_hour);
+  int diff_hour =
+      hour_experiment.metrics().toast_hour - experiment.metrics().toast_hour;
+  EXPECT_EQ(1, diff_hour);
+}
+
+TEST(ExperimentTest, TestSetUserSessionUptime) {
+  Experiment experiment;
+  experiment.AssignGroup(5);
+  experiment.SetUserSessionUptime(base::TimeDelta::FromMinutes(0));
+  EXPECT_EQ(0, experiment.metrics().session_length_bucket);
+  EXPECT_EQ(0, experiment.user_session_uptime().InMinutes());
+
+  experiment.SetUserSessionUptime(base::TimeDelta::FromMinutes(60));
+  EXPECT_EQ(24, experiment.metrics().session_length_bucket);
+  EXPECT_EQ(60, experiment.user_session_uptime().InMinutes());
+
+  experiment.SetUserSessionUptime(base::TimeDelta::FromMinutes(60 * 24));
+  EXPECT_EQ(43, experiment.metrics().session_length_bucket);
+  EXPECT_EQ(60 * 24, experiment.user_session_uptime().InMinutes());
+
+  experiment.SetUserSessionUptime(
+      base::TimeDelta::FromMinutes(ExperimentMetrics::kMaxSessionLength));
+  EXPECT_EQ(63, experiment.metrics().session_length_bucket);
+  EXPECT_EQ(ExperimentMetrics::kMaxSessionLength,
+            experiment.user_session_uptime().InMinutes());
+
+  experiment.SetUserSessionUptime(
+      base::TimeDelta::FromMinutes(2 * ExperimentMetrics::kMaxSessionLength));
+  EXPECT_EQ(63, experiment.metrics().session_length_bucket);
+  EXPECT_EQ(2 * ExperimentMetrics::kMaxSessionLength,
+            experiment.user_session_uptime().InMinutes());
+}
+
+TEST(ExperimentTest, TestSetActionDelay) {
+  Experiment experiment;
+  experiment.AssignGroup(5);
+  experiment.SetActionDelay(base::TimeDelta::FromSeconds(0));
+  EXPECT_EQ(0, experiment.metrics().action_delay_bucket);
+  EXPECT_EQ(0, experiment.user_session_uptime().InSeconds());
+
+  experiment.SetActionDelay(base::TimeDelta::FromSeconds(60));
+  EXPECT_EQ(10, experiment.metrics().action_delay_bucket);
+  EXPECT_EQ(60, experiment.action_delay().InSeconds());
+
+  experiment.SetActionDelay(base::TimeDelta::FromSeconds(60 * 60));
+  EXPECT_EQ(19, experiment.metrics().action_delay_bucket);
+  EXPECT_EQ(60 * 60, experiment.action_delay().InSeconds());
+
+  experiment.SetActionDelay(
+      base::TimeDelta::FromSeconds(ExperimentMetrics::kMaxActionDelay));
+  EXPECT_EQ(31, experiment.metrics().action_delay_bucket);
+  EXPECT_EQ(ExperimentMetrics::kMaxActionDelay,
+            experiment.action_delay().InSeconds());
+
+  experiment.SetActionDelay(
+      base::TimeDelta::FromSeconds(2 * ExperimentMetrics::kMaxActionDelay));
+  EXPECT_EQ(31, experiment.metrics().action_delay_bucket);
+  EXPECT_EQ(2 * ExperimentMetrics::kMaxActionDelay,
+            experiment.action_delay().InSeconds());
+}
+
+TEST(ExperimentTest, TestAllSetters) {
+  Experiment experiment;
+  experiment.AssignGroup(5);
+  experiment.SetState(ExperimentMetrics::kGroupAssigned);
+  experiment.SetToastLocation(ExperimentMetrics::kOverTaskbarPin);
+  experiment.SetInactiveDays(1621);
+  experiment.SetToastCount(1);
+  base::Time test_display_time =
+      (base::Time::UnixEpoch() +
+       base::TimeDelta::FromSeconds(
+           ExperimentMetrics::kExperimentStartSeconds) +
+       base::TimeDelta::FromDays(30) + base::TimeDelta::FromHours(3));
+  experiment.SetDisplayTime(test_display_time);
+  experiment.SetUserSessionUptime(base::TimeDelta::FromMinutes(3962));
+  experiment.SetActionDelay(base::TimeDelta::FromSeconds(32875));
+
+  EXPECT_EQ(ExperimentMetrics::kGroupAssigned, experiment.state());
+  EXPECT_EQ(5, experiment.group());
+  EXPECT_EQ(ExperimentMetrics::kOverTaskbarPin, experiment.toast_location());
+  EXPECT_EQ(1621, experiment.inactive_days());
+  EXPECT_EQ(1, experiment.toast_count());
+  EXPECT_EQ(test_display_time, experiment.first_display_time());
+  EXPECT_EQ(test_display_time, experiment.latest_display_time());
+  EXPECT_EQ(base::TimeDelta::FromMinutes(3962),
+            experiment.user_session_uptime());
+  EXPECT_EQ(base::TimeDelta::FromSeconds(32875), experiment.action_delay());
+
+  EXPECT_EQ(ExperimentMetrics::kGroupAssigned, experiment.metrics().state);
+  EXPECT_EQ(5, experiment.metrics().group);
+  EXPECT_EQ(ExperimentMetrics::kOverTaskbarPin,
+            experiment.metrics().toast_location);
+  EXPECT_EQ(1, experiment.metrics().toast_count);
+  EXPECT_EQ(30, experiment.metrics().first_toast_offset_days);
+  EXPECT_EQ(125, experiment.metrics().last_used_bucket);
+  EXPECT_EQ(24, experiment.metrics().action_delay_bucket);
+  EXPECT_EQ(49, experiment.metrics().session_length_bucket);
+}
+
+}  // namespace installer