test/models/dns/test_zones.rb in fog-azure-rm-0.0.4 vs test/models/dns/test_zones.rb in fog-azure-rm-0.0.5

@@ -28,10 +28,11 @@
       end
     end
   end
 
   def test_get_method_response
-    @service.stub :list_zones, @response do
+    response = ApiStub::Models::DNS::Zone.create_zone_obj
+    @service.stub :get_zone, response do
       assert_instance_of Fog::DNS::AzureRM::Zone, @zones.get('fog-test-rg', 'fog-test-zone.com')
     end
   end
 
   def test_check_for_zone_true_response