oneacct-export 0.3.0 → 0.4.2

data/oneacct-export.gemspec

@@ -20,14 +20,15 @@ Gem::Specification.new do |spec|
   spec.required_ruby_version = '>= 2.0.0'
 
   spec.add_development_dependency 'bundler', '~> 1.6'
-  spec.add_development_dependency 'rake', '~> 10.0'
-  spec.add_development_dependency 'rspec', '~> 3.0.0'
+  spec.add_development_dependency 'rake', '~> 10.4'
+  spec.add_development_dependency 'rspec', '~> 3.3'
   spec.add_development_dependency 'fuubar', '~> 2.0.0'
-  spec.add_development_dependency 'simplecov', '~> 0.9.0'
+  spec.add_development_dependency 'simplecov', '~> 0.10'
   spec.add_development_dependency 'rubygems-tasks', '~> 0.2.4'
 
   spec.add_runtime_dependency 'opennebula', '~> 4.6.0'
-  spec.add_runtime_dependency 'syslogger', '~> 1.6.0'
-  spec.add_runtime_dependency 'sidekiq', '~> 3.2.6'
+  spec.add_runtime_dependency 'syslogger', '~> 1.6'
+  spec.add_runtime_dependency 'sidekiq', '~> 3.4'
   spec.add_runtime_dependency 'settingslogic', '~> 2.0.9'
+  spec.add_runtime_dependency 'chronic_duration', '~> 0.10'
 end

data/spec/data_validators/apel_data_validator_spec.rb

@@ -50,6 +50,11 @@ module DataValidators
         data['user_dn'] = '/MY=STuPID/CN=DN/CN=HERE'
         data['image_name'] = 'https://appdb.egi.eu/store/vo/image/blablablabla/'
 
+        data['cloud_compute_service'] = 'CloudComputeServiceValue'
+
+        data['benchmark_type'] = 'b_type_999'
+        data['benchmark_value'] = '132.68'
+
         data
       end
 
@@ -74,6 +79,7 @@ module DataValidators
         valid_data['network_inbound'] = 3
         valid_data['network_outbound'] = 5
         valid_data['memory'] = '1736960'
+        valid_data['number_of_public_ips'] = 0
         history = []
         rec = {}
         rec['start_time'] = '1383741169'
@@ -101,6 +107,11 @@ module DataValidators
         valid_data['suspend'] = 425
         valid_data['disk_size'] = 52608
 
+        valid_data['cloud_compute_service'] = 'CloudComputeServiceValue'
+
+        valid_data['benchmark_type'] = 'b_type_999'
+        valid_data['benchmark_value'] = '132.68'
+
         valid_data
       end
 
@@ -492,6 +503,67 @@ module DataValidators
           expect(subject.validate_data(data)).to eq(valid_data)
         end
       end
+
+      context 'without cloud compute service' do
+        before :example do
+          data['cloud_compute_service'] = nil
+          valid_data['cloud_compute_service'] = 'NULL'
+        end
+
+        it 'replaces cloud compute service with "NULL"' do
+          expect(subject.validate_data(data)).to eq(valid_data)
+        end
+      end
+
+      context 'without benchmark_type' do
+        before :example do
+          data['benchmark_type'] = nil
+          valid_data['benchmark_type'] = 'NULL'
+          valid_data['benchmark_value'] = 'NULL'
+        end
+
+        it 'replaced benchmark type and benchmark value with "NULL"' do
+          expect(subject.validate_data(data)).to eq(valid_data)
+        end
+      end
+
+      context 'without benchmark_value' do
+        before :example do
+          data['benchmark_value'] = nil
+          valid_data['benchmark_type'] = 'NULL'
+          valid_data['benchmark_value'] = 'NULL'
+        end
+
+        it 'replaced benchmark type and benchmark value with "NULL"' do
+          expect(subject.validate_data(data)).to eq(valid_data)
+        end
+      end
+
+      context 'without benchmark_type and benchmark_value' do
+        before :example do
+          data['benchmark_type'] = nil
+          data['benchmark_value'] = nil
+          valid_data['benchmark_type'] = 'NULL'
+          valid_data['benchmark_value'] = 'NULL'
+        end
+
+        it 'replaced benchmark type and benchmark value with "NULL"' do
+          expect(subject.validate_data(data)).to eq(valid_data)
+        end
+      end
+
+      context 'with benchmark_value that is not a number' do
+        before :example do
+          data['benchmark_value'] = 'string'
+          valid_data['benchmark_type'] = 'NULL'
+          valid_data['benchmark_value'] = 'NULL'
+        end
+
+        it 'replaced benchmark type and benchmark value with "NULL"' do
+          expect(subject.validate_data(data)).to eq(valid_data)
+        end
+      end
+
     end
   end
 end

