@@ -35,7 +35,10 @@ import (
3535)
3636
3737var _ = Describe ("AggregatesController" , func () {
38- var tc * AggregatesController
38+ var (
39+ tc * AggregatesController
40+ fakeServer testhelper.FakeServer
41+ )
3942 const EOF = "EOF"
4043 const AggregateListBodyEmpty = `
4144{
@@ -101,19 +104,19 @@ var _ = Describe("AggregatesController", func() {
101104
102105 BeforeEach (func (ctx context.Context ) {
103106 By ("Setting up the OpenStack http mock server" )
104- testhelper .SetupHTTP ()
107+ fakeServer = testhelper .SetupHTTP ()
105108
106109 By ("Creating the AggregatesController" )
107110 tc = & AggregatesController {
108111 Client : k8sClient ,
109112 Scheme : k8sClient .Scheme (),
110- computeClient : client .ServiceClient (),
113+ computeClient : client .ServiceClient (fakeServer ),
111114 }
112115 })
113116
114117 AfterEach (func () {
115118 By ("Tearing down the OpenStack http mock server" )
116- testhelper . TeardownHTTP ()
119+ fakeServer . Teardown ()
117120
118121 By ("Deleting the Hypervisor resource" )
119122 hypervisor := & kvmv1.Hypervisor {}
@@ -139,14 +142,14 @@ var _ = Describe("AggregatesController", func() {
139142 Expect (k8sClient .Create (ctx , hypervisor )).To (Succeed ())
140143
141144 // Mock resourceproviders.GetAggregates
142- testhelper .Mux .HandleFunc ("GET /os-aggregates" , func (w http.ResponseWriter , r * http.Request ) {
145+ fakeServer .Mux .HandleFunc ("GET /os-aggregates" , func (w http.ResponseWriter , r * http.Request ) {
143146 w .Header ().Add ("Content-Type" , "application/json" )
144147 w .WriteHeader (http .StatusOK )
145148
146149 _ , err := fmt .Fprint (w , AggregateListBodyEmpty )
147150 Expect (err ).NotTo (HaveOccurred ())
148151 })
149- testhelper .Mux .HandleFunc ("POST /os-aggregates" , func (w http.ResponseWriter , r * http.Request ) {
152+ fakeServer .Mux .HandleFunc ("POST /os-aggregates" , func (w http.ResponseWriter , r * http.Request ) {
150153 w .Header ().Add ("Content-Type" , "application/json" )
151154 w .WriteHeader (http .StatusOK )
152155
@@ -155,7 +158,7 @@ var _ = Describe("AggregatesController", func() {
155158 })
156159
157160 // Mock resourceproviders.UpdateAggregates
158- testhelper .Mux .HandleFunc ("POST /os-aggregates/42/action" , func (w http.ResponseWriter , r * http.Request ) {
161+ fakeServer .Mux .HandleFunc ("POST /os-aggregates/42/action" , func (w http.ResponseWriter , r * http.Request ) {
159162 // parse request
160163 Expect (r .Header .Get ("Content-Type" )).To (Equal ("application/json" ))
161164 expectedBody := `{"add_host":{"host":"hv-test"}}`
@@ -204,7 +207,7 @@ var _ = Describe("AggregatesController", func() {
204207 Expect (k8sClient .Status ().Update (ctx , hypervisor )).To (Succeed ())
205208
206209 // Mock resourceproviders.GetAggregates
207- testhelper .Mux .HandleFunc ("GET /os-aggregates" , func (w http.ResponseWriter , r * http.Request ) {
210+ fakeServer .Mux .HandleFunc ("GET /os-aggregates" , func (w http.ResponseWriter , r * http.Request ) {
208211 w .Header ().Add ("Content-Type" , "application/json" )
209212 w .WriteHeader (http .StatusOK )
210213
@@ -227,8 +230,8 @@ var _ = Describe("AggregatesController", func() {
227230 _ , err = fmt .Fprint (w , AggregateRemoveHostBody )
228231 Expect (err ).NotTo (HaveOccurred ())
229232 }
230- testhelper .Mux .HandleFunc ("POST /os-aggregates/100001/action" , expectRemoveHostFromAggregate )
231- testhelper .Mux .HandleFunc ("POST /os-aggregates/99/action" , expectRemoveHostFromAggregate )
233+ fakeServer .Mux .HandleFunc ("POST /os-aggregates/100001/action" , expectRemoveHostFromAggregate )
234+ fakeServer .Mux .HandleFunc ("POST /os-aggregates/99/action" , expectRemoveHostFromAggregate )
232235 })
233236
234237 It ("should update Aggregates and set status condition when Aggregates differ" , func () {
0 commit comments