1// Copyright 2014 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
5#include "components/invalidation/invalidation_test_util.h"
6
7#include "base/basictypes.h"
8#include "base/json/json_writer.h"
9#include "base/json/string_escape.h"
10#include "base/values.h"
11#include "components/invalidation/invalidation.h"
12
13namespace syncer {
14
15using ::testing::MakeMatcher;
16using ::testing::MatchResultListener;
17using ::testing::Matcher;
18using ::testing::MatcherInterface;
19using ::testing::PrintToString;
20
21namespace {
22
23class AckHandleEqMatcher : public MatcherInterface<const AckHandle&> {
24 public:
25  explicit AckHandleEqMatcher(const AckHandle& expected);
26
27  virtual bool MatchAndExplain(const AckHandle& actual,
28                               MatchResultListener* listener) const;
29  virtual void DescribeTo(::std::ostream* os) const;
30  virtual void DescribeNegationTo(::std::ostream* os) const;
31
32 private:
33  const AckHandle expected_;
34
35  DISALLOW_COPY_AND_ASSIGN(AckHandleEqMatcher);
36};
37
38AckHandleEqMatcher::AckHandleEqMatcher(const AckHandle& expected)
39    : expected_(expected) {
40}
41
42bool AckHandleEqMatcher::MatchAndExplain(const AckHandle& actual,
43                                         MatchResultListener* listener) const {
44  return expected_.Equals(actual);
45}
46
47void AckHandleEqMatcher::DescribeTo(::std::ostream* os) const {
48  *os << " is equal to " << PrintToString(expected_);
49}
50
51void AckHandleEqMatcher::DescribeNegationTo(::std::ostream* os) const {
52  *os << " isn't equal to " << PrintToString(expected_);
53}
54
55class InvalidationEqMatcher : public MatcherInterface<const Invalidation&> {
56 public:
57  explicit InvalidationEqMatcher(const Invalidation& expected);
58
59  virtual bool MatchAndExplain(const Invalidation& actual,
60                               MatchResultListener* listener) const;
61  virtual void DescribeTo(::std::ostream* os) const;
62  virtual void DescribeNegationTo(::std::ostream* os) const;
63
64 private:
65  const Invalidation expected_;
66
67  DISALLOW_COPY_AND_ASSIGN(InvalidationEqMatcher);
68};
69
70InvalidationEqMatcher::InvalidationEqMatcher(const Invalidation& expected)
71    : expected_(expected) {
72}
73
74bool InvalidationEqMatcher::MatchAndExplain(
75    const Invalidation& actual,
76    MatchResultListener* listener) const {
77  if (!(expected_.object_id() == actual.object_id())) {
78    return false;
79  }
80  if (expected_.is_unknown_version() && actual.is_unknown_version()) {
81    return true;
82  } else if (expected_.is_unknown_version() != actual.is_unknown_version()) {
83    return false;
84  } else {
85    // Neither is unknown version.
86    return expected_.payload() == actual.payload() &&
87           expected_.version() == actual.version();
88  }
89}
90
91void InvalidationEqMatcher::DescribeTo(::std::ostream* os) const {
92  *os << " is equal to " << PrintToString(expected_);
93}
94
95void InvalidationEqMatcher::DescribeNegationTo(::std::ostream* os) const {
96  *os << " isn't equal to " << PrintToString(expected_);
97}
98
99}  // namespace
100
101void PrintTo(const AckHandle& ack_handle, ::std::ostream* os) {
102  scoped_ptr<base::Value> value(ack_handle.ToValue());
103  std::string printable_ack_handle;
104  base::JSONWriter::Write(value.get(), &printable_ack_handle);
105  *os << "{ ack_handle: " << printable_ack_handle << " }";
106}
107
108Matcher<const AckHandle&> Eq(const AckHandle& expected) {
109  return MakeMatcher(new AckHandleEqMatcher(expected));
110}
111
112void PrintTo(const Invalidation& inv, ::std::ostream* os) {
113  *os << "{ payload: " << inv.ToString() << " }";
114}
115
116Matcher<const Invalidation&> Eq(const Invalidation& expected) {
117  return MakeMatcher(new InvalidationEqMatcher(expected));
118}
119
120}  // namespace syncer
121