data/spec/one_data_accessor_spec.rb

@@ -155,42 +155,44 @@ describe OneDataAccessor do
     before :example do
       allow(vm_pool).to receive(:info) { 'valid_rc' }
       allow(vm_pool).to receive(:to) { vm_pool }
+      allow(vm_pool).to receive(:count) { 100 }
+      allow(vm_pool).to receive(:entries) { [last] }
       allow(OpenNebula::VirtualMachinePool).to receive(:new) { vm_pool }
     end
 
     let(:vm_pool) { double('vm_pool') }
+    let(:last) { double('last') }
 
     context 'without compatibility mode' do
-      context 'with valid batch number' do
-        it 'requests vms with correct range' do
-          expect(vm_pool).to receive(:info).with(OpenNebula::Pool::INFO_ALL, 0, 99, OpenNebula::VirtualMachinePool::INFO_ALL_VM)
-          subject.load_vm_pool(0)
+      it 'requests vms with correct range' do
+        expect(vm_pool).to receive(:info).with(OpenNebula::Pool::INFO_ALL, 0, -100, OpenNebula::VirtualMachinePool::INFO_ALL_VM)
+        allow(last).to receive(:id) {99}
+        subject.load_vm_pool
 
-          expect(vm_pool).to receive(:info).with(OpenNebula::Pool::INFO_ALL, 100, 199, OpenNebula::VirtualMachinePool::INFO_ALL_VM)
-          subject.load_vm_pool(1)
+        expect(vm_pool).to receive(:info).with(OpenNebula::Pool::INFO_ALL, 100, -100, OpenNebula::VirtualMachinePool::INFO_ALL_VM)
+        allow(last).to receive(:id) {199}
+        subject.load_vm_pool
 
-          expect(vm_pool).to receive(:info).with(OpenNebula::Pool::INFO_ALL, 300, 399, OpenNebula::VirtualMachinePool::INFO_ALL_VM)
-          subject.load_vm_pool(3)
+        expect(vm_pool).to receive(:info).with(OpenNebula::Pool::INFO_ALL, 200, -100, OpenNebula::VirtualMachinePool::INFO_ALL_VM)
+        allow(last).to receive(:id) {299}
+        subject.load_vm_pool
 
-          expect(vm_pool).to receive(:info).with(OpenNebula::Pool::INFO_ALL, 500, 599, OpenNebula::VirtualMachinePool::INFO_ALL_VM)
-          subject.load_vm_pool(5)
+        expect(vm_pool).to receive(:info).with(OpenNebula::Pool::INFO_ALL, 300, -100, OpenNebula::VirtualMachinePool::INFO_ALL_VM)
+        allow(last).to receive(:id) {399}
+        subject.load_vm_pool
 
-          expect(vm_pool).to receive(:info).with(OpenNebula::Pool::INFO_ALL, 1000, 1099, OpenNebula::VirtualMachinePool::INFO_ALL_VM)
-          subject.load_vm_pool(10)
+        expect(vm_pool).to receive(:info).with(OpenNebula::Pool::INFO_ALL, 400, -100, OpenNebula::VirtualMachinePool::INFO_ALL_VM)
+        allow(last).to receive(:id) {499}
+        subject.load_vm_pool
 
-          expect(vm_pool).to receive(:info).with(OpenNebula::Pool::INFO_ALL, 1200, 1299, OpenNebula::VirtualMachinePool::INFO_ALL_VM)
-          subject.load_vm_pool(12)
-        end
-
-        it 'returns obtained vm pool' do
-          expect(subject.load_vm_pool(0)).to eq(vm_pool)
-        end
+        expect(vm_pool).to receive(:info).with(OpenNebula::Pool::INFO_ALL, 500, -100, OpenNebula::VirtualMachinePool::INFO_ALL_VM)
+        allow(last).to receive(:id) {599}
+        subject.load_vm_pool
       end
 
