def test_vm_in_range_accepted_after_didnot_run(self):
vm = self._get_vm(0, 0)
stime = int(self.w_end + HOUR)
vm.find('slice/' + Computer.VM_STARTTIME_ELEM).text = str(stime)
cmptr = Computer(self.w_start, self.w_end, '', True)
cmptr._query_etime_from_vm_details = cmptr.get_etime
assert False == cmptr.vm_in_range(vm)
def test_vm_in_range_started_before_still_running(self):
stime = int(self.w_start - HOUR)
etime = 0
vm = self._get_vm(stime, etime)
cmptr = Computer(self.w_start, self.w_end, '', True)
cmptr._query_etime_from_vm_details = cmptr.get_etime
assert True == cmptr.vm_in_range(vm)
def _update_and_assert(self, vm, delta_time_hours):
cmptr = Computer(self.w_start, self.w_end, '', True)
cmptr._query_etime_from_vm_details = cmptr.get_etime
cmptr._update_time_on_vm(vm)
assert delta_time_hours == int(vm.find('time').text)
def test_get_size_from_marketplace(self):
cmptr = Computer(self.w_start, self.w_end, '', True)
cmptr._query_etime_from_vm_details = cmptr.get_etime
cmptr._get_url = Mock(return_value=IMAGE_MANIFEST)
assert DISK_SIZE == cmptr.get_size_from_marketplace('http://foo.bar/baz')
assert DISK_SIZE == cmptr.marketplaceSizeCache['http://foo.bar/baz']
