policy_manager/generic_variables_unittest.cc

// Copyright (c) 2014 The Chromium OS 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 "update_engine/policy_manager/generic_variables.h"

#include <base/memory/scoped_ptr.h>
#include <gtest/gtest.h>

#include "update_engine/policy_manager/pmtest_utils.h"

using base::TimeDelta;

namespace chromeos_policy_manager {

class PmCopyVariableTest : public ::testing::Test {
 protected:
  TimeDelta default_timeout_ = TimeDelta::FromSeconds(1);
};


TEST_F(PmCopyVariableTest, SimpleTest) {
  // Tests that copies are generated as intended.
  int source = 5;
  CopyVariable<int> var("var", kVariableModePoll, source);

  // Generate and validate a copy.
  scoped_ptr<const int> copy_1(var.GetValue(default_timeout_, NULL));
  PMTEST_ASSERT_NOT_NULL(copy_1.get());
  EXPECT_EQ(5, *copy_1);

  // Assign a different value to the source variable.
  source = 42;

  // Check that the content of the copy was not affected (distinct instance).
  EXPECT_EQ(5, *copy_1);

  // Generate and validate a second copy.
  scoped_ptr<const int> copy_2(var.GetValue(default_timeout_, NULL));
  PMTEST_ASSERT_NOT_NULL(copy_2.get());
  EXPECT_EQ(42, *copy_2);
}

TEST_F(PmCopyVariableTest, SetFlagTest) {
  // Tests that the set flag is being referred to as expected.
  int source = 5;
  bool is_set = false;
  CopyVariable<int> var("var", kVariableModePoll, source, &is_set);

  // Flag marked unset, nothing should be returned.
  PMTEST_ASSERT_NULL(var.GetValue(default_timeout_, NULL));

  // Flag marked set, we should be getting a value.
  is_set = true;
  scoped_ptr<const int> copy(var.GetValue(default_timeout_, NULL));
  PMTEST_ASSERT_NOT_NULL(copy.get());
  EXPECT_EQ(5, *copy);
}


class CopyConstructorTestClass {
 public:
  CopyConstructorTestClass(void) : copied_(false) {}
  CopyConstructorTestClass(const CopyConstructorTestClass& /* ignored */)
      : copied_(true) {}

  // Tells if the instance was constructed using the copy-constructor.
  bool copied_;
};


TEST_F(PmCopyVariableTest, UseCopyConstructorTest) {
  // Ensures that CopyVariables indeed uses the copy contructor.
  const CopyConstructorTestClass source;
  ASSERT_FALSE(source.copied_);

  CopyVariable<CopyConstructorTestClass> var("var", kVariableModePoll, source);
  scoped_ptr<const CopyConstructorTestClass> copy(
      var.GetValue(default_timeout_, NULL));
  PMTEST_ASSERT_NOT_NULL(copy.get());
  EXPECT_TRUE(copy->copied_);
}


class PmConstCopyVariableTest : public ::testing::Test {
 protected:
  TimeDelta default_timeout_ = TimeDelta::FromSeconds(1);
};

TEST_F(PmConstCopyVariableTest, SimpleTest) {
  int source = 5;
  ConstCopyVariable<int> var("var", source);
  scoped_ptr<const int> copy(var.GetValue(default_timeout_, NULL));
  PMTEST_ASSERT_NOT_NULL(copy.get());
  EXPECT_EQ(5, *copy);

  // Ensure the value is cached.
  source = 42;
  copy.reset(var.GetValue(default_timeout_, NULL));
  PMTEST_ASSERT_NOT_NULL(copy.get());
  EXPECT_EQ(5, *copy);
}

}  // namespace chromeos_policy_manager