-      context 'with invalid batch number' do
-        it 'fails with ArgumentError' do
-          expect { subject.load_vm_pool('invalid_number') }.to raise_error(ArgumentError)
-        end
+      it 'returns obtained vm pool' do
+        allow(last).to receive(:id) {99}
+        expect(subject.load_vm_pool).to eq(vm_pool)
       end
     end
 
@@ -199,17 +201,19 @@ describe OneDataAccessor do
         Settings.output['num_of_vms_per_file'] = 3
         allow(OpenNebula::VirtualMachinePool).to receive(:new) { compatibility_vm_pool }
         allow(compatibility_vm_pool).to receive(:info) { 'valid_rc' }
+        allow(last).to receive(:id).and_return(2, 5, 8, 10)
       end
 
       let(:one_data_accessor) { OneDataAccessor.new(true, Logger.new('/dev/null')) }
-      let(:compatibility_vm_pool) { [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10] }
+      let(:compatibility_vm_pool) { [0, 1, last, 3, 4, last, 6, 7, last, 9, last] }
+      let(:last) { double('last') }
 
       it 'returns correct parts of vm pool' do
         expect(compatibility_vm_pool).to receive(:info).with(OpenNebula::Pool::INFO_ALL, -1, -1, OpenNebula::VirtualMachinePool::INFO_ALL_VM)
-        expect(subject.load_vm_pool(0)).to eq([0, 1, 2])
-        expect(subject.load_vm_pool(1)).to eq([3, 4, 5])
-        expect(subject.load_vm_pool(2)).to eq([6, 7, 8])
-        expect(subject.load_vm_pool(3)).to eq([9, 10])
+        expect(subject.load_vm_pool).to eq([0, 1, last])
+        expect(subject.load_vm_pool).to eq([3, 4, last])
+        expect(subject.load_vm_pool).to eq([6, 7, last])
+        expect(subject.load_vm_pool).to eq([9, last])
       end
     end
   end
@@ -333,23 +337,12 @@ describe OneDataAccessor do
     let(:vm2) { { 'ID' => '2' } }
     let(:vm3) { { 'ID' => '3' } }
     let(:vm_pool) { [vm1, vm2, vm3] }
-    let(:batch_number) { 5 }
-
-    context 'is called with some batch number' do
-      before :example do
-        expect(subject).to receive(:load_vm_pool).with(5) { vm_pool }
-      end
-
-      it 'calls load_vm_pool with that batch number' do
-        subject.vms(batch_number, nil, nil)
-      end
-    end
 
     context 'when vm pool is empty' do
       let(:vm_pool) { [] }
 
       it 'returns nil' do
-        expect(subject.vms(batch_number, nil, nil)).to be_nil
+        expect(subject.vms(nil, nil)).to be_nil
       end
     end
 
@@ -360,14 +353,14 @@ describe OneDataAccessor do
         expect(subject).to receive(:want?).with(vm3, nil, nil).once
       end
       it 'calls want?' do
-        subject.vms(batch_number, nil, nil)
+        subject.vms(nil, nil)
       end
     end
 
     context 'for every vm obtained from vm pool' do
       context 'with ID attribute' do
         it 'returns ID of those vms' do
-          expect(subject.vms(batch_number, nil, nil)).to eq([1, 2, 3])
+          expect(subject.vms(nil, nil)).to eq([1, 2, 3])
         end
       end
 
@@ -376,7 +369,7 @@ describe OneDataAccessor do
         let(:vm3) { {} }
 
         it 'skips vms without ID attribute and returns only thouse with it' do
-          expect(subject.vms(batch_number, nil, nil)).to eq([2])
+          expect(subject.vms(nil, nil)).to eq([2])
         end
       end
     end
@@ -438,4 +431,82 @@ describe OneDataAccessor do
       end
     end
   end
