コード例 #1
0
ファイル: test_linux.py プロジェクト: holandes22/dmtcore
    def test__extract_all_hctls_from_proc_scsi_file(self, _populate_disks_entries_mock):
        with patch("dmtcore.os.disk.linux.open", create=True) as open_mock:
            open_mock.return_value = MagicMock(spec=file)

            file_handle = open_mock.return_value.__enter__.return_value
            file_handle.readlines.return_value = FAKE_CONTENT_OF_PROC_SCSI.splitlines()

            dq = LinuxDiskDeviceQueries()
            expected_results = [
                HctlInfo(0, 0, 0, 0),
                HctlInfo(1, 0, 2, 0),
                HctlInfo(1, 0, 2, 1),
                HctlInfo(1, 0, 3, 0),
                HctlInfo(1, 0, 3, 1),
                HctlInfo(0, 0, 0, 0),
            ]
            actual_results = dq._extract_all_hctls_from_proc_scsi_file()
            self.assertEqual(len(expected_results), len(actual_results))

            for expected_result, actual_result in map(None, expected_results, actual_results):
                self.assertEqual(expected_result, actual_result)