refactor fake Delete function

This commit is contained in:
Jingtao Ren 2017-11-16 11:33:48 -08:00
parent 6b36a70d79
commit 83f18ca3f0

View File

@ -71,9 +71,9 @@ func (fLBC fakeAzureLBClient) Delete(resourceGroupName string, loadBalancerName
close(respChan) close(respChan)
close(errChan) close(errChan)
}() }()
if _, ok := fLBC.FakeStore[resourceGroupName]; ok { if rgLBs, ok := fLBC.FakeStore[resourceGroupName]; ok {
if _, ok := fLBC.FakeStore[resourceGroupName][loadBalancerName]; ok { if _, ok := rgLBs[loadBalancerName]; ok {
delete(fLBC.FakeStore[resourceGroupName], loadBalancerName) delete(rgLBs, loadBalancerName)
resp.Response = &http.Response{ resp.Response = &http.Response{
StatusCode: http.StatusAccepted, StatusCode: http.StatusAccepted,
} }
@ -207,9 +207,9 @@ func (fAPC fakeAzurePIPClient) Delete(resourceGroupName string, publicIPAddressN
close(respChan) close(respChan)
close(errChan) close(errChan)
}() }()
if _, ok := fAPC.FakeStore[resourceGroupName]; ok { if rgPIPs, ok := fAPC.FakeStore[resourceGroupName]; ok {
if _, ok := fAPC.FakeStore[resourceGroupName][publicIPAddressName]; ok { if _, ok := rgPIPs[publicIPAddressName]; ok {
delete(fAPC.FakeStore[resourceGroupName], publicIPAddressName) delete(rgPIPs, publicIPAddressName)
resp.Response = &http.Response{ resp.Response = &http.Response{
StatusCode: http.StatusAccepted, StatusCode: http.StatusAccepted,
} }
@ -434,9 +434,9 @@ func (fASC fakeAzureSubnetsClient) Delete(resourceGroupName string, virtualNetwo
}() }()
rgVnet := strings.Join([]string{resourceGroupName, virtualNetworkName}, "AND") rgVnet := strings.Join([]string{resourceGroupName, virtualNetworkName}, "AND")
if _, ok := fASC.FakeStore[rgVnet]; ok { if rgSubnets, ok := fASC.FakeStore[rgVnet]; ok {
if _, ok := fASC.FakeStore[rgVnet][subnetName]; ok { if _, ok := rgSubnets[subnetName]; ok {
delete(fASC.FakeStore[rgVnet], subnetName) delete(rgSubnets, subnetName)
resp.Response = &http.Response{ resp.Response = &http.Response{
StatusCode: http.StatusAccepted, StatusCode: http.StatusAccepted,
} }
@ -532,9 +532,9 @@ func (fNSG fakeAzureNSGClient) Delete(resourceGroupName string, networkSecurityG
close(respChan) close(respChan)
close(errChan) close(errChan)
}() }()
if _, ok := fNSG.FakeStore[resourceGroupName]; ok { if rgSGs, ok := fNSG.FakeStore[resourceGroupName]; ok {
if _, ok := fNSG.FakeStore[resourceGroupName][networkSecurityGroupName]; ok { if _, ok := rgSGs[networkSecurityGroupName]; ok {
delete(fNSG.FakeStore[resourceGroupName], networkSecurityGroupName) delete(rgSGs, networkSecurityGroupName)
resp.Response = &http.Response{ resp.Response = &http.Response{
StatusCode: http.StatusAccepted, StatusCode: http.StatusAccepted,
} }