test/models/storage/test_storage_accounts.rb in fog-azure-rm-0.0.5 vs test/models/storage/test_storage_accounts.rb in fog-azure-rm-0.0.6

@@ -43,12 +43,12 @@
       end
     end
   end
 
   def test_get_method_response
-    @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
+    response = ApiStub::Models::Storage::StorageAccount.create_storage_account(@storage_mgmt_client)
+    @service.stub :get_storage_account, response do
+      assert_instance_of Fog::Storage::AzureRM::StorageAccount, @storage_accounts.get('fog-test-rg', 'fog-test-storage-account')
     end
   end
 
   def test_check_name_availability_true_case
     @service.stub :check_storage_account_name_availability, true do