azure-plus 0.6.5

data/test/support/name_generator.rb

@@ -0,0 +1,160 @@
+#-------------------------------------------------------------------------
+# # Copyright (c) Microsoft and contributors. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#--------------------------------------------------------------------------
+class NameGenerator
+  def initialize(&cleanup_proc)
+    @cleanup_proc = cleanup_proc
+    @names = []
+  end
+
+  def name
+    alpha = ('a'..'z').to_a
+    name = 10.times.map { alpha[Random.rand(alpha.size)] }.join
+    @names << name
+    name
+  end
+
+  def clean
+    @names.reject! do |name|
+      @cleanup_proc.call(name)
+    end
+    @names = []
+  end
+end
+
+TableNameHelper = NameGenerator.new do |name|
+  svc = Azure::Table::TableService.new
+  begin
+    svc.delete_table name
+  rescue
+  end
+end
+
+ContainerNameHelper = NameGenerator.new do |name|
+  svc = Azure::Blob::BlobService.new
+  begin
+    svc.delete_container name
+  rescue
+  end
+end
+
+QueueNameHelper = NameGenerator.new do |name|
+  svc = Azure::Queue::QueueService.new
+  begin
+    svc.delete_queue name
+  rescue
+  end
+end
+
+ServiceBusQueueNameHelper = NameGenerator.new do |name|
+  svc = Azure::ServiceBus::ServiceBusService.new
+  begin
+    svc.delete_queue name
+  rescue
+  end
+end
+
+ServiceBusTopicNameHelper = NameGenerator.new do |name|
+  svc = Azure::ServiceBus::ServiceBusService.new
+  begin
+    svc.delete_topic name
+  rescue
+  end
+end
+
+AffinityGroupNameHelper = NameGenerator.new do |name|
+  svc = Azure::BaseManagementService.new
+  begin
+    svc.delete_affinity_group name
+  rescue
+  end
+end
+
+class GB18030TestStringsClass
+  def get
+    {
+      'Chinese2B1'       => [0x2488, 0x2460, 0x216B, 0x3128,
+                             0x3129].pack('U*'),
+      'Chinese2B2'       => [0x554A, 0x963F, 0x9F3E, 0x9F44].pack('U*'),
+      'Chinese2B3'       => [0x4E02, 0x4E04, 0x72DA, 0x72DB].pack('U*'),
+      'Chinese2B4'       => [0x72DC, 0x72DD, 0xFA28, 0xFA29].pack('U*'),
+      'Chinese2B5'       => [0x02CA, 0x02CB, 0x02D9, 0x2013,
+                             0xE7F3, 0x3007].pack('U*'),
+      'Chinese4BExtA'    => [0x3400, 0x3401, 0x4DB4, 0x4DB5].pack('U*'),
+      'ChineseExtB'      => [0x020000, 0x020001, 0x02A6D5,
+                             0x02A6D6].pack('U*'),
+      'Tibetan'          => [0x0F56, 0x0F7C, 0x0F51, 0x0F0B, 0x0F61,
+                             0x0F72, 0x0F42, 0x0020, 0x0020, 0x0020,
+                             0x0F00, 0x0F01, 0x0F02, 0x0F03, 0x0F04,
+                             0x0F05, 0x0FA2, 0x0FA3, 0x0FA4,
+                             0x0FA5].pack('U*'),
+      'Uyghur'           => [0x0626, 0x06C7, 0x064A, 0x063A,
+                             0x06C7, 0x0631, 0x0020, 0x062A,
+                             0x0649, 0x0644, 0x0649, 0x0020,
+                             0x0020, 0x0020, 0x003B, 0x060B,
+                             0x0020, 0x060C, 0x0020, 0x060D,
+                             0x0020, 0x060E, 0x0020, 0x060F,
+                             0x0610, 0x0020, 0x0020, 0x0020,
+                             0xFEF2, 0x0020, 0xFEF3, 0x0020,
+                             0xFEF4, 0x0020, 0xFEF5, 0x0020,
+                             0xFEF6, 0x0020, 0xFEF7, 0x0020,
+                             0xFEF8, 0x0020, 0xFEF9, 0x0020,
+                             0xFEFA, 0x0020, 0xFEFB].pack('U*'),
+      'Mongolian'        => [0x182E, 0x1823, 0x1829, 0x182D,
+                             0x1823, 0x182F, 0x0020, 0x0020,
+                             0x1826, 0x1830, 0x1826, 0x182D,
+                             0x003B, 0x0020, 0x17F8, 0x17F9,
+                             0x0020, 0x188E, 0x188F, 0x1890,
+                             0x1891, 0x1892, 0x1893, 0x1894,
+                             0x1895, 0x1896, 0x1897].pack('U*'),
+      'Yi'               => [0xA188, 0xA320, 0xA071, 0xA0B7,
+                             0x003B, 0xA000, 0xA001, 0xA002,
+                             0xA003, 0xA004, 0xA005, 0xA006,
+                             0xA4C1, 0xA4C2, 0xA4C3, 0xA4C4,
+                             0xA4C5, 0xA4C6].pack('U*'),
+      'DehongDai'        => [0x1958, 0x1963, 0x196D, 0x1970,
+                             0x0020, 0x1956, 0x196D, 0x1970,
+                             0x0020, 0x1956, 0x196C, 0x1972,
+                             0x1951, 0x1968, 0x1952, 0x1970].pack('U*'),
+      'XishuangbannaDai' => [0x198E, 0x19B7, 0x1991, 0x19BA,
+                             0x1996, 0x19BA, 0x19C8, 0x1989,
+                             0x19B2, 0x19C7, 0x1989, 0x19B8,
+                             0x19C2, 0x1997, 0x19C3, 0x1993, 0x19B1].pack('U*')
+    }
+  end
+
+  def get_xml_10_fourth_ed_identifiers
+    # No characters from the following sets are valid,
+    # according to the XML library that the service uses:
+    #   * Chinese_4B_ExtA
+    #   * Chinese_ExtB
+    #   * Mongolian
+    #   * Yi
+    #   * Dehong_Dai
+    #   * Xishuangbanna_Dai
+    {
+      'Chinese_2B1' => [0x3128, 0x3129].pack('U*'),
+      'Chinese_2B2' => [0x554A, 0x963F, 0x9F3E, 0x9F44].pack('U*'),
+      'Chinese_2B3' => [0x4E02, 0x4E04, 0x72DA, 0x72DB].pack('U*'),
+      'Chinese_2B4' => [0x72DC, 0x72DD].pack('U*'),
+      'Chinese_2B5' => [0x3007].pack('U*'),
+      'Tibetan'     => [0x0F56, 0x0F7C, 0x0F51, 0x0F61, 0x0F72, 0x0F42,
+                        0x0FA2, 0x0FA3, 0x0FA4, 0x0FA5].pack('U*'),
+      'Uyghur'      => [0x0626, 0x06C7, 0x064A, 0x063A, 0x06C7, 0x0631,
+                        0x062A, 0x0649, 0x0644, 0x0649].pack('U*'),
+    }
+  end
+end
+
+GB18030TestStrings = GB18030TestStringsClass.new

