OLD | NEW |
1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/extensions/api/identity/identity_mint_queue.h" | 5 #include "chrome/browser/extensions/api/identity/identity_mint_queue.h" |
6 | 6 |
| 7 #include <memory> |
7 #include <vector> | 8 #include <vector> |
8 | 9 |
9 #include "base/memory/scoped_ptr.h" | |
10 #include "testing/gmock/include/gmock/gmock.h" | 10 #include "testing/gmock/include/gmock/gmock.h" |
11 #include "testing/gtest/include/gtest/gtest.h" | 11 #include "testing/gtest/include/gtest/gtest.h" |
12 | 12 |
13 using extensions::ExtensionTokenKey; | 13 using extensions::ExtensionTokenKey; |
14 using extensions::IdentityMintRequestQueue; | 14 using extensions::IdentityMintRequestQueue; |
15 | 15 |
16 namespace { | 16 namespace { |
17 | 17 |
18 class MockRequest : public extensions::IdentityMintRequestQueue::Request { | 18 class MockRequest : public extensions::IdentityMintRequestQueue::Request { |
19 public: | 19 public: |
20 MOCK_METHOD1(StartMintToken, void(IdentityMintRequestQueue::MintType)); | 20 MOCK_METHOD1(StartMintToken, void(IdentityMintRequestQueue::MintType)); |
21 }; | 21 }; |
22 | 22 |
23 scoped_ptr<ExtensionTokenKey> ExtensionIdToKey( | 23 std::unique_ptr<ExtensionTokenKey> ExtensionIdToKey( |
24 const std::string& extension_id) { | 24 const std::string& extension_id) { |
25 return scoped_ptr<ExtensionTokenKey>(new ExtensionTokenKey( | 25 return std::unique_ptr<ExtensionTokenKey>( |
26 extension_id, "user_id", std::set<std::string>())); | 26 new ExtensionTokenKey(extension_id, "user_id", std::set<std::string>())); |
27 } | 27 } |
28 | 28 |
29 } // namespace | 29 } // namespace |
30 | 30 |
31 TEST(IdentityMintQueueTest, SerialRequests) { | 31 TEST(IdentityMintQueueTest, SerialRequests) { |
32 IdentityMintRequestQueue::MintType type = | 32 IdentityMintRequestQueue::MintType type = |
33 IdentityMintRequestQueue::MINT_TYPE_NONINTERACTIVE; | 33 IdentityMintRequestQueue::MINT_TYPE_NONINTERACTIVE; |
34 IdentityMintRequestQueue queue; | 34 IdentityMintRequestQueue queue; |
35 scoped_ptr<ExtensionTokenKey> key(ExtensionIdToKey("ext_id")); | 35 std::unique_ptr<ExtensionTokenKey> key(ExtensionIdToKey("ext_id")); |
36 MockRequest request1; | 36 MockRequest request1; |
37 MockRequest request2; | 37 MockRequest request2; |
38 | 38 |
39 EXPECT_CALL(request1, StartMintToken(type)).Times(1); | 39 EXPECT_CALL(request1, StartMintToken(type)).Times(1); |
40 queue.RequestStart(type, *key, &request1); | 40 queue.RequestStart(type, *key, &request1); |
41 queue.RequestComplete(type, *key, &request1); | 41 queue.RequestComplete(type, *key, &request1); |
42 | 42 |
43 EXPECT_CALL(request2, StartMintToken(type)).Times(1); | 43 EXPECT_CALL(request2, StartMintToken(type)).Times(1); |
44 queue.RequestStart(type, *key, &request2); | 44 queue.RequestStart(type, *key, &request2); |
45 queue.RequestComplete(type, *key, &request2); | 45 queue.RequestComplete(type, *key, &request2); |
46 } | 46 } |
47 | 47 |
48 TEST(IdentityMintQueueTest, InteractiveType) { | 48 TEST(IdentityMintQueueTest, InteractiveType) { |
49 IdentityMintRequestQueue::MintType type = | 49 IdentityMintRequestQueue::MintType type = |
50 IdentityMintRequestQueue::MINT_TYPE_INTERACTIVE; | 50 IdentityMintRequestQueue::MINT_TYPE_INTERACTIVE; |
51 IdentityMintRequestQueue queue; | 51 IdentityMintRequestQueue queue; |
52 scoped_ptr<ExtensionTokenKey> key(ExtensionIdToKey("ext_id")); | 52 std::unique_ptr<ExtensionTokenKey> key(ExtensionIdToKey("ext_id")); |
53 MockRequest request1; | 53 MockRequest request1; |
54 | 54 |
55 EXPECT_CALL(request1, StartMintToken(type)).Times(1); | 55 EXPECT_CALL(request1, StartMintToken(type)).Times(1); |
56 queue.RequestStart(type, *key, &request1); | 56 queue.RequestStart(type, *key, &request1); |
57 queue.RequestComplete(type, *key, &request1); | 57 queue.RequestComplete(type, *key, &request1); |
58 } | 58 } |
59 | 59 |
60 TEST(IdentityMintQueueTest, ParallelRequests) { | 60 TEST(IdentityMintQueueTest, ParallelRequests) { |
61 IdentityMintRequestQueue::MintType type = | 61 IdentityMintRequestQueue::MintType type = |
62 IdentityMintRequestQueue::MINT_TYPE_NONINTERACTIVE; | 62 IdentityMintRequestQueue::MINT_TYPE_NONINTERACTIVE; |
63 IdentityMintRequestQueue queue; | 63 IdentityMintRequestQueue queue; |
64 scoped_ptr<ExtensionTokenKey> key(ExtensionIdToKey("ext_id")); | 64 std::unique_ptr<ExtensionTokenKey> key(ExtensionIdToKey("ext_id")); |
65 MockRequest request1; | 65 MockRequest request1; |
66 MockRequest request2; | 66 MockRequest request2; |
67 MockRequest request3; | 67 MockRequest request3; |
68 | 68 |
69 EXPECT_CALL(request1, StartMintToken(type)).Times(1); | 69 EXPECT_CALL(request1, StartMintToken(type)).Times(1); |
70 queue.RequestStart(type, *key, &request1); | 70 queue.RequestStart(type, *key, &request1); |
71 queue.RequestStart(type, *key, &request2); | 71 queue.RequestStart(type, *key, &request2); |
72 queue.RequestStart(type, *key, &request3); | 72 queue.RequestStart(type, *key, &request3); |
73 | 73 |
74 EXPECT_CALL(request2, StartMintToken(type)).Times(1); | 74 EXPECT_CALL(request2, StartMintToken(type)).Times(1); |
75 queue.RequestComplete(type, *key, &request1); | 75 queue.RequestComplete(type, *key, &request1); |
76 | 76 |
77 EXPECT_CALL(request3, StartMintToken(type)).Times(1); | 77 EXPECT_CALL(request3, StartMintToken(type)).Times(1); |
78 queue.RequestComplete(type, *key, &request2); | 78 queue.RequestComplete(type, *key, &request2); |
79 | 79 |
80 queue.RequestComplete(type, *key, &request3); | 80 queue.RequestComplete(type, *key, &request3); |
81 } | 81 } |
82 | 82 |
83 TEST(IdentityMintQueueTest, ParallelRequestsFromTwoKeys) { | 83 TEST(IdentityMintQueueTest, ParallelRequestsFromTwoKeys) { |
84 IdentityMintRequestQueue::MintType type = | 84 IdentityMintRequestQueue::MintType type = |
85 IdentityMintRequestQueue::MINT_TYPE_NONINTERACTIVE; | 85 IdentityMintRequestQueue::MINT_TYPE_NONINTERACTIVE; |
86 IdentityMintRequestQueue queue; | 86 IdentityMintRequestQueue queue; |
87 scoped_ptr<ExtensionTokenKey> key1(ExtensionIdToKey("ext_id_1")); | 87 std::unique_ptr<ExtensionTokenKey> key1(ExtensionIdToKey("ext_id_1")); |
88 scoped_ptr<ExtensionTokenKey> key2(ExtensionIdToKey("ext_id_2")); | 88 std::unique_ptr<ExtensionTokenKey> key2(ExtensionIdToKey("ext_id_2")); |
89 MockRequest request1; | 89 MockRequest request1; |
90 MockRequest request2; | 90 MockRequest request2; |
91 | 91 |
92 EXPECT_CALL(request1, StartMintToken(type)).Times(1); | 92 EXPECT_CALL(request1, StartMintToken(type)).Times(1); |
93 EXPECT_CALL(request2, StartMintToken(type)).Times(1); | 93 EXPECT_CALL(request2, StartMintToken(type)).Times(1); |
94 queue.RequestStart(type, *key1, &request1); | 94 queue.RequestStart(type, *key1, &request1); |
95 queue.RequestStart(type, *key2, &request2); | 95 queue.RequestStart(type, *key2, &request2); |
96 | 96 |
97 queue.RequestComplete(type, *key1, &request1); | 97 queue.RequestComplete(type, *key1, &request1); |
98 queue.RequestComplete(type, *key2, &request2); | 98 queue.RequestComplete(type, *key2, &request2); |
99 } | 99 } |
100 | 100 |
101 TEST(IdentityMintQueueTest, Empty) { | 101 TEST(IdentityMintQueueTest, Empty) { |
102 IdentityMintRequestQueue::MintType type = | 102 IdentityMintRequestQueue::MintType type = |
103 IdentityMintRequestQueue::MINT_TYPE_INTERACTIVE; | 103 IdentityMintRequestQueue::MINT_TYPE_INTERACTIVE; |
104 IdentityMintRequestQueue queue; | 104 IdentityMintRequestQueue queue; |
105 scoped_ptr<ExtensionTokenKey> key(ExtensionIdToKey("ext_id")); | 105 std::unique_ptr<ExtensionTokenKey> key(ExtensionIdToKey("ext_id")); |
106 MockRequest request1; | 106 MockRequest request1; |
107 | 107 |
108 EXPECT_TRUE(queue.empty(type, *key)); | 108 EXPECT_TRUE(queue.empty(type, *key)); |
109 EXPECT_CALL(request1, StartMintToken(type)).Times(1); | 109 EXPECT_CALL(request1, StartMintToken(type)).Times(1); |
110 queue.RequestStart(type, *key, &request1); | 110 queue.RequestStart(type, *key, &request1); |
111 EXPECT_FALSE(queue.empty(type, *key)); | 111 EXPECT_FALSE(queue.empty(type, *key)); |
112 queue.RequestComplete(type, *key, &request1); | 112 queue.RequestComplete(type, *key, &request1); |
113 EXPECT_TRUE(queue.empty(type, *key)); | 113 EXPECT_TRUE(queue.empty(type, *key)); |
114 } | 114 } |
115 | 115 |
116 TEST(IdentityMintQueueTest, Cancel) { | 116 TEST(IdentityMintQueueTest, Cancel) { |
117 IdentityMintRequestQueue::MintType type = | 117 IdentityMintRequestQueue::MintType type = |
118 IdentityMintRequestQueue::MINT_TYPE_NONINTERACTIVE; | 118 IdentityMintRequestQueue::MINT_TYPE_NONINTERACTIVE; |
119 IdentityMintRequestQueue queue; | 119 IdentityMintRequestQueue queue; |
120 scoped_ptr<ExtensionTokenKey> key(ExtensionIdToKey("ext_id")); | 120 std::unique_ptr<ExtensionTokenKey> key(ExtensionIdToKey("ext_id")); |
121 MockRequest request1; | 121 MockRequest request1; |
122 MockRequest request2; | 122 MockRequest request2; |
123 MockRequest request3; | 123 MockRequest request3; |
124 MockRequest request4; | 124 MockRequest request4; |
125 | 125 |
126 EXPECT_CALL(request1, StartMintToken(type)).Times(1); | 126 EXPECT_CALL(request1, StartMintToken(type)).Times(1); |
127 queue.RequestStart(type, *key, &request1); | 127 queue.RequestStart(type, *key, &request1); |
128 queue.RequestStart(type, *key, &request2); | 128 queue.RequestStart(type, *key, &request2); |
129 queue.RequestStart(type, *key, &request3); | 129 queue.RequestStart(type, *key, &request3); |
130 | 130 |
131 // request1: cancel the running head | 131 // request1: cancel the running head |
132 // request3: cancel a request that is not the head | 132 // request3: cancel a request that is not the head |
133 // request2: cancel the last request | 133 // request2: cancel the last request |
134 // request4: cancel a request that is not in the queue at all | 134 // request4: cancel a request that is not in the queue at all |
135 queue.RequestCancel(*key, &request1); | 135 queue.RequestCancel(*key, &request1); |
136 queue.RequestCancel(*key, &request3); | 136 queue.RequestCancel(*key, &request3); |
137 queue.RequestCancel(*key, &request2); | 137 queue.RequestCancel(*key, &request2); |
138 queue.RequestCancel(*key, &request4); | 138 queue.RequestCancel(*key, &request4); |
139 } | 139 } |
OLD | NEW |