+
+  describe '.benchmark_map' do
+    before :example do
+      allow(OpenNebula::HostPool).to receive(:new) { host_pool }
+      allow(OpenNebula::Cluster).to receive(:new) { searched_cluster }
+      expect(host_pool).to receive(:info)
+    end
+
+    let(:host_pool) {
+      file = File.read("#{GEM_DIR}/mock/#{host_filename}")
+      xml = OpenNebula::XMLElement.new(OpenNebula::XMLElement.build_xml(file, 'HOST'))
+      [ xml ]
+    }
+
+    let(:searched_cluster) {
+      file = File.read("#{GEM_DIR}/mock/#{cluster_filename}")
+      OpenNebula::XMLElement.new(OpenNebula::XMLElement.build_xml(file, 'CLUSTER'))
+    }
+
+    context 'with correct data on a host with one benchmark value' do
+      let(:host_filename) { 'one_data_accessor_host_01.xml' }
+      let(:expected) { { '1' => { :benchmark_type => 'bench_type_1',
+                                  :mixins => { 'mixin1' => '36.9' } } } }
+
+      it 'creates correct benchmark_map' do
+        expect(subject.benchmark_map).to eq(expected)
+      end
+    end
+
+    context 'with correct data on a host with two benchmark values' do
+      let(:host_filename) { 'one_data_accessor_host_02.xml' }
+      let(:expected) { { '2' => { :benchmark_type => 'bench_type_2',
+                                  :mixins => { 'mixin1' => '788.6', 'mixin2' => '123.123' } } } }
+
+      it 'creates correct benchmark_map' do
+        expect(subject.benchmark_map).to eq(expected)
+      end
+    end
+
+    context 'with no data on a host and correct data on host\'s cluster' do
+      before :example do
+        expect(searched_cluster).to receive(:info)
+      end
+
+      let(:host_filename) { 'one_data_accessor_host_03.xml' }
+      let(:cluster_filename) { 'one_data_accessor_cluster_01.xml' }
+      let(:expected) { { '3' => { :benchmark_type => 'bench_type_cluster_1',
+                                  :mixins => { 'mixin1' => '21.7' } } } }
+
+      it 'creates correct benchmark_map' do
+        expect(subject.benchmark_map).to eq(expected)
+      end
+    end
+
+    context 'with no data on a host and with no data on host\'s cluster' do
+      before :example do
+        expect(searched_cluster).to receive(:info)
+      end
+
+      let(:host_filename) { 'one_data_accessor_host_04.xml' }
+      let(:cluster_filename) { 'one_data_accessor_cluster_02.xml' }
+      let(:expected) { { '4' => {} } }
+
+      it 'creates correct benchmark_map' do
+        expect(subject.benchmark_map).to eq(expected)
+      end
+    end
+
+    context 'with no data on a host and without host\'s cluster' do
+      let(:host_filename) { 'one_data_accessor_host_05.xml' }
+      let(:expected) { { '5' => {} } }
+      let(:searched_cluster) { nil }
+
+      it 'creates correct benchmark_map' do
+        expect(subject.benchmark_map).to eq(expected)
+      end
+    end
+  end
 end

data/spec/one_worker_spec.rb

@@ -13,9 +13,10 @@ describe OneWorker do
         Settings.output.apel['endpoint'] = 'machine.hogwarts.co.uk'
         Settings.output.apel['site_name'] = 'Hogwarts'
         Settings.output.apel['cloud_type'] = 'OpenNebula'
+        Settings.output.apel['cloud_compute_service'] = 'CloudComputeServiceValue'
       end
 
-      let(:output_type_specific_data) { {'endpoint' => 'machine.hogwarts.co.uk', 'site_name' => 'Hogwarts', 'cloud_type' => 'OpenNebula'} }
+      let(:output_type_specific_data) { {'endpoint' => 'machine.hogwarts.co.uk', 'site_name' => 'Hogwarts', 'cloud_type' => 'OpenNebula', 'cloud_compute_service' => 'CloudComputeServiceValue'} }
 
       it 'returns data specific for apel output type in form of hash' do
         expect(subject.output_type_specific_data).to eq(output_type_specific_data)
@@ -53,6 +54,13 @@ describe OneWorker do
     end
   end
 
