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

@@ -2,12 +2,13 @@
 
 # Test class for Sql Server Collection
 class TestSqlServers < Minitest::Test
   def setup
     @service = Fog::Sql::AzureRM.new(credentials)
-    @sql_servers = Fog::Sql::AzureRM::SqlServers.new(resource_group: 'fog-test-rg', name: 'database-name', location: 'eastus', version: '2.0', administrator_login: 'test-admin', administrator_login_password: 'test-apase2', service: @service)
-    @list_sql_server_response = [ApiStub::Models::Sql::SqlServer.create_sql_server]
+    @sql_servers = sql_servers(@service)
+    @sql_server_client = @service.instance_variable_get(:@sql_mgmt_client)
+    @sql_server_response = ApiStub::Models::Sql::SqlServer.create_sql_server(@sql_server_client)
   end
 
   def test_collection_methods
     methods = [
       :all,
@@ -21,21 +22,20 @@
   def test_collection_attributes
     assert_respond_to @sql_servers, :resource_group
   end
 
   def test_all_method_response
-    @service.stub :list_sql_servers, @list_sql_server_response do
+    @service.stub :list_sql_servers, [@sql_server_response] do
       assert_instance_of Fog::Sql::AzureRM::SqlServers, @sql_servers.all
       assert @sql_servers.all.size >= 1
       @sql_servers.all.each do |s|
         assert_instance_of Fog::Sql::AzureRM::SqlServer, s
       end
     end
   end
 
   def test_get_method_response
-    response = ApiStub::Models::Sql::SqlServer.create_sql_server
-    @service.stub :get_sql_server, response do
+    @service.stub :get_sql_server, @sql_server_response do
       assert_instance_of Fog::Sql::AzureRM::SqlServer, @sql_servers.get('fog-test-rg', 'fog-test-server-name')
     end
   end
 end