google-cloud-cpp/google/cloud/internal/rest_context_test.cc at 2808537e1b8dd182d2637e6e66156f22dd91fddf · ajayky-os/google-cloud-cpp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
// Copyright 2022 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#include "google/cloud/internal/rest_context.h"
#include <gmock/gmock.h>

namespace google {
namespace cloud {
namespace rest_internal {
GOOGLE_CLOUD_CPP_INLINE_NAMESPACE_BEGIN
namespace {

using ::testing::Contains;
using ::testing::ElementsAre;
using ::testing::Eq;
using ::testing::Not;
using ::testing::Pair;

class RestContextTest : public ::testing::Test {
 protected:
  void SetUp() override {
    headers_["header1"] = HttpHeader{"header1", "value1"};
    headers_["header2"] = HttpHeader{"header2", {"value2a", "value2b"}};
  }

  HttpHeaders headers_;
};

TEST_F(RestContextTest, RvalueBuilder) {
  auto context = RestContext()
                     .AddHeader("header1", "value1")
                     .AddHeader(HttpHeader("header2", "value2a"))
                     .AddHeader("header2", "value2b");
  EXPECT_THAT(context.headers(),
              Contains(Pair("header1", ElementsAre("value1"))));
  EXPECT_THAT(context.headers(),
              Contains(Pair("header2", ElementsAre("value2a", "value2b"))));
}

TEST_F(RestContextTest, GetHeaderNotFound) {
  RestContext context(headers_);
  auto result = context.GetHeader("NotFound");
  EXPECT_TRUE(result.empty());
  result = context.GetHeader("notfound");
  EXPECT_TRUE(result.empty());
}

TEST_F(RestContextTest, GetHeaderFound) {
  RestContext context(headers_);
  auto result = context.GetHeader("Header1");
  EXPECT_THAT(result.size(), Eq(1));
  EXPECT_THAT(result, Contains("value1"));
  result = context.GetHeader("header1");
  EXPECT_THAT(result.size(), Eq(1));
  EXPECT_THAT(result, Contains("value1"));
}

TEST_F(RestContextTest, Equality) {
  RestContext lhs(headers_);
  EXPECT_THAT(lhs, Eq(lhs));
  RestContext rhs;
  EXPECT_THAT(lhs, Not(Eq(rhs)));
  rhs.AddHeader("header1", "value1");
  rhs.AddHeader({"header2", "value2a"});
  EXPECT_THAT(lhs, Not(Eq(rhs)));
  rhs.AddHeader("header2", "value2b");
  EXPECT_THAT(lhs, Eq(rhs));
}

TEST_F(RestContextTest, CancellationToken) {
  RestContext context;
  EXPECT_EQ(context.cancellation_token(), nullptr);

  auto token = std::make_shared<std::atomic<bool>>(false);
  context.set_cancellation_token(token);
  EXPECT_EQ(context.cancellation_token(), token);
}

}  // namespace
GOOGLE_CLOUD_CPP_INLINE_NAMESPACE_END
}  // namespace rest_internal
}  // namespace cloud
}  // namespace google