test/models/compute/test_servers.rb in fog-azure-rm-0.0.4 vs test/models/compute/test_servers.rb in fog-azure-rm-0.0.5

@@ -3,11 +3,12 @@
 # Test class for Servers Collection
 class TestServers < Minitest::Test
   def setup
     @service = Fog::Compute::AzureRM.new(credentials)
     @servers = Fog::Compute::AzureRM::Servers.new(resource_group: 'fog-test-rg', service: @service)
-    @response = [ApiStub::Models::Compute::Server.create_windows_virtual_machine_response]
+    @compute_client = @service.instance_variable_get(:@compute_mgmt_client)
+    @response = [ApiStub::Models::Compute::Server.create_windows_virtual_machine_response(@compute_client)]
   end
 
   def test_collection_methods
     methods = [
       :all,
@@ -39,10 +40,10 @@
       assert @servers.get('wrong-name').nil?, true
     end
   end
 
   def test_get_from_remote_method_response
-    response = ApiStub::Models::Compute::Server.create_windows_virtual_machine_response
+    response = ApiStub::Models::Compute::Server.create_windows_virtual_machine_response(@compute_client)
     @service.stub :get_virtual_machine, response do
       assert_instance_of Fog::Compute::AzureRM::Server, @servers.get_from_remote('fog-test-rg', 'fog-test-server')
     end
   end
 end