+  describe '.create_cluster_map' do
+    it 'returns cluster map' do
+      expect(subject).to receive(:create_map).with(OpenNebula::ClusterPool, anything, anything) { 'map' }
+      expect(subject.create_cluster_map(oda)).to eq('map')
+    end
+  end
+
   describe '.create_map' do
     let(:pool_type) { double('pool_type') }
     let(:mapping) { double('mapping') }
@@ -136,8 +144,9 @@ describe OneWorker do
     before :example do
       Settings.output['output_type'] = 'apel-0.2'
       Settings.output.apel['endpoint'] = 'machine.hogwarts.co.uk'
-      Settings.output.apel['site_name'] = 'Hogwarts'
+      Settings.output.apel['site_name'] = 'site-name-from-config'
       Settings.output.apel['cloud_type'] = 'OpenNebula'
+      Settings.output.apel['cloud_compute_service'] = nil
 
       allow(vm).to receive(:state_str) { 'DONE' }
     end
@@ -151,8 +160,9 @@ describe OneWorker do
       data = {}
 
       data['endpoint'] = 'machine.hogwarts.co.uk'
-      data['site_name'] = 'Hogwarts'
+      data['site_name'] = 'site-name-from-cluster'
       data['cloud_type'] = 'OpenNebula'
+      data['cloud_compute_service'] = nil
 
       data['vm_uuid'] = '36551'
       data['start_time'] = '1383741160'
@@ -168,6 +178,7 @@ describe OneWorker do
       data['network_inbound'] = '43557888'
       data['network_outbound'] = '376832'
       data['memory'] = '1736960'
+      data['number_of_public_ips'] = 1
       history = []
       rec = {}
       rec['start_time'] = '1383741169'
@@ -191,17 +202,24 @@ describe OneWorker do
       data['user_dn'] = '/Dn=FrOm/CN=DN/CN=TeMpLaTe'
       data['image_name'] = 'https://appdb.egi.eu/store/vo/image/image_name_from_VMCATCHER_EVENT_AD_MPURI_tag/'
 
+      data['benchmark_type'] = nil
+      data['benchmark_value'] = nil
+
+      data['oneacct_export_version'] = ::OneacctExporter::VERSION
+
       data
     end
 
     let(:user_map) { {'120' => '/Dn=FrOm/CN=DN/CN=MaP'} }
     let(:image_map) { {'31' => 'image_name_from_map'} }
+    let(:cluster_map) { {'100' => 'site-name-from-cluster'}}
+    let(:benchmark_map) { {'123' => nil}  }
 
     context 'with apel specific data' do
       let(:filename) { 'one_worker_vm_dn01.xml' }
 
       it 'returns correct vm data with apel specific data' do
-        expect(subject.process_vm(vm, user_map, image_map)).to eq(data)
+        expect(subject.process_vm(vm, user_map, image_map, cluster_map, benchmark_map)).to eq(data)
       end
     end
 
@@ -219,14 +237,14 @@ describe OneWorker do
         data['host'] = 'on_localhost'
 
         data.delete 'endpoint'
-        data.delete 'site_name'
         data.delete 'cloud_type'
+        data.delete 'cloud_compute_service'
       end
 
       let(:filename) { 'one_worker_vm_dn01.xml' }
 
       it 'returns correct vm data with pbs specific data' do
-        expect(subject.process_vm(vm, user_map, image_map)).to eq(data)
+        expect(subject.process_vm(vm, user_map, image_map, cluster_map, benchmark_map)).to eq(data)
       end
     end
 
@@ -234,7 +252,7 @@ describe OneWorker do
       let(:filename) { 'one_worker_vm_dn01.xml' }
 
       it 'returns correct vm data with user\'s dn from template' do
-        expect(subject.process_vm(vm, user_map, image_map)).to eq(data)
+        expect(subject.process_vm(vm, user_map, image_map, cluster_map, benchmark_map)).to eq(data)
       end
     end
 
@@ -246,7 +264,7 @@ describe OneWorker do
       end
 
       it 'returns correct vm data with user\'s dn from map' do
-        expect(subject.process_vm(vm, user_map, image_map)).to eq(data)
+        expect(subject.process_vm(vm, user_map, image_map, cluster_map, benchmark_map)).to eq(data)
       end
     end
 
