test/models/sql/test_firewall_rule.rb in fog-azure-rm-0.1.2 vs test/models/sql/test_firewall_rule.rb in fog-azure-rm-0.2.0

@@ -3,10 +3,11 @@
 # Test class for Sql Server Firewall Rule Model
 class TestFirewallRule < Minitest::Test
   def setup
     @service = Fog::Sql::AzureRM.new(credentials)
     @firewall_rule = sql_server_firewall_rule(@service)
+    @firewall_client = @service.instance_variable_get(:@sql_mgmt_client)
   end
 
   def test_model_methods
     methods = [
       :save,
@@ -32,10 +33,10 @@
       assert_respond_to @firewall_rule, attribute
     end
   end
 
   def test_save_method_response
-    create_response = ApiStub::Models::Sql::SqlFirewallRule.create_firewall_rule
+    create_response = ApiStub::Models::Sql::SqlFirewallRule.create_firewall_rule(@firewall_client)
     @service.stub :create_or_update_firewall_rule, create_response do
       assert_instance_of Fog::Sql::AzureRM::FirewallRule, @firewall_rule.save
     end
   end