test_storage_accounts.rb in fog-azure-rm-0.0.5

- old
+ new

@@ -2,12 +2,13 @@
 
 # Test class for Availability Set Collection
 class TestStorageAccounts < Minitest::Test
   def setup
     @service = Fog::Storage::AzureRM.new(credentials)
+    @storage_mgmt_client = @service.instance_variable_get(:@storage_mgmt_client)
     @storage_accounts = Fog::Storage::AzureRM::StorageAccounts.new(resource_group: 'fog-test-rg', service: @service)
-    @response = [ApiStub::Models::Storage::StorageAccount.list_storage_account]
+    @list_storage_account_response = [ApiStub::Models::Storage::StorageAccount.create_storage_account(@storage_mgmt_client)]
   end
 
   def test_collection_methods
     methods = [
       :all,
@@ -21,32 +22,31 @@
   def test_collection_attributes
     assert @storage_accounts.respond_to? :resource_group, true
   end
 
   def test_all_method_response_for_rg
-    @service.stub :list_storage_account_for_rg, @response do
+    @service.stub :list_storage_account_for_rg, @list_storage_account_response do
       assert_instance_of Fog::Storage::AzureRM::StorageAccounts, @storage_accounts.all
       assert @storage_accounts.all.size >= 1
       @storage_accounts.all.each do |s|
         assert_instance_of Fog::Storage::AzureRM::StorageAccount, s
       end
     end
   end
 
   def test_all_method_response
     storage_accounts = Fog::Storage::AzureRM::StorageAccounts.new(service: @service)
-    @service.stub :list_storage_accounts, @response do
+    @service.stub :list_storage_accounts, @list_storage_account_response do
       assert_instance_of Fog::Storage::AzureRM::StorageAccounts, storage_accounts.all
       assert storage_accounts.all.size >= 1
       storage_accounts.all.each do |s|
         assert_instance_of Fog::Storage::AzureRM::StorageAccount, s
       end
     end
   end
 
   def test_get_method_response
-    response = [ApiStub::Models::Storage::StorageAccount.list_storage_account]
-    @service.stub :list_storage_account_for_rg, response do
+    @service.stub :list_storage_account_for_rg, @list_storage_account_response do
       assert_instance_of Fog::Storage::AzureRM::StorageAccount, @storage_accounts.get('fog-test-storage-account')
       assert @storage_accounts.get('wrong-name').nil?, true
     end
   end
 