data/test/support/stubs.rb

@@ -0,0 +1,42 @@
+#-------------------------------------------------------------------------
+# # Copyright (c) Microsoft and contributors. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#--------------------------------------------------------------------------
+require "ostruct"
+
+module Stubs
+  def double(messages={})
+    OpenStruct.new(messages)
+  end
+end
+
+class MiniTest::Mock
+  def stub(name, value=nil)
+    metaclass = class << self; self; end
+    metaclass.send :define_method, name do |*args|
+      value
+    end
+  end
+
+  alias_method :expect_without_stub_support, :expect
+
+  def expect(name, *args)
+    metaclass = class << self; self; end
+    metaclass.send :undef_method, name if respond_to?(name)
+    expect_without_stub_support(name, *args)
+  end
+end
+
+class MiniTest::Unit::TestCase
+  include Stubs
+end

data/test/support/virtual_machine_name_generator.rb

@@ -0,0 +1,102 @@
+#-------------------------------------------------------------------------
+# # Copyright (c) Microsoft and contributors. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#--------------------------------------------------------------------------
+class VirtualMachineNameGenerator
+  def initialize(&cleanup_proc)
+    @cleanup_proc = cleanup_proc
+    @names = []
+    @count = rand(900)
+  end
+
+  def random_text(text)
+    @count += 1
+    text + @count.to_s + "-#{Array.new(32) { rand(36).to_s(36) }.join[0..5]}"
+  end
+
+  def name
+    vm_name = random_text('test-')
+    cloud_name = random_text(vm_name + '-service-')
+    @names << [vm_name, cloud_name]
+    [vm_name, cloud_name]
+  end
+
+  def clean
+    @names.each do |name|
+      @cleanup_proc.call(name.first, name.last)
+    end
+    @names = []
+  end
+
+  def self.cleanup
+    puts 'Running after test cleanup.'
+    # Delete virtual machines
+    virtual_machine_service = Azure::VirtualMachineManagementService.new
+    virtualmachines = virtual_machine_service.list_virtual_machines
+    azure_cloud_service = Azure::CloudServiceManagementService.new
+    virtualmachines.each do |vm|
+      vm_name = vm.vm_name + '-service-'
+      csn = vm.cloud_service_name
+      if vm.vm_name.include?('test-') && csn.include?(vm_name)
+        azure_cloud_service.delete_cloud_service_deployment(csn) rescue nil
+      end
+    end
+
+    # Delete cloud services
+    cloud_services = azure_cloud_service.list_cloud_services
+    cloud_services.each do |cloud_service|
+      csn = cloud_service.name
+      if csn.include?('test-') && csn.include?('-service-')
+        azure_cloud_service.delete_cloud_service(csn)  rescue nil
+      end
+    end
+
+    # Delete disks
+    disk_service = VirtualMachineDiskManagementService.new
+    sleep 60
+    disks = disk_service.list_virtual_machine_disks
+    disks.each do |disk|
+      name = disk.name
+      if name.include?('-service-') && name.include?('test-') && !disk.attached
+        disk_service.delete_virtual_machine_disk(name) rescue nil
+      end
+    end
+
+    # Delete storage account
+    storage_service = Azure::StorageManagementService.new
+    storage_accounts = storage_service.list_storage_accounts
+    storage_accounts.each do |storage_account|
+      if storage_account.name.include?('storagetest')
+        storage_service.delete_storage_account(storage_account.name) rescue nil
+      end
+    end
+
+    # Delete affinity groups
+    ag_service = Azure::BaseManagementService.new
+    affinity_groups = ag_service.list_affinity_groups
+    affinity_groups.each do |affinity_group|
+      if affinity_group.name.include?('affinity-group-')
+        ag_service.delete_affinity_group(affinity_group.name) rescue nil
+      end
+    end
+  end
+end
+
+VirtualMachineNameHelper = VirtualMachineNameGenerator.new do |name, cloud_name|
+  cloud_service = Azure::CloudServiceManagementService.new
+  begin
+    cloud_service.delete_cloud_service_deployment(cloud_name)
+    cloud_service.delete_cloud_service(cloud_name)
+  rescue
+  end
+end

