lib/fog/azurerm/models/network/virtual_network_gateways.rb in fog-azure-rm-0.0.9 vs lib/fog/azurerm/models/network/virtual_network_gateways.rb in fog-azure-rm-0.1.0

@@ -1,28 +1,25 @@
-require 'fog/core/collection'
-require 'fog/azurerm/models/network/virtual_network_gateway'
-
 module Fog
   module Network
     class AzureRM
       # VirtualNetworkGateways collection class for Network Service
       class VirtualNetworkGateways < Fog::Collection
-        model Fog::Network::AzureRM::VirtualNetworkGateway
+        model VirtualNetworkGateway
         attribute :resource_group
 
         def all
           requires :resource_group
           network_gateways = []
           service.list_virtual_network_gateways(resource_group).each do |gateway|
-            network_gateways << Fog::Network::AzureRM::VirtualNetworkGateway.parse(gateway)
+            network_gateways << VirtualNetworkGateway.parse(gateway)
           end
           load(network_gateways)
         end
 
         def get(resource_group_name, name)
           network_gateway = service.get_virtual_network_gateway(resource_group_name, name)
-          virtual_network_gateway = Fog::Network::AzureRM::VirtualNetworkGateway.new(service: service)
-          virtual_network_gateway.merge_attributes(Fog::Network::AzureRM::VirtualNetworkGateway.parse(network_gateway))
+          virtual_network_gateway = VirtualNetworkGateway.new(service: service)
+          virtual_network_gateway.merge_attributes(VirtualNetworkGateway.parse(network_gateway))
         end
       end
     end
   end
 end