Add method all to images collection

fog · Oct 30, 2018 · d530738 · d530738
1 parent 8ff81af
commit d530738
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/lib/fog/azurerm/models/compute/images.rb b/lib/fog/azurerm/models/compute/images.rb
@@ -10,6 +10,14 @@ def create(attributes)
           image.save
         end
 
+        def all(async = false)
+          requires :resource_group
+          images = service.list_images(resource_group, async).map do |image|
+            Fog::Compute::AzureRM::Image.parse(image)
+          end
+          load(images)
+        end
+
         def get(resource_group_name, image_name)
           response = service.get_image(resource_group_name, image_name)
           image = Fog::Compute::AzureRM::Image.new(service: service)

diff --git a/test/models/compute/test_images.rb b/test/models/compute/test_images.rb
@@ -13,12 +13,23 @@ def test_collection_attributes
   end
 
   def test_collection_methods
-    methods = %i(get create)
+    methods = %i(all get create)
     methods.each do |method|
       assert_respond_to @images, method
     end
   end
 
+  def test_all_method_response
+    @service.stub :list_images, [@image] do
+      images_list = @images.all
+      assert_instance_of Fog::Compute::AzureRM::Images, images_list
+      assert images_list.size >= 1
+      images_list.each do |image|
+        assert_instance_of Fog::Compute::AzureRM::Image, image
+      end
+    end
+  end
+
   def test_get_method_response
     @service.stub :get_image, @image do
       assert_instance_of Fog::Compute::AzureRM::Image, @images.get('fog-test-rg', 'fog-test-image')