data/test/support/virtual_network_helper.rb

@@ -0,0 +1,73 @@
+#-------------------------------------------------------------------------
+# # Copyright (c) Microsoft and contributors. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#--------------------------------------------------------------------------
+
+class VirtualNetworkHelper
+  def self.check_config(virtual_networks,
+                        in_vnet_name,
+                        in_affinity_name,
+                        created_state,
+                        in_address_space,
+                        inputoptions)
+    virtual_networks.wont_be_nil
+    virtual_networks.must_be_kind_of Array
+    virtual_networks[0].must_be_kind_of(
+      Azure::VirtualNetworkManagement::VirtualNetwork
+    )
+
+    created_virtual_network = virtual_networks.select do |x|
+      x.name == in_vnet_name
+    end
+
+    created_virtual_network.wont_be_nil
+    created_virtual_network.size.must_equal 1
+    created_virtual_network[0].must_be_kind_of(
+      Azure::VirtualNetworkManagement::VirtualNetwork
+    )
+
+    # Verify global properties
+    created_virtual_network[0].name.must_equal in_vnet_name
+    created_virtual_network[0].affinity_group.must_equal in_affinity_name
+    created_virtual_network[0].state.must_equal created_state
+
+    # Verify address_space
+    created_virtual_network[0].address_space.must_be_kind_of Array
+    created_virtual_network[0].address_space.length.must_equal 1
+    created_virtual_network[0].address_space.first.must_be_kind_of String
+    created_virtual_network[0].address_space.first.must_equal(
+      in_address_space[0]
+    )
+
+    # Verify subnets
+    created_virtual_network[0].subnets.must_be_kind_of Array
+    created_virtual_network[0].subnets.length.must_equal 1
+    created_virtual_network[0].subnets.first[:name].must_equal(
+      inputoptions[:subnet][0][:name]
+    )
+    created_virtual_network[0].subnets.first[:address_prefix].must_equal(
+      "#{inputoptions[:subnet][0][:ip_address]}/"\
+      "#{inputoptions[:subnet][0][:cidr]}"
+    )
+
+    # Verify DNS Servers
+    created_virtual_network[0].dns_servers.must_be_kind_of Array
+    created_virtual_network[0].dns_servers.length.must_equal 1
+    created_virtual_network[0].dns_servers.first[:name].must_equal(
+      inputoptions[:dns][0][:name]
+    )
+    created_virtual_network[0].dns_servers.first[:ip_address].must_equal(
+      inputoptions[:dns][0][:ip_address]
+    )
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,53 @@
+#-------------------------------------------------------------------------
+# # Copyright (c) Microsoft and contributors. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#--------------------------------------------------------------------------
+require "minitest/autorun"
+require "mocha/setup"
+
+# Attempt to load turn to show formatted test results
+begin
+  require "turn"
+  Turn.config.format = :pretty
+  Turn.config.natural = true
+rescue LoadError
+end
+
+# add to the MiniTest DSL
+module Kernel
+  def need_tests_for(name)
+    describe "##{name}" do
+      it "needs unit tests" do
+        skip ""
+      end
+    end
+  end
+end
+
+
+Dir["./test/support/**/*.rb"].each { |dep| require dep }
+
+# mock configuration setup
+require 'azure'
+
+Azure.config.storage_account_name     = "accountname"
+Azure.config.storage_access_key       = "YWNjZXNzLWtleQ=="
+Azure.config.management_certificate   = File.expand_path("test/fixtures/management_certificate.pem")
+Azure.config.management_endpoint      = "https://management.core.windows.net"
+Azure.config.subscription_id          = "subscription-id"
+# Azure.config.storage_table_host     = 
+# Azure.config.storage_blob_host      = 
+# Azure.config.storage_queue_host     = 
+# Azure.config.sb_namespace           = 
+# Azure.config.sb_access_key          = 
+# Azure.config.sb_issuer              = 

