fog-azure-rm 0.1.2 → 0.2.0

data/test/requests/sql/test_create_or_update_firewall_rule.rb

@@ -4,24 +4,22 @@ require File.expand_path '../../test_helper', __dir__
 class TestCreateOrUpdateFirewallRule < Minitest::Test
   def setup
     @service = Fog::Sql::AzureRM.new(credentials)
-    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+    @sql_manager_client = @service.instance_variable_get(:@sql_mgmt_client)
+    @firewall_rule = @sql_manager_client.servers
+    @data_hash = ApiStub::Requests::Sql::FirewallRule.firewall_rule_hash
   end
 
   def test_create_or_update_sql_server_firewall_rule_success
-    firewall_rule_response = ApiStub::Requests::Sql::FirewallRule.create_firewall_rule_response
-    data_hash = ApiStub::Requests::Sql::FirewallRule.firewall_rule_hash
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      RestClient.stub :put, firewall_rule_response do
-        assert_equal @service.create_or_update_firewall_rule(data_hash), JSON.parse(firewall_rule_response)
-      end
+    firewall_rule_response = ApiStub::Requests::Sql::FirewallRule.create_firewall_rule_response(@sql_manager_client)
+    @firewall_rule.stub :create_or_update_firewall_rule, firewall_rule_response do
+      assert_equal @service.create_or_update_firewall_rule(@data_hash), firewall_rule_response
     end
   end
 
   def test_create_or_update_sql_server_firewall_rule_failure
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      assert_raises ArgumentError do
-        @service.create_or_update_firewall_rule('test-resource-group', 'test-server-name')
-      end
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @firewall_rule.stub :create_or_update_firewall_rule, response do
+      assert_raises(RuntimeError) { @service.create_or_update_firewall_rule(@data_hash) }
     end
   end
 end

data/test/requests/sql/test_create_or_update_sql_server.rb

@@ -4,25 +4,22 @@ require File.expand_path '../../test_helper', __dir__
 class TestCreateOrUpdateSqlServer < Minitest::Test
   def setup
     @service = Fog::Sql::AzureRM.new(credentials)
-    @databases = @service.sql_servers
-    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+    @sql_manager_client = @service.instance_variable_get(:@sql_mgmt_client)
+    @server = @sql_manager_client.servers
+    @data_hash = ApiStub::Requests::Sql::SqlServer.sql_server_hash
   end
 
   def test_create_or_update_sql_server_success
-    server_response = ApiStub::Requests::Sql::SqlServer.create_sql_server_response
-    data_hash = ApiStub::Requests::Sql::SqlServer.sql_server_hash
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      RestClient.stub :put, server_response do
-        assert_equal @service.create_or_update_sql_server(data_hash), JSON.parse(server_response)
-      end
+    server_response = ApiStub::Requests::Sql::SqlServer.create_sql_server_response(@sql_manager_client)
+    @server.stub :create_or_update, server_response do
+      assert_equal @service.create_or_update_sql_server(@data_hash), server_response
     end
   end
 
   def test_create_or_update_sql_server_failure
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      assert_raises ArgumentError do
-        @service.create_or_update_sql_server('test-resource-group', 'test-server-name')
-      end
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @server.stub :create_or_update, response do
+      assert_raises(RuntimeError) { @service.create_or_update_sql_server(@data_hash) }
     end
   end
 end

data/test/requests/sql/test_delete_database.rb

@@ -4,31 +4,20 @@ require File.expand_path '../../test_helper', __dir__
 class TestDeleteDatabase < Minitest::Test
   def setup
     @service = Fog::Sql::AzureRM.new(credentials)
-    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+    @sql_manager_client = @service.instance_variable_get(:@sql_mgmt_client)
+    @databases = @sql_manager_client.databases
   end
 
   def test_delete_database_success
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      RestClient.stub :delete, true do
-        assert @service.delete_database('fog-test-rg', 'fog-test-server-name', 'database-name')
-      end
+    @databases.stub :delete, true do
+      assert @service.delete_database('fog-test-rg', 'fog-test-server-name', 'database-name')
     end
   end
 
   def test_delete_database_failure
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      assert_raises ArgumentError do
-        @service.delete_database('fog-test-zone')
-      end
-    end
-  end
-
-  def test_delete_database_exception
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @token_provider.stub :get_authentication_header, response do
-      assert_raises Exception do
-        assert @service.delete_database('fog-test-rg', 'fog-test-server-name', 'database-name')
-      end
+    @databases.stub :delete, response do
+      assert_raises(RuntimeError) { @service.delete_database('fog-test-rg', 'fog-test-server-name', 'database-name') }
     end
   end
 end

