stuartpreston-azure-sdk-for-ruby 0.6.5

data/test/integration/vm/VM_Operations_test.rb

@@ -0,0 +1,173 @@
+#-------------------------------------------------------------------------
+# 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 'integration/test_helper'
+
+describe Azure::VirtualMachineManagementService do
+  let(:vm_name) { 'test-operation-vm' }
+  let(:csn) { vm_name + '-service-qwert' }
+  subject { Azure::VirtualMachineManagementService.new }
+
+  before do
+    Loggerx.expects(:puts).returns(nil).at_least(0)
+    params = {
+      vm_name: vm_name,
+      vm_user: 'user',
+      image: LinuxImage.name,
+      password: 'User123',
+      location: LinuxImageLocation
+    }
+    options = {
+      storage_account_name: StorageAccountName,
+      cloud_service_name: csn,
+    }
+    vm = subject.get_virtual_machine(vm_name, csn)
+    if vm.nil?
+      subject.create_virtual_machine(params, options)
+      sleep 60 # Wait to finish provision.
+    end
+  end
+
+  describe 'Operations on virtual machine' do
+
+    describe '#shutdown_virtual_machine' do
+      it 'shuts down virtual machine' do
+        subject.shutdown_virtual_machine(vm_name, csn)
+        vm = subject.get_virtual_machine(vm_name, csn)
+        %w(StoppedVM StoppedDeallocated).must_include  vm.status
+
+        msg = subject.shutdown_virtual_machine(vm_name, csn)
+        emsg = 'Cannot perform the shutdown operation on a stopped virtual machine'
+        assert_match(/#{emsg}/, msg)
+      end
+
+      it 'raises an error if virtual machine doesn\'t exists' do
+        name = 'test-shutdown'
+        msg = subject.shutdown_virtual_machine(name, csn)
+        assert_match(/Cannot find virtual machine \"#{name}\" under cloud service \"#{csn}\"/, msg)
+      end
+
+    end
+
+    describe '#start_virtual_machine' do
+      before do
+        subject.shutdown_virtual_machine(vm_name, csn)
+      end
+
+      it 'starts virtual machine' do
+        subject.start_virtual_machine(vm_name, csn)
+        vm = subject.get_virtual_machine(vm_name, csn)
+        statuses = %w(ReadyRole Provisioning CreatingVM RoleStateUnknown)
+        statuses.must_include  vm.status
+      end
+
+    end
+
+    describe '#restart_virtual_machine' do
+      before do
+        subject.restart_virtual_machine(vm_name, csn)
+      end
+
+      it 'restarts virtual machine' do
+        vm = subject.get_virtual_machine(vm_name, csn)
+        %w(ReadyRole Provisioning RoleStateUnknown).must_include  vm.status
+      end
+    end
+
+    describe '#get_virtual_machine' do
+      it 'should return virtual machine object' do
+        vm = subject.get_virtual_machine(vm_name, csn)
+        vm.vm_name.must_equal vm_name
+        vm.cloud_service_name.must_equal csn
+      end
+    end
+
+    describe '#list_virtual_machines' do
+      it 'returns a list of virtual machines' do
+        vms = subject.list_virtual_machines
+        vms.wont_be_nil
+        vms.must_be_kind_of Array
+        vm = vms.first
+        vm.must_be_kind_of Azure::VirtualMachineManagement::VirtualMachine
+        vm_names = vms.map(&:vm_name)
+        vm_names.must_include  vm.vm_name
+      end
+    end
+
+    describe '#add_data_disk' do
+      it 'add data disk to virtual machine' do
+        others = { disk_size: 100 }
+        subject.add_data_disk(vm_name, csn, others)
+        dms = VirtualMachineDiskManagementService.new
+        disks = dms.list_virtual_machine_disks
+        disks = disks.select { |x| (/#{csn}/ =~ x.name) && x.os_type.empty? }
+        assert_operator disks.size, :>=, 1
+        disks.first.size.must_equal others[:disk_size].to_s
+      end
+    end
+
+    describe 'Add, Update, Delete endpoints' do
+
+      it 'should add endpoints to virtual machine.' do
+        ep1 =  {
+          name: 'endpoint-1',
+          public_port: 777,
+          local_port: 777,
+          protocol: 'UDP',
+        }
+        ep2 =  {
+          name: 'endpoint-2',
+          public_port: 888,
+          local_port: 889,
+          protocol: 'UDP',
+          load_balancer_name: 'lb',
+          load_balancer: { protocol: 'http', path: 'mypath' }
+        }
+        subject.update_endpoints(vm_name, csn,  ep1, ep2)
+        vm = subject.get_virtual_machine(vm_name, csn)
+        vm.udp_endpoints.size.must_equal 2
+        vm.udp_endpoints.first[:name].must_equal 'endpoint-1'
+        vm.udp_endpoints.last[:name].must_equal 'endpoint-2'
+      end
+
+      it 'should update existing endpoints of virtual machine.' do
+        ep1 =  {
+          name: 'SSH',
+          public_port: 2222,
+          local_port: 22,
+          protocol: 'TCP',
+        }
+        subject.update_endpoints(vm_name, csn,  ep1)
+        vm = subject.get_virtual_machine(vm_name, csn)
+        vm.tcp_endpoints.size.must_equal 1
+        vm.tcp_endpoints.first[:name].must_equal 'SSH'
+        vm.tcp_endpoints.first[:public_port].must_equal '2222'
+      end
+
+      it 'should delete endpoint of virtual machine.' do
+        subject.delete_endpoint(vm_name, csn,  'SSH')
+        vm = subject.get_virtual_machine(vm_name, csn)
+        vm.tcp_endpoints.size.must_equal 0
+        ep1 =  {
+          name: 'SSH',
+          public_port: 22,
+          local_port: 22,
+          protocol: 'TCP',
+        }
+        subject.update_endpoints(vm_name, csn,  ep1)
+      end
+    end
+
+  end
+end # end

data/test/integration/vm_image/virtual_machine_disk_test.rb

@@ -0,0 +1,37 @@
+#-------------------------------------------------------------------------
+# # 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 "integration/test_helper"
+
+describe Azure::VirtualMachineImageManagement::VirtualMachineDiskManagementService do
+
+  subject {
+    Azure::VirtualMachineImageManagement::VirtualMachineDiskManagementService.new
+  }
+
+  describe "#list_virtual_machine_disks" do
+
+    it "returns a list of virtual machine disks" do
+      vm_disks = subject.list_virtual_machine_disks
+      vm_disks.wont_be_nil
+      vm_disks.must_be_kind_of Array
+      disk = vm_disks.first
+      disk.must_be_kind_of Azure::VirtualMachineImageManagement::VirtualMachineDisk
+      disk.name.wont_be_nil
+      [true, false].must_include disk.attached
+    end
+
+  end
+
+end

data/test/integration/vm_image/virtual_machine_image_test.rb

@@ -0,0 +1,37 @@
+#-------------------------------------------------------------------------
+# # 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 "integration/test_helper"
+
+describe Azure::VirtualMachineImageManagementService do
+
+  subject { Azure::VirtualMachineImageManagementService.new }
+
+  describe "#virtual_machine_images" do
+
+    it "returns a list of virtual machine images" do
+      virtualImages = subject.list_virtual_machine_images
+      virtualImages.wont_be_nil
+      virtualImages.must_be_kind_of Array
+      image = virtualImages.first
+      image.must_be_kind_of Azure::VirtualMachineImageManagement::VirtualMachineImage
+      image.os_type.wont_be_nil
+      image.category.wont_be_nil
+      image.name.wont_be_nil
+      refute_equal virtualImages.length, 0
+    end
+
+  end
+
+end

data/test/integration/vnet/Virtual_Network_Create_test.rb

@@ -0,0 +1,122 @@
+#-------------------------------------------------------------------------
+# 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 'integration/test_helper'
+require 'support/virtual_network_helper'
+
+describe Azure::VirtualNetworkManagement::VirtualNetwork do
+  let(:in_vnet_name) { 'newvnet' }
+  let(:in_affinity_name) { 'integration-test-affinity-group' }
+  let(:in_address_space) { ['172.16.0.0/12'] }
+  let(:invalid_address_space) { ['', ''] }
+  let(:invalid_cidr_address_space) { ['10.0.0.0'] }
+  let(:created_state) { 'Created' }
+  let(:xml_err_msg) { 'XML Schema validation error in network configuration' }
+
+  inputoptions = {
+      subnet: [{ name: 'Subnet-1', ip_address: '172.16.0.0', cidr: 12 }],
+      dns: [{ name: 'DNS', ip_address: '1.2.3.4' }]
+  }
+
+  subject { Azure::VirtualNetworkManagementService.new }
+
+  before do
+    Loggerx.expects(:puts).returns(nil).at_least(0)
+
+    affinity_group_service = Azure::BaseManagementService.new
+    affinity_group_service.create_affinity_group(in_affinity_name,
+                                                 WindowsImageLocation,
+                                                 'Label'
+                                                ) rescue nil
+  end
+
+  describe 'Create virtual network valid params' do
+    it 'Create virtual network with valid params' do
+      subject.set_network_configuration(in_vnet_name,
+                                        in_affinity_name,
+                                        in_address_space,
+                                        inputoptions)
+
+      VirtualNetworkHelper.check_config(subject.list_virtual_networks,
+                                        in_vnet_name,
+                                        in_affinity_name,
+                                        created_state,
+                                        in_address_space,
+                                        inputoptions)
+    end
+  end
+
+  describe 'Create virtual network with invalid subnet' do
+    it 'checks for invalid subnet and throws an exception' do
+      options = {
+          subnet: [{ name: 'Subnet-1', ip_address: '10.0.0.0/8', cidr: 12 }],
+          dns: [{ name: 'demodns', ip_address: '2.3.4.5' }]
+      }
+
+      exception = assert_raises(RuntimeError) do
+        subject.set_network_configuration(in_vnet_name,
+                                          in_affinity_name,
+                                          in_address_space,
+                                          options)
+      end
+      #assert_match(xml_err_msg, exception.message)
+    end
+  end
+
+  describe 'Create virtual network with no address space provided options ' do
+    it 'Create virtual network with invalid options' do
+      exception = assert_raises(RuntimeError) do
+        subject.set_network_configuration(in_vnet_name,
+                                          in_affinity_name,
+                                          invalid_address_space,
+                                          inputoptions)
+      end
+      assert_match(xml_err_msg, exception.message)
+    end
+  end
+
+  describe 'Update virtual network valid params ' do
+    it 'Update virtual network with valid params' do
+      update_address_space = ['10.0.0.0/8']
+      update_options = {
+          subnet: [{ name: 'Subnet-1', ip_address: '10.0.0.0', cidr: 16 }],
+          dns: [{ name: 'DNS', ip_address: '1.2.3.4' }]
+      }
+
+      subject.set_network_configuration(in_vnet_name,
+                                        in_affinity_name,
+                                        update_address_space,
+                                        update_options)
+      VirtualNetworkHelper.check_config(subject.list_virtual_networks,
+                                        in_vnet_name,
+                                        in_affinity_name,
+                                        created_state,
+                                        update_address_space,
+                                        update_options)
+    end
+  end
+
+  describe 'Create virtual network invalid cidr' do
+    it 'Create virtual network with invalid options' do
+      exception = assert_raises(RuntimeError) do
+        subject.set_network_configuration(in_vnet_name,
+                                          in_affinity_name,
+                                          invalid_cidr_address_space,
+                                          inputoptions)
+      end
+      assert_match("Cidr is invalid for IP #{invalid_cidr_address_space[0]}",
+                   exception.message)
+    end
+  end
+end

data/test/integration/vnet/Virtual_Network_list_test.rb

@@ -0,0 +1,53 @@
+#-------------------------------------------------------------------------
+# 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 'integration/test_helper'
+
+describe Azure::VirtualNetworkManagement::VirtualNetwork do
+
+  subject { Azure::VirtualNetworkManagementService.new }
+
+  let(:affinity_group_name) { 'my-affinity-group' }
+  let(:location) { 'West US' }
+  let(:vnet_name) { 'vnet-integration-test' }
+
+  before do
+    Loggerx.expects(:puts).returns(nil).at_least(0)
+    affinity_group_service = Azure::BaseManagementService.new
+    affinity_group_service.create_affinity_group(affinity_group_name,
+                                                 location,
+                                                 'Label Name'
+                                                ) rescue nil
+    address_space = %w(172.16.0.0/12 10.0.0.0/8 192.168.0.0/24)
+    unless subject.list_virtual_networks.map(&:name).include?(vnet_name)
+      subject.set_network_configuration(
+        vnet_name,
+        affinity_group_name,
+        address_space
+      )
+    end
+  end
+
+  describe '#list_virtual_networks' do
+    it 'Gets a list of virtual networks for the current subscription.' do
+      virtual_networks = subject.list_virtual_networks
+      virtual_networks.wont_be_nil
+      virtual_networks.must_be_kind_of Array
+      virtual_networks.first.must_be_kind_of(
+        Azure::VirtualNetworkManagement::VirtualNetwork
+      )
+      assert_operator virtual_networks.size, :>=, 1
+    end
+  end
+end

data/test/support/env.rb

@@ -0,0 +1,19 @@
+#-------------------------------------------------------------------------
+# # 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.
+#--------------------------------------------------------------------------
+def skip_if_emulator
+ if Azure.config.storage_table_host =~ /localhost/
+   skip "This test will only run in Live environment"
+ end
+end