@@ -254,7 +272,7 @@ describe OneWorker do
       let(:filename) { 'one_worker_vm_image_name01.xml' }
 
       it 'returns correct vm data with image name from VMCATCHER_EVENT_AD_MPURI tag' do
-        expect(subject.process_vm(vm, user_map, image_map)).to eq(data)
+        expect(subject.process_vm(vm, user_map, image_map, cluster_map, benchmark_map)).to eq(data)
       end
     end
 
@@ -266,7 +284,7 @@ describe OneWorker do
       end
 
       it 'returns correct vm data with image name from map' do
-        expect(subject.process_vm(vm, user_map, image_map)).to eq(data)
+        expect(subject.process_vm(vm, user_map, image_map, cluster_map, benchmark_map)).to eq(data)
       end
     end
 
@@ -279,7 +297,7 @@ describe OneWorker do
       end
 
       it 'returns correct vm data with image name from USER_TEMPLATE/OCCI_COMPUTE_MIXINS tag' do
-        expect(subject.process_vm(vm, user_map, image_map)).to eq(data)
+        expect(subject.process_vm(vm, user_map, image_map, cluster_map, benchmark_map)).to eq(data)
       end
     end
 
@@ -292,7 +310,7 @@ describe OneWorker do
       end
 
       it 'returns correct vm data with image name from USER_TEMPLATE/OCCI_MIXIN tag' do
-        expect(subject.process_vm(vm, user_map, image_map)).to eq(data)
+        expect(subject.process_vm(vm, user_map, image_map, cluster_map, benchmark_map)).to eq(data)
       end
     end
 
@@ -305,7 +323,7 @@ describe OneWorker do
       end
 
       it 'returns correct vm data with image name from TEMPLATE/OCCI_MIXIN tag' do
-        expect(subject.process_vm(vm, user_map, image_map)).to eq(data)
+        expect(subject.process_vm(vm, user_map, image_map, cluster_map, benchmark_map)).to eq(data)
       end
     end
 
@@ -317,7 +335,47 @@ describe OneWorker do
       end
 
       it 'returns correct vm data with image name as image id' do
-        expect(subject.process_vm(vm, user_map, image_map)).to eq(data)
+        expect(subject.process_vm(vm, user_map, image_map, cluster_map, benchmark_map)).to eq(data)
+      end
+    end
+
+    context 'without site-name on cluster (APEL)' do
+      let(:filename) { 'one_worker_vm_dn01.xml' }
+      let(:cluster_map) { {} }
+
+      before :example do
+        data['site_name'] = 'site-name-from-config'
+      end
+
+      it 'returns correct vm data with site-name from configuration file' do
+        expect(subject.process_vm(vm, user_map, image_map, cluster_map, benchmark_map)).to eq(data)
+      end
+    end
+
+    context 'without site-name on cluster (NON APEL)' do
+      let(:filename) { 'one_worker_vm_dn01.xml' }
+      let(:cluster_map) { {} }
+
+      before :example do
+        Settings.output['output_type'] = 'pbs-0.1'
+        Settings.output.pbs['realm'] = 'REALM'
+        Settings.output.pbs['queue'] = 'cloud'
+        Settings.output.pbs['scratch_type'] = 'local'
+        Settings.output.pbs['host_identifier'] = 'on_localhost'
+
+        data['realm'] = 'REALM'
+        data['pbs_queue'] = 'cloud'
+        data['scratch_type'] = 'local'
+        data['host'] = 'on_localhost'
+
+        data.delete 'endpoint'
+        data.delete 'cloud_type'
+        data.delete 'cloud_compute_service'
+        data.delete 'site_name'
+      end
+
+      it 'returns correct vm data without any site-name' do
+        expect(subject.process_vm(vm, user_map, image_map, cluster_map, benchmark_map)).to eq(data)
       end
     end
   end
@@ -402,19 +460,46 @@ describe OneWorker do
     end
   end
 
