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

@@ -3,10 +3,11 @@
 # Test class for Sql Server Model
 class TestSqlServer < Minitest::Test
   def setup
     @service = Fog::Sql::AzureRM.new(credentials)
     @sql_server = sql_server(@service)
+    @sql_server_client = @service.instance_variable_get(:@sql_mgmt_client)
   end
 
   def test_model_methods
     methods = [
       :save,
@@ -34,10 +35,10 @@
       assert_respond_to @sql_server, attribute
     end
   end
 
   def test_save_method_response
-    create_response = ApiStub::Models::Sql::SqlServer.create_sql_server
+    create_response = ApiStub::Models::Sql::SqlServer.create_sql_server(@sql_server_client)
     @service.stub :create_or_update_sql_server, create_response do
       assert_instance_of Fog::Sql::AzureRM::SqlServer, @sql_server.save
     end
   end