Add module for counting date-last-roll-call and persisting those counts

components/update_client/persisted_data_unittest.cc

+// Copyright 2016 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 <string>
+#include <vector>
+
+#include "base/files/file_path.h"
+#include "base/files/scoped_temp_dir.h"
+#include "base/run_loop.h"
+#include "base/thread_task_runner_handle.h"
+#include "components/update_client/persisted_data.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace update_client {
+
+class PersistedDataTest : public testing::Test {
+ public:
+  PersistedDataTest();
+  ~PersistedDataTest() override;
+
+ private:
+  base::MessageLoop used_for_threadtaskrunnerhandle_;
+
+  DISALLOW_COPY_AND_ASSIGN(PersistedDataTest);
+};
+
+PersistedDataTest::PersistedDataTest() {
+}
+
+PersistedDataTest::~PersistedDataTest() {
+}
+
+TEST_F(PersistedDataTest, InMemory) {
+  scoped_refptr<PersistedData> metadata(new PersistedData(
+      base::FilePath(), base::ThreadTaskRunnerHandle::Get()));
+  base::RunLoop().RunUntilIdle();  // To let Initialize() finish.
+  EXPECT_EQ(-2, metadata->DateLastRollCall("someappid"));
+  std::vector<std::string> items;
+  items.push_back("someappid");
+  metadata->SetDateLastRollCall(items, 3383);
+  EXPECT_EQ(3383, metadata->DateLastRollCall("someappid"));
+  EXPECT_EQ(-2, metadata->DateLastRollCall("someotherappid"));
+  metadata->SetDateLastRollCall(items, 3386);
+  EXPECT_EQ(3386, metadata->DateLastRollCall("someappid"));
+  EXPECT_EQ(-2, metadata->DateLastRollCall("someotherappid"));
+}
+
+TEST_F(PersistedDataTest, InFile) {
+  base::ScopedTempDir dir;
+  EXPECT_TRUE(dir.CreateUniqueTempDir());
+  base::FilePath path = dir.path().AppendASCII("t.json");
+  scoped_refptr<PersistedData> metadata(new PersistedData(
+      path, base::ThreadTaskRunnerHandle::Get()));
+  base::RunLoop().RunUntilIdle();  // To let Initialize() finish.
+  EXPECT_EQ(-2, metadata->DateLastRollCall("someappid"));
+  std::vector<std::string> items;
+  items.push_back("someappid");
+  metadata->SetDateLastRollCall(items, 3383);
+  base::RunLoop().RunUntilIdle();
+
+  // Now, create a new PersistedData reading from the same path, verify
+  // that it loads the value.
+  metadata = new PersistedData(path, base::ThreadTaskRunnerHandle::Get());
+  base::RunLoop().RunUntilIdle();  // To let Initialize() finish.
+  EXPECT_EQ(3383, metadata->DateLastRollCall("someappid"));
+  EXPECT_EQ(-2, metadata->DateLastRollCall("someotherappid"));
+}
+
+}  // namespace update_client