class TestLinuxCollectors(): def setup(self): from zabby.hostos.linux import Linux self.linux = Linux() def test_disk_device_collector_collection(self): self.linux._disk_device_stats_collector._collect() devices = self.linux.disk_device_names() now = int(time.time()) (stats, timestamp) = self.linux.disk_device_stats_shifted( devices.pop(), 60, now) assert_is_instance(stats, DiskDeviceStats) def test_cpu_times_collector_collection(self): self.linux._cpu_times_collector._collect() cpu_id = list(range(self.linux.cpu_count())).pop() times = self.linux._cpu_times_collector.get_times(cpu_id, 60) assert_is_instance(times, CpuTimes)
class TestLinuxCollectors(): def setup(self): from zabby.hostos.linux import Linux self.linux = Linux() def test_disk_device_collector_collection(self): self.linux._disk_device_stats_collector._collect() devices = self.linux.disk_device_names() now = int(time.time()) (stats, timestamp) = self.linux.disk_device_stats_shifted(devices.pop(), 60, now) assert_is_instance(stats, DiskDeviceStats) def test_cpu_times_collector_collection(self): self.linux._cpu_times_collector._collect() cpu_id = list(range(self.linux.cpu_count())).pop() times = self.linux._cpu_times_collector.get_times(cpu_id, 60) assert_is_instance(times, CpuTimes)
class TestLinux(): def setup(self): from zabby.hostos.linux import Linux self.linux = Linux() def test_linux_is_detected_correctly(self): detected_os = detect_host_os() assert_is_instance(detected_os, self.linux.__class__) def test_fs_size_returns_tuple_of_integers_one_less_than_other(self): free, total = self.linux.fs_size(PRESENT_FILESYSTEM) assert_less_equal(free, total) def test_fs_inodes_returns_tuple_of_integers_one_less_than_other(self): free, total = self.linux.fs_inodes(PRESENT_FILESYSTEM) assert_less_equal(free, total) def test_net_interface_names_returns_set_of_strings(self): interface_names = self.linux.net_interface_names() assert_is_instance(interface_names, set) for interface_name in interface_names: assert_is_instance(interface_name, string_types) assert_in(PRESENT_INTERFACE, interface_names) def test_net_interface_info_returns_NetworkInterfaceInfo(self): interface_names = self.linux.net_interface_names() interface_info = self.linux.net_interface_info(interface_names.pop()) assert_is_instance(interface_info, NetworkInterfaceInfo) for key, value in interface_info._asdict().items(): assert_is_instance(value, integer_types) NET_INFO_JOINED_NAME = [ "Inter-| Receive | Transmit", " face |bytes packets errs drop fifo frame compressed multicast|bytes packets errs drop fifo colls carrier compressed", " lo:9091406708 32855976 0 0 0 0 0 0 9091406708 32855976 0 0 0 0 0 0", ] @patch('zabby.hostos.linux.lines_from_file') def test_net_interface_infos_works_with_joined_names(self, mock_lines): mock_lines.return_value = self.NET_INFO_JOINED_NAME assert_in(PRESENT_INTERFACE, self.linux.net_interface_names()) def test_process_infos_returns_iterable_of_ProcessInfo(self): process_infos = list(self.linux.process_infos()) for process_info in process_infos: assert_is_instance(process_info, ProcessInfo) def test_process_infos_contains_processes_run_by_root(self): process_infos = [ proc_info for proc_info in self.linux.process_infos() if proc_info.uid == 0 ] # at least init should be here assert_less(0, len(process_infos)) @patch('zabby.hostos.linux.os.listdir') def test_process_infos_skips_expired_pids(self, mock_listdir): mock_listdir.return_value = ['0', '1'] assert_equal(1, len(list(self.linux.process_infos()))) def test_uid_returns_integer(self): uid = self.linux.uid('root') assert_is_instance(uid, integer_types) def test_uid_raises_exception_on_invalid_username(self): assert_raises(OperatingSystemError, self.linux.uid, '') def test_memory_returns_dict_of_numbers(self): d = self.linux.memory() assert_is_instance(d, dict) for key, value in d.items(): assert_is_instance(value, (integer_types, float)) def test_memory_dict_contains_all_available_memory_types(self): d = self.linux.memory() for memory_type in self.linux.AVAILABLE_MEMORY_TYPES: assert_in(memory_type, d) def test_disk_device_names_returns_set_of_strings(self): device_names = self.linux.disk_device_names() assert_is_instance(device_names, set) for device_name in device_names: assert_is_instance(device_name, string_types) def test_disk_device_stats_returns_DiskDeviceStats(self): device_names = self.linux.disk_device_names() disk_device_stats = self.linux.disk_device_stats(device_names.pop()) assert_is_instance(disk_device_stats, DiskDeviceStats) for key, value in disk_device_stats._asdict().items(): assert_is_instance(value, integer_types) def test_cpu_count_returns_integer(self): cpu_count = self.linux.cpu_count() assert_is_instance(cpu_count, integer_types) def test_cpu_times(self): cpu_id = list(range(self.linux.cpu_count())).pop() cpu_times = self.linux.cpu_times(cpu_id) assert_is_instance(cpu_times, CpuTimes) def test_hostname(self): hostname = self.linux.hostname('host') assert_is_instance(hostname, string_types) def test_uname(self): uname = self.linux.uname() assert_is_instance(uname, collections.Iterable) def test_uptime(self): uptime = self.linux.uptime() assert_is_instance(uptime, integer_types) def test_max_number_of_running_processes(self): maxproc = self.linux.max_number_of_running_processes() assert_is_instance(maxproc, integer_types) def test_system_load(self): system_load = self.linux.system_load() assert_is_instance(system_load, SystemLoad) def test_swap_size(self): free, total = self.linux.swap_size('all') assert_less_equal(free, total) def test_swap_info(self): swap_info = self.linux.swap_info() assert_is_instance(swap_info, SwapInfo) def test_swap_device_names(self): swap_devices = self.linux.swap_device_names() disk_devices = self.linux.disk_device_names() for swap_device in swap_devices: assert_in(swap_device, disk_devices) @patch('zabby.hostos.linux.lines_from_file') def test_swap_device_names_skips_files(self, mock_lists): swap_file_path = '/mnt/swap_file' mock_lists.return_value = [[ 'Filename', 'Type', 'Size', 'Used', 'Priority' ], ['/dev/dm-0', 'partition', '10485756', '311756', '-1'], [swap_file_path, 'file', '524284', '0', '-2']] swap_devices = self.linux.swap_device_names() assert_not_in(swap_file_path.split('/')[-1], swap_devices)
class TestLinux(): def setup(self): from zabby.hostos.linux import Linux self.linux = Linux() def test_linux_is_detected_correctly(self): detected_os = detect_host_os() assert_is_instance(detected_os, self.linux.__class__) def test_fs_size_returns_tuple_of_integers_one_less_than_other(self): free, total = self.linux.fs_size(PRESENT_FILESYSTEM) assert_less_equal(free, total) def test_fs_inodes_returns_tuple_of_integers_one_less_than_other(self): free, total = self.linux.fs_inodes(PRESENT_FILESYSTEM) assert_less_equal(free, total) def test_net_interface_names_returns_set_of_strings(self): interface_names = self.linux.net_interface_names() assert_is_instance(interface_names, set) for interface_name in interface_names: assert_is_instance(interface_name, string_types) assert_in(PRESENT_INTERFACE, interface_names) def test_net_interface_info_returns_NetworkInterfaceInfo(self): interface_names = self.linux.net_interface_names() interface_info = self.linux.net_interface_info(interface_names.pop()) assert_is_instance(interface_info, NetworkInterfaceInfo) for key, value in interface_info._asdict().items(): assert_is_instance(value, integer_types) NET_INFO_JOINED_NAME = [ "Inter-| Receive | Transmit", " face |bytes packets errs drop fifo frame compressed multicast|bytes packets errs drop fifo colls carrier compressed", " lo:9091406708 32855976 0 0 0 0 0 0 9091406708 32855976 0 0 0 0 0 0", ] @patch('zabby.hostos.linux.lines_from_file') def test_net_interface_infos_works_with_joined_names(self, mock_lines): mock_lines.return_value = self.NET_INFO_JOINED_NAME assert_in(PRESENT_INTERFACE, self.linux.net_interface_names()) def test_process_infos_returns_iterable_of_ProcessInfo(self): process_infos = list(self.linux.process_infos()) for process_info in process_infos: assert_is_instance(process_info, ProcessInfo) def test_process_infos_contains_processes_run_by_root(self): process_infos = [proc_info for proc_info in self.linux.process_infos() if proc_info.uid == 0] # at least init should be here assert_less(0, len(process_infos)) @patch('zabby.hostos.linux.os.listdir') def test_process_infos_skips_expired_pids(self, mock_listdir): mock_listdir.return_value = ['0', '1'] assert_equal(1, len(list(self.linux.process_infos()))) def test_uid_returns_integer(self): uid = self.linux.uid('root') assert_is_instance(uid, integer_types) def test_uid_raises_exception_on_invalid_username(self): assert_raises(OperatingSystemError, self.linux.uid, '') def test_memory_returns_dict_of_numbers(self): d = self.linux.memory() assert_is_instance(d, dict) for key, value in d.items(): assert_is_instance(value, (integer_types, float)) def test_memory_dict_contains_all_available_memory_types(self): d = self.linux.memory() for memory_type in self.linux.AVAILABLE_MEMORY_TYPES: assert_in(memory_type, d) def test_disk_device_names_returns_set_of_strings(self): device_names = self.linux.disk_device_names() assert_is_instance(device_names, set) for device_name in device_names: assert_is_instance(device_name, string_types) def test_disk_device_stats_returns_DiskDeviceStats(self): device_names = self.linux.disk_device_names() disk_device_stats = self.linux.disk_device_stats(device_names.pop()) assert_is_instance(disk_device_stats, DiskDeviceStats) for key, value in disk_device_stats._asdict().items(): assert_is_instance(value, integer_types) def test_cpu_count_returns_integer(self): cpu_count = self.linux.cpu_count() assert_is_instance(cpu_count, integer_types) def test_cpu_times(self): cpu_id = list(range(self.linux.cpu_count())).pop() cpu_times = self.linux.cpu_times(cpu_id) assert_is_instance(cpu_times, CpuTimes) def test_hostname(self): hostname = self.linux.hostname('host') assert_is_instance(hostname, string_types) def test_uname(self): uname = self.linux.uname() assert_is_instance(uname, collections.Iterable) def test_uptime(self): uptime = self.linux.uptime() assert_is_instance(uptime, integer_types) def test_max_number_of_running_processes(self): maxproc = self.linux.max_number_of_running_processes() assert_is_instance(maxproc, integer_types) def test_system_load(self): system_load = self.linux.system_load() assert_is_instance(system_load, SystemLoad) def test_swap_size(self): free, total = self.linux.swap_size('all') assert_less_equal(free, total) def test_swap_info(self): swap_info = self.linux.swap_info() assert_is_instance(swap_info, SwapInfo) def test_swap_device_names(self): swap_devices = self.linux.swap_device_names() disk_devices = self.linux.disk_device_names() for swap_device in swap_devices: assert_in(swap_device, disk_devices) @patch('zabby.hostos.linux.lines_from_file') def test_swap_device_names_skips_files(self, mock_lists): swap_file_path = '/mnt/swap_file' mock_lists.return_value = [ ['Filename', 'Type', 'Size', 'Used', 'Priority'], ['/dev/dm-0', 'partition', '10485756', '311756', '-1'], [swap_file_path, 'file', '524284', '0', '-2']] swap_devices = self.linux.swap_device_names() assert_not_in(swap_file_path.split('/')[-1], swap_devices)