@@ -84,6 +84,7 @@ var _ = ginkgo.Describe("Network", func() {
8484 PublicIpAddresses : []* csapi.PublicIpAddress {{Id : dummies .PublicIPID , Ipaddress : "fakeIP" }}}, nil )
8585 as .EXPECT ().NewAssociateIpAddressParams ().Return (& csapi.AssociateIpAddressParams {})
8686 as .EXPECT ().AssociateIpAddress (gomock .Any ())
87+ ns .EXPECT ().GetNetworkByID (dummies .ISONet1 .ID , gomock .Any ()).Return (& csapi.Network {Egressdefaultpolicy : false }, 1 , nil )
8788 fs .EXPECT ().NewCreateEgressFirewallRuleParams (dummies .ISONet1 .ID , gomock .Any ()).
8889 DoAndReturn (func (_ string , protocol string ) * csapi.CreateEgressFirewallRuleParams {
8990 p := & csapi.CreateEgressFirewallRuleParams {}
@@ -138,6 +139,7 @@ var _ = ginkgo.Describe("Network", func() {
138139 ginkgo .Context ("for a closed firewall" , func () {
139140 ginkgo .It ("OpenFirewallRule asks CloudStack to open the firewall" , func () {
140141 dummies .Zone1 .Network = dummies .ISONet1
142+ ns .EXPECT ().GetNetworkByID (dummies .ISONet1 .ID , gomock .Any ()).Return (& csapi.Network {Egressdefaultpolicy : false }, 1 , nil )
141143 fs .EXPECT ().NewCreateEgressFirewallRuleParams (dummies .ISONet1 .ID , gomock .Any ()).
142144 DoAndReturn (func (_ string , protocol string ) * csapi.CreateEgressFirewallRuleParams {
143145 p := & csapi.CreateEgressFirewallRuleParams {}
@@ -165,6 +167,7 @@ var _ = ginkgo.Describe("Network", func() {
165167 ginkgo .It ("OpenFirewallRule asks CloudStack to open the firewall anyway, but doesn't fail" , func () {
166168 dummies .Zone1 .Network = dummies .ISONet1
167169
170+ ns .EXPECT ().GetNetworkByID (dummies .ISONet1 .ID , gomock .Any ()).Return (& csapi.Network {Egressdefaultpolicy : false }, 1 , nil )
168171 fs .EXPECT ().NewCreateEgressFirewallRuleParams (dummies .ISONet1 .ID , gomock .Any ()).
169172 DoAndReturn (func (_ string , protocol string ) * csapi.CreateEgressFirewallRuleParams {
170173 p := & csapi.CreateEgressFirewallRuleParams {}
0 commit comments