@@ -13,10 +13,10 @@ import (
1313)
1414
1515func TestListArchivePolicies (t * testing.T ) {
16- th .SetupHTTP ()
17- defer th . TeardownHTTP ()
16+ fakeServer := th .SetupHTTP ()
17+ defer fakeServer . Teardown ()
1818
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 ) {
2020 th .TestMethod (t , r , "GET" )
2121 th .TestHeader (t , r , "X-Auth-Token" , fake .TokenID )
2222
@@ -28,7 +28,7 @@ func TestListArchivePolicies(t *testing.T) {
2828
2929 expected := ListArchivePoliciesExpected
3030 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 ) {
3232 pages ++
3333
3434 actual , err := archivepolicies .ExtractArchivePolicies (page )
@@ -46,10 +46,10 @@ func TestListArchivePolicies(t *testing.T) {
4646}
4747
4848func TestListArchivePoliciesAllPages (t * testing.T ) {
49- th .SetupHTTP ()
50- defer th . TeardownHTTP ()
49+ fakeServer := th .SetupHTTP ()
50+ defer fakeServer . Teardown ()
5151
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 ) {
5353 th .TestMethod (t , r , "GET" )
5454 th .TestHeader (t , r , "X-Auth-Token" , fake .TokenID )
5555
@@ -59,17 +59,17 @@ func TestListArchivePoliciesAllPages(t *testing.T) {
5959 fmt .Fprintf (w , ArchivePoliciesListResult )
6060 })
6161
62- allPages , err := archivepolicies .List (fake .ServiceClient ()).AllPages (context .TODO ())
62+ allPages , err := archivepolicies .List (fake .ServiceClient (fakeServer )).AllPages (context .TODO ())
6363 th .AssertNoErr (t , err )
6464 _ , err = archivepolicies .ExtractArchivePolicies (allPages )
6565 th .AssertNoErr (t , err )
6666}
6767
6868func TestGetArchivePolicy (t * testing.T ) {
69- th .SetupHTTP ()
70- defer th . TeardownHTTP ()
69+ fakeServer := th .SetupHTTP ()
70+ defer fakeServer . Teardown ()
7171
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 ) {
7373 th .TestMethod (t , r , "GET" )
7474 th .TestHeader (t , r , "X-Auth-Token" , fake .TokenID )
7575
@@ -79,7 +79,7 @@ func TestGetArchivePolicy(t *testing.T) {
7979 fmt .Fprintf (w , ArchivePolicyGetResult )
8080 })
8181
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 ()
8383 th .AssertNoErr (t , err )
8484
8585 th .AssertDeepEquals (t , s .AggregationMethods , []string {
@@ -104,10 +104,10 @@ func TestGetArchivePolicy(t *testing.T) {
104104}
105105
106106func TestCreate (t * testing.T ) {
107- th .SetupHTTP ()
108- defer th . TeardownHTTP ()
107+ fakeServer := th .SetupHTTP ()
108+ defer fakeServer . Teardown ()
109109
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 ) {
111111 th .TestMethod (t , r , "POST" )
112112 th .TestHeader (t , r , "X-Auth-Token" , fake .TokenID )
113113 th .TestHeader (t , r , "Content-Type" , "application/json" )
@@ -139,7 +139,7 @@ func TestCreate(t *testing.T) {
139139 },
140140 Name : "test_policy" ,
141141 }
142- s , err := archivepolicies .Create (context .TODO (), fake .ServiceClient (), opts ).Extract ()
142+ s , err := archivepolicies .Create (context .TODO (), fake .ServiceClient (fakeServer ), opts ).Extract ()
143143 th .AssertNoErr (t , err )
144144
145145 th .AssertDeepEquals (t , s .AggregationMethods , []string {
@@ -164,10 +164,10 @@ func TestCreate(t *testing.T) {
164164}
165165
166166func TestUpdateArchivePolicy (t * testing.T ) {
167- th .SetupHTTP ()
168- defer th . TeardownHTTP ()
167+ fakeServer := th .SetupHTTP ()
168+ defer fakeServer . Teardown ()
169169
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 ) {
171171 th .TestMethod (t , r , "PATCH" )
172172 th .TestHeader (t , r , "X-Auth-Token" , fake .TokenID )
173173 th .TestHeader (t , r , "Content-Type" , "application/json" )
@@ -192,7 +192,7 @@ func TestUpdateArchivePolicy(t *testing.T) {
192192 },
193193 },
194194 }
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 ()
196196 th .AssertNoErr (t , err )
197197
198198 th .AssertDeepEquals (t , s .AggregationMethods , []string {
@@ -215,15 +215,15 @@ func TestUpdateArchivePolicy(t *testing.T) {
215215}
216216
217217func TestDelete (t * testing.T ) {
218- th .SetupHTTP ()
219- defer th . TeardownHTTP ()
218+ fakeServer := th .SetupHTTP ()
219+ defer fakeServer . Teardown ()
220220
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 ) {
222222 th .TestMethod (t , r , "DELETE" )
223223 th .TestHeader (t , r , "X-Auth-Token" , fake .TokenID )
224224 w .WriteHeader (http .StatusNoContent )
225225 })
226226
227- res := archivepolicies .Delete (context .TODO (), fake .ServiceClient (), "test_policy" )
227+ res := archivepolicies .Delete (context .TODO (), fake .ServiceClient (fakeServer ), "test_policy" )
228228 th .AssertNoErr (t , res .Err )
229229}
0 commit comments