@@ -13,10 +13,10 @@ import (
13
13
)
14
14
15
15
func TestListArchivePolicies (t * testing.T ) {
16
- th .SetupHTTP ()
17
- defer th . TeardownHTTP ()
16
+ fakeServer := th .SetupHTTP ()
17
+ defer fakeServer . Teardown ()
18
18
19
- th .Mux .HandleFunc ("/v1/archive_policy" , func (w http.ResponseWriter , r * http.Request ) {
19
+ fakeServer .Mux .HandleFunc ("/v1/archive_policy" , func (w http.ResponseWriter , r * http.Request ) {
20
20
th .TestMethod (t , r , "GET" )
21
21
th .TestHeader (t , r , "X-Auth-Token" , fake .TokenID )
22
22
@@ -28,7 +28,7 @@ func TestListArchivePolicies(t *testing.T) {
28
28
29
29
expected := ListArchivePoliciesExpected
30
30
pages := 0
31
- err := archivepolicies .List (fake .ServiceClient ()).EachPage (context .TODO (), func (_ context.Context , page pagination.Page ) (bool , error ) {
31
+ err := archivepolicies .List (fake .ServiceClient (fakeServer )).EachPage (context .TODO (), func (_ context.Context , page pagination.Page ) (bool , error ) {
32
32
pages ++
33
33
34
34
actual , err := archivepolicies .ExtractArchivePolicies (page )
@@ -46,10 +46,10 @@ func TestListArchivePolicies(t *testing.T) {
46
46
}
47
47
48
48
func TestListArchivePoliciesAllPages (t * testing.T ) {
49
- th .SetupHTTP ()
50
- defer th . TeardownHTTP ()
49
+ fakeServer := th .SetupHTTP ()
50
+ defer fakeServer . Teardown ()
51
51
52
- th .Mux .HandleFunc ("/v1/archive_policy" , func (w http.ResponseWriter , r * http.Request ) {
52
+ fakeServer .Mux .HandleFunc ("/v1/archive_policy" , func (w http.ResponseWriter , r * http.Request ) {
53
53
th .TestMethod (t , r , "GET" )
54
54
th .TestHeader (t , r , "X-Auth-Token" , fake .TokenID )
55
55
@@ -59,17 +59,17 @@ func TestListArchivePoliciesAllPages(t *testing.T) {
59
59
fmt .Fprintf (w , ArchivePoliciesListResult )
60
60
})
61
61
62
- allPages , err := archivepolicies .List (fake .ServiceClient ()).AllPages (context .TODO ())
62
+ allPages , err := archivepolicies .List (fake .ServiceClient (fakeServer )).AllPages (context .TODO ())
63
63
th .AssertNoErr (t , err )
64
64
_ , err = archivepolicies .ExtractArchivePolicies (allPages )
65
65
th .AssertNoErr (t , err )
66
66
}
67
67
68
68
func TestGetArchivePolicy (t * testing.T ) {
69
- th .SetupHTTP ()
70
- defer th . TeardownHTTP ()
69
+ fakeServer := th .SetupHTTP ()
70
+ defer fakeServer . Teardown ()
71
71
72
- th .Mux .HandleFunc ("/v1/archive_policy/test_policy" , func (w http.ResponseWriter , r * http.Request ) {
72
+ fakeServer .Mux .HandleFunc ("/v1/archive_policy/test_policy" , func (w http.ResponseWriter , r * http.Request ) {
73
73
th .TestMethod (t , r , "GET" )
74
74
th .TestHeader (t , r , "X-Auth-Token" , fake .TokenID )
75
75
@@ -79,7 +79,7 @@ func TestGetArchivePolicy(t *testing.T) {
79
79
fmt .Fprintf (w , ArchivePolicyGetResult )
80
80
})
81
81
82
- s , err := archivepolicies .Get (context .TODO (), fake .ServiceClient (), "test_policy" ).Extract ()
82
+ s , err := archivepolicies .Get (context .TODO (), fake .ServiceClient (fakeServer ), "test_policy" ).Extract ()
83
83
th .AssertNoErr (t , err )
84
84
85
85
th .AssertDeepEquals (t , s .AggregationMethods , []string {
@@ -104,10 +104,10 @@ func TestGetArchivePolicy(t *testing.T) {
104
104
}
105
105
106
106
func TestCreate (t * testing.T ) {
107
- th .SetupHTTP ()
108
- defer th . TeardownHTTP ()
107
+ fakeServer := th .SetupHTTP ()
108
+ defer fakeServer . Teardown ()
109
109
110
- th .Mux .HandleFunc ("/v1/archive_policy" , func (w http.ResponseWriter , r * http.Request ) {
110
+ fakeServer .Mux .HandleFunc ("/v1/archive_policy" , func (w http.ResponseWriter , r * http.Request ) {
111
111
th .TestMethod (t , r , "POST" )
112
112
th .TestHeader (t , r , "X-Auth-Token" , fake .TokenID )
113
113
th .TestHeader (t , r , "Content-Type" , "application/json" )
@@ -139,7 +139,7 @@ func TestCreate(t *testing.T) {
139
139
},
140
140
Name : "test_policy" ,
141
141
}
142
- s , err := archivepolicies .Create (context .TODO (), fake .ServiceClient (), opts ).Extract ()
142
+ s , err := archivepolicies .Create (context .TODO (), fake .ServiceClient (fakeServer ), opts ).Extract ()
143
143
th .AssertNoErr (t , err )
144
144
145
145
th .AssertDeepEquals (t , s .AggregationMethods , []string {
@@ -164,10 +164,10 @@ func TestCreate(t *testing.T) {
164
164
}
165
165
166
166
func TestUpdateArchivePolicy (t * testing.T ) {
167
- th .SetupHTTP ()
168
- defer th . TeardownHTTP ()
167
+ fakeServer := th .SetupHTTP ()
168
+ defer fakeServer . Teardown ()
169
169
170
- th .Mux .HandleFunc ("/v1/archive_policy/test_policy" , func (w http.ResponseWriter , r * http.Request ) {
170
+ fakeServer .Mux .HandleFunc ("/v1/archive_policy/test_policy" , func (w http.ResponseWriter , r * http.Request ) {
171
171
th .TestMethod (t , r , "PATCH" )
172
172
th .TestHeader (t , r , "X-Auth-Token" , fake .TokenID )
173
173
th .TestHeader (t , r , "Content-Type" , "application/json" )
@@ -192,7 +192,7 @@ func TestUpdateArchivePolicy(t *testing.T) {
192
192
},
193
193
},
194
194
}
195
- s , err := archivepolicies .Update (context .TODO (), fake .ServiceClient (), "test_policy" , updateOpts ).Extract ()
195
+ s , err := archivepolicies .Update (context .TODO (), fake .ServiceClient (fakeServer ), "test_policy" , updateOpts ).Extract ()
196
196
th .AssertNoErr (t , err )
197
197
198
198
th .AssertDeepEquals (t , s .AggregationMethods , []string {
@@ -215,15 +215,15 @@ func TestUpdateArchivePolicy(t *testing.T) {
215
215
}
216
216
217
217
func TestDelete (t * testing.T ) {
218
- th .SetupHTTP ()
219
- defer th . TeardownHTTP ()
218
+ fakeServer := th .SetupHTTP ()
219
+ defer fakeServer . Teardown ()
220
220
221
- th .Mux .HandleFunc ("/v1/archive_policy/test_policy" , func (w http.ResponseWriter , r * http.Request ) {
221
+ fakeServer .Mux .HandleFunc ("/v1/archive_policy/test_policy" , func (w http.ResponseWriter , r * http.Request ) {
222
222
th .TestMethod (t , r , "DELETE" )
223
223
th .TestHeader (t , r , "X-Auth-Token" , fake .TokenID )
224
224
w .WriteHeader (http .StatusNoContent )
225
225
})
226
226
227
- res := archivepolicies .Delete (context .TODO (), fake .ServiceClient (), "test_policy" )
227
+ res := archivepolicies .Delete (context .TODO (), fake .ServiceClient (fakeServer ), "test_policy" )
228
228
th .AssertNoErr (t , res .Err )
229
229
}
0 commit comments