data/test/requests/sql/test_delete_firewall_rule.rb

@@ -4,31 +4,20 @@ require File.expand_path '../../test_helper', __dir__
 class TestDeleteFirewallRule < Minitest::Test
   def setup
     @service = Fog::Sql::AzureRM.new(credentials)
-    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+    @sql_manager_client = @service.instance_variable_get(:@sql_mgmt_client)
+    @firewall_rules = @sql_manager_client.servers
   end
 
   def test_delete_sql_server_firewall_rule_success
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      RestClient.stub :delete, true do
-        assert @service.delete_firewall_rule('fog-test-rg', 'fog-test-server-name', 'rule-name')
-      end
+    @firewall_rules.stub :delete_firewall_rule, true do
+      assert @service.delete_firewall_rule('fog-test-rg', 'fog-test-server-name', 'rule-name')
     end
   end
 
   def test_delete_sql_server_firewall_rule_failure
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      assert_raises ArgumentError do
-        @service.delete_firewall_rule('fog-test-name')
-      end
-    end
-  end
-
-  def test_delete_sql_server_firewall_rule_exception
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @token_provider.stub :get_authentication_header, response do
-      assert_raises Exception do
-        assert @service.delete_firewall_rule('fog-test-rg', 'fog-test-server-name', 'rule-name')
-      end
+    @firewall_rules.stub :delete_firewall_rule, response do
+      assert_raises(RuntimeError) { @service.delete_firewall_rule('fog-test-rg', 'fog-test-server-name', 'rule-name') }
     end
   end
 end

data/test/requests/sql/test_delete_sql_server.rb

@@ -4,31 +4,20 @@ require File.expand_path '../../test_helper', __dir__
 class TestDeleteSqlServer < Minitest::Test
   def setup
     @service = Fog::Sql::AzureRM.new(credentials)
-    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+    @sql_manager_client = @service.instance_variable_get(:@sql_mgmt_client)
+    @servers = @sql_manager_client.servers
   end
 
   def test_delete_sql_server_success
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      RestClient.stub :delete, true do
-        assert @service.delete_sql_server('fog-test-rg', 'fog-test-server-name')
-      end
+    @servers.stub :delete, true do
+      assert @service.delete_sql_server('fog-test-rg', 'fog-test-server-name')
     end
   end
 
   def test_delete_sql_server_failure
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      assert_raises ArgumentError do
-        @service.delete_sql_server('fog-test-zone')
-      end
-    end
-  end
-
-  def test_delete_sql_server_exception
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @token_provider.stub :get_authentication_header, response do
-      assert_raises Exception do
-        assert @service.delete_sql_server('fog-test-rg', 'fog-test-server-name')
-      end
+    @servers.stub :delete, response do
+      assert_raises(RuntimeError) { @service.delete_sql_server('fog-test-rg', 'fog-test-server-name') }
     end
   end
 end

data/test/requests/sql/test_get_database.rb

@@ -4,32 +4,21 @@ require File.expand_path '../../test_helper', __dir__
 class TestGetDatabase < Minitest::Test
   def setup
     @service = Fog::Sql::AzureRM.new(credentials)
-    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+    @sql_manager_client = @service.instance_variable_get(:@sql_mgmt_client)
+    @databases = @sql_manager_client.databases
   end
 
   def test_get_database_success
-    create_response = ApiStub::Requests::Sql::SqlDatabase.create_database_response
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      RestClient.stub :get, create_response do
-        assert_equal @service.get_database('fog-test-rg', 'fog-test-server-name', 'fog-test-database-name'), JSON.parse(create_response)
-      end
+    create_response = ApiStub::Requests::Sql::SqlDatabase.create_database_response(@sql_manager_client)
+    @databases.stub :get, create_response do
+      assert_equal @service.get_database('fog-test-rg', 'fog-test-server-name', 'fog-test-database-name'), create_response
     end
   end
 
   def test_get_database_failure
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      assert_raises ArgumentError do
-        @service.get_database('fog-test-rg')
-      end
-    end
-  end
-
-  def test_get_database_exception
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @token_provider.stub :get_authentication_header, response do
-      assert_raises Exception do
-        @service.get_database('fog-test-rg', 'fog-test-server-name', 'fog-test-database-name')
-      end
+    @databases.stub :get, response do
+      assert_raises(RuntimeError) { @service.get_database('fog-test-rg', 'fog-test-server-name', 'fog-test-database-name') }
     end
   end
 end