data/test/unit/affinity_group/affinity_group_test.rb

@@ -0,0 +1,192 @@
+#-------------------------------------------------------------------------
+# Copyright 2013 Microsoft Open Technologies, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#--------------------------------------------------------------------------
+require 'test_helper'
+
+describe Azure::BaseManagementService do
+
+  subject { Azure::BaseManagementService.new }
+
+  let(:mock_request) { mock }
+  let(:response) { mock }
+  let(:response_xml) { nil }
+
+  before do
+    Loggerx.expects(:puts).returns(nil).at_least(0)
+    mock_request.expects(:call).returns(Nokogiri::XML response_xml).at_least(0)
+  end
+
+  describe '#list_affinity_groups' do
+    let(:request_path) { '/affinitygroups' }
+    let(:response_xml) { Fixtures['list_affinity_groups'] }
+    let(:method) { :get }
+
+    before do
+      ManagementHttpRequest.stubs(:new).with(method,
+                                             request_path,
+                                             nil).returns(mock_request)
+    end
+
+    it 'returns a list of affinity groups for the subscription' do
+      results = subject.list_affinity_groups
+      results.must_be_kind_of Array
+      results.length.must_equal 2
+      results.first.must_be_kind_of Azure::BaseManagement::AffinityGroup
+      affinity_group = results.first
+      affinity_group.name.must_equal 'AG1'
+      affinity_group.description.must_equal 'First Affinity Group'
+      affinity_group.location.must_equal 'West US'
+      affinity_group.capability.must_equal %w(PersistentVMRole HighMemory)
+    end
+  end
+
+  describe '#delete_affinity_group' do
+    before do
+      Azure::BaseManagementService.any_instance.stubs(
+        :list_affinity_groups
+      ).returns([])
+    end
+
+    it 'error if affinity group does not exists' do
+      affinity_group_name = 'unknown-affinity-group'
+      exception = assert_raises(Azure::Error::Error) do
+        subject.delete_affinity_group affinity_group_name
+      end
+      assert_match(/The affinity group does not exist./i, exception.message)
+    end
+
+    it 'error if affinity group name is empty' do
+      affinity_group_name = ''
+      exception = assert_raises(Azure::Error::Error) do
+        subject.delete_affinity_group affinity_group_name
+      end
+      assert_match(/The affinity group does not exist./i, exception.message)
+    end
+  end
+
+  describe '#create_affinity_group' do
+    let(:method) { :post }
+    let(:request_path) { '/affinitygroups' }
+    let(:name) { 'AG1' }
+    let(:location) { 'West US' }
+    let(:label) { 'Label Name' }
+
+    let(:location_request_path) { '/locations' }
+    let(:locations_xml) { Fixtures['list_locations'] }
+    let(:location_response) do
+      response = mock
+      response.stubs(:body).returns(locations_xml)
+      response
+    end
+    let(:location_response_body) { Nokogiri::XML location_response.body }
+
+    before do
+      ManagementHttpRequest.stubs(:new).with(method,
+                                             request_path,
+                                             anything).returns(mock_request)
+      mock_request.expects(:call).returns(
+        Nokogiri::XML response_xml
+      ).at_least(0)
+      mock_request = mock
+
+      ManagementHttpRequest.stubs(:new).with(:get,
+                                             location_request_path,
+                                             nil).returns(mock_request)
+      mock_request.expects(:call).returns(location_response_body).at_least(0)
+      mock_request = mock
+
+      ManagementHttpRequest.stubs(:new).with(:get,
+                                             request_path,
+                                             nil).returns(mock_request)
+      mock_request.expects(:call).returns(
+        Nokogiri::XML Fixtures['list_affinity_groups']
+      ).at_least(0)
+    end
+
+    it 'error if affinity group already exists' do
+      exception = assert_raises(Azure::Error::Error) do
+        subject.create_affinity_group(name, location, label)
+      end
+      assert_match(
+       /An affinity group #{name} already exists in the current subscription/i,
+       exception.message
+      )
+    end
+
+    it 'error if location is invalid' do
+      location = 'invalid-location'
+      exception = assert_raises(RuntimeError) do
+        subject.create_affinity_group('AGN', location, label)
+      end
+      assert_match(
+        /Value '#{location}' specified for parameter 'location' is invalid/i,
+        exception.message
+      )
+    end
+  end
+
+  describe '#get_affinity_group' do
+    let(:affinity_group_name) { 'AG1' }
+    let(:request_path) { '/affinitygroups/' + affinity_group_name }
+    let(:response_xml) { Fixtures['affinity_group'] }
+    let(:method) { :get }
+
+    before do
+      Azure::BaseManagementService.any_instance.stubs(
+        :affinity_group
+      ).returns(true)
+      ManagementHttpRequest.stubs(:new).with(
+        :get,
+        request_path,
+        nil
+      ).returns(mock_request)
+      mock_request.expects(:call).returns(
+        Nokogiri::XML response_xml
+      ).at_least(0)
+    end
+
+    it 'returns a properties of given affinity group' do
+      affinity_group = subject.get_affinity_group affinity_group_name
+      affinity_group.must_be_kind_of Azure::BaseManagement::AffinityGroup
+      affinity_group.name.must_equal 'AG1'
+      affinity_group.label.must_equal 'AG1'
+      affinity_group.description.must_equal 'First Affinity Group'
+      affinity_group.location.must_equal 'West US'
+      affinity_group.hosted_services.wont_be_nil
+      affinity_group.hosted_services.size.must_equal 1
+      affinity_group.storage_services.wont_be_nil
+      affinity_group.storage_services.size.must_equal 1
+      affinity_group.capability.must_equal %w[PersistentVMRole HighMemory]
+    end
+  end
+
+  describe '#update_affinity_group' do
+    let(:affinity_group_name) { 'AG1' }
+
+    it 'error if affinity group label is invalid' do
+      exception = assert_raises(RuntimeError) do
+        subject.update_affinity_group(affinity_group_name, nil)
+      end
+      assert_match(/Label name cannot be empty/i, exception.message)
+    end
+
+    it 'error if affinity group does not exist' do
+      exception = assert_raises(Azure::Error::Error) do
+        subject.update_affinity_group(nil, 'label')
+      end
+      assert_match(/The affinity group does not exist./i, exception.message)
+    end
+  end
+
+end