+  describe ".number_of_public_ips" do
+    let(:vm) do
+      xml = File.read("#{GEM_DIR}/mock/#{filename}")
+      OpenNebula::XMLElement.new(OpenNebula::XMLElement.build_xml(xml, 'VM'))
+    end
+
+    context "with multiple NICs (multiple IPs with duplicates) " do
+      let(:filename) { 'one_worker_vm_number_of_public_ips_01.xml' }
+      it "returns the correct number of public IPs" do
+        expect(subject.number_of_public_ips(vm)).to eq 8
+      end
+    end
+    context "with no NICs" do
+      let(:filename) { 'one_worker_vm_number_of_public_ips_02.xml' }
+      it "returns 0, the correct number of IPs" do
+        expect(subject.number_of_public_ips(vm)).to eq 0
+      end
+    end
+    context "with single NIC (multiple IPs with duplicates)" do
+      let(:filename) { 'one_worker_vm_number_of_public_ips_03.xml' }
+      it "returns the correct number of public IPs" do
+        expect(subject.number_of_public_ips(vm)).to eq 8
+      end
+    end
+  end
+
   describe '.perform' do
     before :example do
       Settings.output['output_type'] = 'unknown'
       allow(OneDataAccessor).to receive(:new) { oda }
       allow(subject).to receive(:create_user_map) { 'user_map' }
       allow(subject).to receive(:create_image_map) { 'image_map' }
-      allow(subject).to receive(:load_vm).and_return(:default)
+      allow(subject).to receive(:create_cluster_map) { 'cluster_map' }
+      allow(oda).to receive(:benchmark_map) { 'benchmark_map' }
       allow(subject).to receive(:load_vm).with('10', oda).and_return('10')
       allow(subject).to receive(:load_vm).with('20', oda).and_return('20')
       allow(subject).to receive(:load_vm).with('30', oda).and_return('30')
-      allow(subject).to receive(:process_vm).with('10', anything, anything).and_return('data_vm1')
-      allow(subject).to receive(:process_vm).with('20', anything, anything).and_return('data_vm2')
-      allow(subject).to receive(:process_vm).with('30', anything, anything).and_return('data_vm3')
+      allow(subject).to receive(:process_vm).with('10', anything, anything, anything, anything).and_return('data_vm1')
+      allow(subject).to receive(:process_vm).with('20', anything, anything, anything, anything).and_return('data_vm2')
+      allow(subject).to receive(:process_vm).with('30', anything, anything, anything, anything).and_return('data_vm3')
     end
 
     let(:vms) { '10|20|30' }
@@ -498,4 +583,49 @@ describe OneWorker do
       end
     end
   end
+
+  describe '.search_benchmark' do
+    let(:vm) do
+      xml = File.read("#{GEM_DIR}/mock/#{filename}")
+      OpenNebula::XMLElement.new(OpenNebula::XMLElement.build_xml(xml, 'VM'))
+    end
+
+    let(:benchmark_map) do
+      values1 = { :benchmark_type => 'bench_type_1', :mixins => { 'mixin1' => '34.12' } }
+      values2 = { :benchmark_type => 'bench_type_2', :mixins => { 'mixin2' => '123.2', 'mixin3' => '129.6' } }
+      values3 = { }
+
+      benchmark_map = { '19' => values1, '11' => values2, '23' => values3 }
+      benchmark_map
+    end
+
+    context 'with empty benchmark_map' do
+      let(:filename) { 'one_worker_vm_search_benchmark_01.xml' }
+      let(:benchmark_map) { {} }
+      let(:expected) { { :benchmark_type => nil, :benchmark_value => nil } }
+
+      it 'returns array with two nil items' do
+        expect(subject.search_benchmark(vm, benchmark_map)).to eq(expected)
+      end
+    end
+
+    context 'with no data for the virtual machine in benchmark_map' do
+      let(:filename) { 'one_worker_vm_search_benchmark_02.xml' }
+      let(:expected) { { :benchmark_type => nil, :benchmark_value => nil } }
+
+      it 'returns array with two nil items' do
+        expect(subject.search_benchmark(vm, benchmark_map)).to eq(expected)
+      end
+    end
+
+    context 'with correct data in vm and benchmark_map' do
+      let(:filename) { 'one_worker_vm_search_benchmark_01.xml' }
+      let(:expected) { { :benchmark_type => 'bench_type_2', :benchmark_value => '129.6' } }
+
+      it 'returns correct benchmark type and value' do
+        expect(subject.search_benchmark(vm, benchmark_map)).to eq(expected)
+      end
+    end
+  end
+
 end