data/test/requests/sql/test_get_firewall_rule.rb

@@ -4,32 +4,21 @@ require File.expand_path '../../test_helper', __dir__
 class TestGetFirewallRule < Minitest::Test
   def setup
     @service = Fog::Sql::AzureRM.new(credentials)
-    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+    @sql_manager_client = @service.instance_variable_get(:@sql_mgmt_client)
+    @servers = @sql_manager_client.servers
   end
 
   def test_get_sql_server_firewall_rule_success
-    create_response = ApiStub::Requests::Sql::FirewallRule.create_firewall_rule_response
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      RestClient.stub :get, create_response do
-        assert_equal @service.get_firewall_rule('fog-test-rg', 'fog-test-server-name', 'rule-name'), JSON.parse(create_response)
-      end
+    create_response = ApiStub::Requests::Sql::FirewallRule.create_firewall_rule_response(@sql_manager_client)
+    @servers.stub :get_firewall_rule, create_response do
+      assert_equal @service.get_firewall_rule('fog-test-rg', 'fog-test-server-name', 'rule-name'), create_response
     end
   end
 
   def test_get_sql_server_firewall_rule_failure
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      assert_raises ArgumentError do
-        @service.get_firewall_rule('fog-test-rg')
-      end
-    end
-  end
-
-  def test_get_sql_server_firewall_rule_exception
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @token_provider.stub :get_authentication_header, response do
-      assert_raises Exception do
-        @service.get_firewall_rule('fog-test-rg', 'fog-test-server-name', 'rule-name')
-      end
+    @servers.stub :get_firewall_rule, response do
+      assert_raises(RuntimeError) { @service.get_firewall_rule('fog-test-rg', 'fog-test-server-name', 'rule-name') }
     end
   end
 end

data/test/requests/sql/test_get_sql_server.rb

@@ -4,32 +4,21 @@ require File.expand_path '../../test_helper', __dir__
 class TestGetSqlServer < Minitest::Test
   def setup
     @service = Fog::Sql::AzureRM.new(credentials)
-    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+    @sql_manager_client = @service.instance_variable_get(:@sql_mgmt_client)
+    @servers = @sql_manager_client.servers
   end
 
   def test_get_sql_server_success
-    create_response = ApiStub::Requests::Sql::SqlServer.create_sql_server_response
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      RestClient.stub :get, create_response do
-        assert_equal @service.get_sql_server('fog-test-rg', 'fog-test-server-name'), JSON.parse(create_response)
-      end
+    create_response = ApiStub::Requests::Sql::SqlServer.create_sql_server_response(@sql_manager_client)
+    @servers.stub :get_by_resource_group, create_response do
+      assert_equal @service.get_sql_server('fog-test-rg', 'fog-test-server-name'), create_response
     end
   end
 
   def test_get_sql_server_failure
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      assert_raises ArgumentError do
-        @service.get_sql_server('fog-test-rg')
-      end
-    end
-  end
-
-  def test_get_sql_server_exception
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @token_provider.stub :get_authentication_header, response do
-      assert_raises Exception do
-        @service.get_sql_server('fog-test-rg', 'fog-test-server-name')
-      end
+    @servers.stub :get_by_resource_group, response do
+      assert_raises(RuntimeError) { @service.get_sql_server('fog-test-rg', 'fog-test-server-name') }
     end
   end
 end

data/test/requests/sql/test_list_databases.rb

@@ -4,32 +4,21 @@ require File.expand_path '../../test_helper', __dir__
 class TestListDatabases < Minitest::Test
   def setup
     @service = Fog::Sql::AzureRM.new(credentials)
-    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+    @sql_manager_client = @service.instance_variable_get(:@sql_mgmt_client)
+    @database = @sql_manager_client.databases
   end
 
   def test_list_databases_success
-    list_response = ApiStub::Requests::Sql::SqlDatabase.list_database_response
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      RestClient.stub :get, list_response do
-        assert_equal @service.list_databases('fog-test-rg', 'fog-test-server-name'), JSON.parse(list_response)['value']
-      end
+    list_response = ApiStub::Requests::Sql::SqlDatabase.list_database_response(@sql_manager_client)
+    @database.stub :list_by_server, list_response do
+      assert_equal @service.list_databases('fog-test-rg', 'fog-test-server-name'), list_response
     end
   end
 
   def test_list_databases_failure
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      assert_raises ArgumentError do
-        @service.list_databases('fog-test-rg')
-      end
-    end
-  end
-
-  def test_list_databases_exception
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @token_provider.stub :get_authentication_header, response do
-      assert_raises Exception do
-        @service.list_databases('fog-test-rg', 'fog-test-zone')
-      end
+    @database.stub :list_by_server, response do
+      assert_raises(RuntimeError) { @service.list_databases('fog-test-rg', 'fog-test-server-name') }
     end
   end
 end

data/test/requests/sql/test_list_firewall_rules.rb

@@ -4,32 +4,21 @@ require File.expand_path '../../test_helper', __dir__
 class TestListFirewallRules < Minitest::Test
   def setup
     @service = Fog::Sql::AzureRM.new(credentials)
-    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+    @sql_manager_client = @service.instance_variable_get(:@sql_mgmt_client)
+    @server = @sql_manager_client.servers
   end
 
   def test_list_sql_server_firewall_rules_success
-    list_response = ApiStub::Requests::Sql::FirewallRule.list_firewall_rule_response
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      RestClient.stub :get, list_response do
-        assert_equal @service.list_firewall_rules('fog-test-rg', 'server-name'), JSON.parse(list_response)['value']
-      end
+    list_response = ApiStub::Requests::Sql::FirewallRule.list_firewall_rule_response(@sql_manager_client)
+    @server.stub :list_firewall_rules, list_response do
+      assert_equal @service.list_firewall_rules('fog-test-rg', 'server-name'), list_response
     end
   end
 
   def test_list_sql_servers_firewall_rules_failure
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      assert_raises ArgumentError do
-        @service.list_firewall_rules('fog-test-rg')
-      end
-    end
-  end
-
-  def test_list_sql_servers_firewall_rules_exception
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @token_provider.stub :get_authentication_header, response do
-      assert_raises Exception do
-        @service.list_firewall_rules('fog-test-rg', 'server-name')
-      end
+    @server.stub :list_firewall_rules, response do
+      assert_raises(RuntimeError) { @service.list_firewall_rules('fog-test-rg', 'server-name') }
     end
   end
 end

data/test/requests/sql/test_list_sql_servers.rb

@@ -4,32 +4,21 @@ require File.expand_path '../../test_helper', __dir__
 class TestListSqlServers < Minitest::Test
   def setup
     @service = Fog::Sql::AzureRM.new(credentials)
-    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+    @sql_manager_client = @service.instance_variable_get(:@sql_mgmt_client)
+    @server = @sql_manager_client.servers
   end
 
   def test_list_sql_servers_success
-    list_response = ApiStub::Requests::Sql::SqlServer.list_sql_server_response
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      RestClient.stub :get, list_response do
-        assert_equal @service.list_sql_servers('fog-test-rg'), JSON.parse(list_response)['value']
-      end
+    list_response = ApiStub::Requests::Sql::SqlServer.list_sql_server_response(@sql_manager_client)
+    @server.stub :list_by_resource_group, list_response do
+      assert_equal @service.list_sql_servers('fog-test-rg'), list_response
     end
   end
 
   def test_list_sql_servers_failure
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      assert_raises ArgumentError do
-        @service.list_sql_servers('fog-test-rg', 'server-name')
-      end
-    end
-  end
-
-  def test_list_sql_servers_exception
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @token_provider.stub :get_authentication_header, response do
-      assert_raises Exception do
-        @service.list_sql_servers('fog-test-rg')
-      end
+    @server.stub :list_by_resource_group, response do
+      assert_raises(RuntimeError) { @service.list_sql_servers('fog-test-rg') }
     end
   end
 end

data/test/requests/storage/test_check_container_exists.rb

@@ -0,0 +1,18 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Container Class
+class TestCheckContainerExists < Minitest::Test
+  # This class posesses the test cases for the requests of checking container exists.
+  def setup
+    @service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    @blob_client = @service.instance_variable_get(:@blob_client)
+
+    @container = ApiStub::Requests::Storage::Directory.container
+  end
+
+  def test_check_container_exists_success
+    @blob_client.stub :get_container_properties, @container do
+      assert_equal @service.check_container_exists('test_container'), true
+    end
+  end
+end