Lines Matching refs:atft_manager
120 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
122 atft_manager._GetOs = MagicMock()
123 atft_manager._SetOs = MagicMock()
124 atft_manager._GetOs.return_value = 'Windows'
126 atft_manager.ListDevices()
127 atft_manager.ListDevices()
128 self.assertEqual(atft_manager.atfa_dev.serial_number, self.ATFA_TEST_SERIAL)
129 self.assertEqual(1, len(atft_manager.target_devs))
130 self.assertEqual(atft_manager.target_devs[0].serial_number,
132 self.assertEqual(None, atft_manager._atfa_dev_setting)
133 self.assertEqual(True, atft_manager._atfa_reboot_lock.acquire(False))
134 atft_manager._SetOs.assert_not_called()
140 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
142 atft_manager._GetOs = MagicMock()
143 atft_manager._SetOs = MagicMock()
144 atft_manager._GetOs.return_value = 'Linux'
146 atft_manager.ListDevices()
147 atft_manager.ListDevices()
148 atft_manager._GetOs.assert_called()
149 atft_manager._SetOs.assert_called_once_with(
150 atft_manager.atfa_dev, 'Windows')
151 self.assertEqual(atft_manager.atfa_dev.serial_number, self.ATFA_TEST_SERIAL)
157 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
159 atft_manager._GetOs = MagicMock()
160 atft_manager._GetOs.return_value = 'Windows'
162 atft_manager.ListDevices()
163 atft_manager.ListDevices()
164 self.assertEqual(atft_manager.atfa_dev.serial_number, self.ATFA_TEST_SERIAL)
165 self.assertEqual(0, len(atft_manager.target_devs))
171 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
173 atft_manager._GetOs = MagicMock()
174 atft_manager._GetOs.return_value = 'Windows'
176 atft_manager.ListDevices()
177 atft_manager.ListDevices()
178 self.assertEqual(atft_manager.atfa_dev, None)
179 self.assertEqual(1, len(atft_manager.target_devs))
180 self.assertEqual(atft_manager.target_devs[0].serial_number,
187 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
189 atft_manager._GetOs = MagicMock()
190 atft_manager._GetOs.return_value = 'Windows'
192 atft_manager.ListDevices()
193 atft_manager.ListDevices()
194 self.assertEqual(atft_manager.atfa_dev, None)
195 self.assertEqual(2, len(atft_manager.target_devs))
196 self.assertEqual(atft_manager.target_devs[0].serial_number,
198 self.assertEqual(atft_manager.target_devs[1].serial_number,
206 atft_manager = atftman.AtftManager(mock_fastboot, self.mock_serial_mapper,
208 atft_manager._GetOs = MagicMock()
209 atft_manager._GetOs.return_value = 'Windows'
211 atft_manager.ListDevices()
212 atft_manager.ListDevices()
214 atft_manager.ListDevices()
215 atft_manager.ListDevices()
216 self.assertEqual(atft_manager.atfa_dev, None)
217 self.assertEqual(1, len(atft_manager.target_devs))
218 self.assertEqual(atft_manager.target_devs[0].serial_number,
227 atft_manager = atftman.AtftManager(mock_fastboot, self.mock_serial_mapper,
229 atft_manager._GetOs = MagicMock()
230 atft_manager._GetOs.return_value = 'Windows'
232 atft_manager.ListDevices()
233 atft_manager.ListDevices()
237 atft_manager.ListDevices()
238 atft_manager.ListDevices()
239 self.assertEqual(atft_manager.atfa_dev.serial_number, self.ATFA_TEST_SERIAL)
240 self.assertEqual(1, len(atft_manager.target_devs))
241 self.assertEqual(atft_manager.target_devs[0].serial_number,
250 atft_manager = atftman.AtftManager(mock_fastboot, self.mock_serial_mapper,
252 atft_manager._GetOs = MagicMock()
253 atft_manager._GetOs.return_value = 'Windows'
255 atft_manager.ListDevices()
256 atft_manager.ListDevices()
260 atft_manager.ListDevices()
261 atft_manager.ListDevices()
262 self.assertEqual(atft_manager.atfa_dev.serial_number, self.ATFA_TEST_SERIAL)
263 self.assertEqual(1, len(atft_manager.target_devs))
264 self.assertEqual(atft_manager.target_devs[0].serial_number,
273 atft_manager = atftman.AtftManager(mock_fastboot, self.mock_serial_mapper,
275 atft_manager._GetOs = MagicMock()
276 atft_manager._GetOs.return_value = 'Windows'
280 atft_manager.ListDevices()
281 atft_manager.ListDevices()
285 atft_manager.ListDevices()
286 atft_manager.ListDevices()
287 self.assertEqual(atft_manager.atfa_dev, None)
288 self.assertEqual(2, len(atft_manager.target_devs))
289 self.assertEqual(atft_manager.target_devs[0].serial_number,
291 self.assertEqual(atft_manager.target_devs[1].serial_number,
300 atft_manager = atftman.AtftManager(mock_fastboot, self.mock_serial_mapper,
302 atft_manager._GetOs = MagicMock()
303 atft_manager._GetOs.return_value = 'Windows'
307 atft_manager.ListDevices()
308 atft_manager.ListDevices()
312 atft_manager.ListDevices()
314 self.assertEqual(0, len(atft_manager.target_devs))
316 atft_manager.ListDevices()
317 self.assertEqual(atft_manager.atfa_dev.serial_number, self.ATFA_TEST_SERIAL)
318 self.assertEqual(1, len(atft_manager.target_devs))
319 self.assertEqual(atft_manager.target_devs[0].serial_number,
328 atft_manager = atftman.AtftManager(mock_fastboot, self.mock_serial_mapper,
330 atft_manager._GetOs = MagicMock()
331 atft_manager._GetOs.return_value = 'Windows'
335 atft_manager.ListDevices()
336 atft_manager.ListDevices()
338 atft_manager.ListDevices()
339 self.assertEqual(atft_manager.atfa_dev, None)
340 self.assertEqual(1, len(atft_manager.target_devs))
341 self.assertEqual(atft_manager.target_devs[0].serial_number,
349 atft_manager = atftman.AtftManager(mock_fastboot, self.mock_serial_mapper,
351 atft_manager._GetOs = MagicMock()
352 atft_manager._GetOs.return_value = 'Windows'
356 atft_manager.ListDevices()
357 atft_manager.ListDevices()
359 atft_manager.ListDevices()
360 self.assertEqual(atft_manager.atfa_dev.serial_number, self.ATFA_TEST_SERIAL)
361 self.assertEqual(0, len(atft_manager.target_devs))
368 atft_manager = atftman.AtftManager(mock_fastboot, self.mock_serial_mapper,
370 atft_manager._GetOs = MagicMock()
371 atft_manager._GetOs.return_value = 'Windows'
373 atft_manager.ListDevices()
375 self.assertEqual(0, len(atft_manager.target_devs))
382 atft_manager = atftman.AtftManager(mock_fastboot, self.mock_serial_mapper,
384 atft_manager._GetOs = MagicMock()
385 atft_manager._GetOs.return_value = 'Windows'
387 atft_manager.ListDevices()
388 self.assertEqual(0, len(atft_manager.target_devs))
394 atft_manager.ListDevices()
395 atft_manager.target_devs))
402 atft_manager = atftman.AtftManager(mock_fastboot, self.mock_serial_mapper,
404 atft_manager._GetOs = MagicMock()
405 atft_manager._GetOs.return_value = 'Windows'
407 atft_manager.ListDevices()
408 self.assertEqual(0, len(atft_manager.target_devs))
410 atft_manager.ListDevices()
412 self.assertEqual(0, len(atft_manager.target_devs))
440 atft_manager = atftman.AtftManager(
442 atft_manager._GetOs = MagicMock()
443 atft_manager._GetOs.return_value = 'Windows'
447 atft_manager.ListDevices()
448 atft_manager.ListDevices()
449 self.assertEqual(atft_manager.atfa_dev.location, self.TEST_LOCATION)
450 self.assertEqual(atft_manager.target_devs[0].location, self.TEST_LOCATION2)
467 atft_manager = atftman.AtftManager(
472 atft_manager.ListDevices()
473 atft_manager.ListDevices()
474 self.assertEqual(3, len(atft_manager.target_devs))
475 self.assertEqual(self.TEST_LOCATION, atft_manager.target_devs[0].location)
476 self.assertEqual(self.TEST_LOCATION3, atft_manager.target_devs[1].location)
477 self.assertEqual(self.TEST_LOCATION2, atft_manager.target_devs[2].location)
493 atft_manager = atftman.AtftManager(
498 atft_manager.ListDevices(atft_manager.SORT_BY_LOCATION)
499 atft_manager.ListDevices(atft_manager.SORT_BY_LOCATION)
500 self.assertEqual(3, len(atft_manager.target_devs))
501 self.assertEqual(self.TEST_LOCATION, atft_manager.target_devs[0].location)
502 self.assertEqual(self.TEST_LOCATION3, atft_manager.target_devs[1].location)
503 self.assertEqual(self.TEST_LOCATION2, atft_manager.target_devs[2].location)
522 atft_manager = atftman.AtftManager(
527 atft_manager.ListDevices(atft_manager.SORT_BY_SERIAL)
528 atft_manager.ListDevices(atft_manager.SORT_BY_SERIAL)
529 self.assertEqual(3, len(atft_manager.target_devs))
531 atft_manager.target_devs[0].serial_number)
533 atft_manager.target_devs[1].serial_number)
535 atft_manager.target_devs[2].serial_number)
563 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
579 atft_manager.TransferContent(src, dst)
587 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
591 algorithm = atft_manager._ChooseAlgorithm([p256, curve])
595 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
598 algorithm = atft_manager._ChooseAlgorithm([p256])
602 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
605 algorithm = atft_manager._ChooseAlgorithm([curve])
609 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
612 atft_manager._ChooseAlgorithm([])
615 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
618 atft_manager._ChooseAlgorithm(['abcd'])
624 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
626 algorithm_list = atft_manager._GetAlgorithmList(mock_target)
660 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
663 atft_manager.atfa_dev = mock_atfa_dev
664 test_atfa_device_manager = atftman.AtfaDeviceManager(atft_manager)
665 atft_manager.product_info = MagicMock()
666 atft_manager.product_info.product_id = self.TEST_ID
670 self.assertEqual(100, atft_manager.GetATFAKeysLeft())
673 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
676 atft_manager.atfa_dev = mock_atfa_dev
677 test_atfa_device_manager = atftman.AtfaDeviceManager(atft_manager)
678 atft_manager.product_info = MagicMock()
679 atft_manager.product_info.product_id = self.TEST_ID
683 self.assertEqual(100, atft_manager.GetATFAKeysLeft())
686 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
689 atft_manager.atfa_dev = mock_atfa_dev
690 atft_manager.product_info = None
691 test_atfa_device_manager = atftman.AtfaDeviceManager(atft_manager)
697 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
699 atft_manager.atfa_dev = None
700 atft_manager.product_info = MagicMock()
701 atft_manager.product_info.product_id = self.TEST_ID
702 test_atfa_device_manager = atftman.AtfaDeviceManager(atft_manager)
707 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
710 atft_manager.atfa_dev = mock_atfa_dev
711 test_atfa_device_manager = atftman.AtfaDeviceManager(atft_manager)
712 atft_manager.product_info = MagicMock()
713 atft_manager.product_info.product_id = self.TEST_ID
719 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
722 atft_manager.atfa_dev = mock_atfa_dev
723 test_atfa_device_manager = atftman.AtfaDeviceManager(atft_manager)
724 atft_manager.product_info = MagicMock()
725 atft_manager.product_info.product_id = self.TEST_ID
735 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
749 atft_manager.CheckProvisionStatus(mock_device)
754 atft_manager.CheckProvisionStatus(mock_device)
767 atft_manager.CheckProvisionStatus(mock_device)
780 atft_manager.CheckProvisionStatus(mock_device)
793 atft_manager.CheckProvisionStatus(mock_device)
798 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
812 atft_manager.CheckProvisionStatus(mock_device)
826 atft_manager.CheckProvisionStatus(mock_device)
840 atft_manager.CheckProvisionStatus(mock_device)
845 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
859 atft_manager.CheckProvisionStatus(mock_device)
867 atft_manager.CheckProvisionStatus(mock_device)
882 atft_manager.CheckProvisionStatus(mock_device)
899 atft_manager.CheckProvisionStatus(mock_device)
914 atft_manager.CheckProvisionStatus(mock_device)
934 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
938 atft_manager._atfa_dev_manager = MagicMock()
939 atft_manager.atfa_dev = mock_atfa
940 atft_manager._GetAlgorithmList = MagicMock()
941 atft_manager._GetAlgorithmList.return_value = [
944 atft_manager.TransferContent = MagicMock()
945 atft_manager.CheckProvisionStatus = MagicMock()
946 atft_manager.CheckProvisionStatus.side_effect = self.MockSetProvisionSuccess
948 atft_manager.Provision(mock_target)
951 atft_manager._atfa_dev_manager.SetTime.assert_called_once()
958 atft_manager.TransferContent.assert_has_calls(transfer_content_calls)
972 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
976 atft_manager._atfa_dev_manager = MagicMock()
977 atft_manager.atfa_dev = mock_atfa
978 atft_manager._GetAlgorithmList = MagicMock()
979 atft_manager._GetAlgorithmList.return_value = [
982 atft_manager.TransferContent = MagicMock()
983 atft_manager.CheckProvisionStatus = MagicMock()
984 atft_manager.CheckProvisionStatus.side_effect = self.MockSetProvisionFail
986 atft_manager.Provision(mock_target)
1006 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
1008 atft_manager.product_info = ProductInfo(
1012 atft_manager.CheckProvisionStatus = MagicMock()
1013 atft_manager.CheckProvisionStatus.side_effect = self.MockSetFuseVbootSuccess
1015 atft_manager.FuseVbootKey(mock_target)
1029 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
1031 atft_manager.product_info = ProductInfo(
1035 atft_manager.CheckProvisionStatus = MagicMock()
1036 atft_manager.CheckProvisionStatus.side_effect = self.MockSetFuseVbootFail
1038 atft_manager.FuseVbootKey(mock_target)
1041 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
1044 atft_manager.product_info = None
1046 atft_manager.FuseVbootKey(mock_target)
1055 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
1057 atft_manager.product_info = ProductInfo(
1061 atft_manager.CheckProvisionStatus = MagicMock()
1065 atft_manager.FuseVbootKey(mock_target)
1087 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
1089 atft_manager.product_info = ProductInfo(
1093 atft_manager.CheckProvisionStatus = MagicMock()
1094 atft_manager.CheckProvisionStatus.side_effect = self.MockSetFuseAttrSuccess
1096 atft_manager.FusePermAttr(mock_target)
1110 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
1112 atft_manager.product_info = ProductInfo(
1116 atft_manager.CheckProvisionStatus = MagicMock()
1117 atft_manager.CheckProvisionStatus.side_effect = self.MockSetFuseAttrFail
1119 atft_manager.FusePermAttr(mock_target)
1122 atft_manager
1125 atft_manager.product_info = None
1127 atft_manager.FusePermAttr(mock_target)
1136 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
1138 atft_manager.product_info = ProductInfo(
1142 atft_manager.CheckProvisionStatus = MagicMock()
1146 atft_manager.FusePermAttr(mock_target)
1162 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
1165 atft_manager.CheckProvisionStatus = MagicMock()
1166 atft_manager.CheckProvisionStatus.side_effect = self.MockSetLockAvbSuccess
1167 atft_manager.LockAvb(mock_target)
1173 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
1176 atft_manager.CheckProvisionStatus = MagicMock()
1177 atft_manager.CheckProvisionStatus.side_effect = self.MockSetLockAvbFail
1179 atft_manager.LockAvb(mock_target)
1182 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
1185 atft_manager.CheckProvisionStatus = MagicMock()
1186 atft_manager.CheckProvisionStatus.side_effect = self.MockSetLockAvbSuccess
1189 atft_manager.LockAvb(mock_target)
1216 atft_manager = atftman.AtftManager(
1219 atft_manager.stable_serials = [self.TEST_SERIAL]
1224 atft_manager.target_devs.append(test_device)
1229 atft_manager.Reboot(test_device, timeout, mock_success, mock_fail)
1232 self.assertEqual(1, len(atft_manager.target_devs))
1235 atft_manager.target_devs[0].provision_status)
1237 self.TEST_SERIAL, atft_manager.target_devs[0].serial_number)
1240 atft_manager.stable_serials = [self.TEST_SERIAL]
1241 atft_manager._HandleRebootCallbacks()
1245 self.assertEqual(1, len(atft_manager.target_devs))
1248 atft_manager.target_devs[0].provision_status)
1258 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
1261 atft_manager.stable_serials.append(self.TEST_SERIAL)
1265 atft_manager.target_devs.append(test_device)
1269 atft_manager.CheckProvisionStatus = MagicMock()
1270 atft_manager.CheckProvisionStatus.side_effect = self.MockSetFuseVbootSuccess
1273 atft_manager.Reboot(test_device, timeout, mock_success, mock_fail)
1274 atft_manager.stable_serials = []
1276 atft_manager._HandleRebootCallbacks()
1281 self.assertEqual(0, len(atft_manager.target_devs))
1288 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
1291 atft_manager.stable_serials.append(self.TEST_SERIAL)
1295 atft_manager.target_devs.append(test_device)
1299 atft_manager.CheckProvisionStatus = MagicMock()
1300 atft_manager.CheckProvisionStatus.side_effect = self.MockSetFuseVbootSuccess
1303 atft_manager.Reboot(test_device, timeout, mock_success, mock_fail)
1304 atft_manager.stable_serials = []
1308 atft_manager._HandleRebootCallbacks()
1310 self.assertEqual(0, len(atft_manager.target_devs))
1317 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
1322 atft_manager.stable_serials.append(self.TEST_SERIAL)
1324 atft_manager.target_devs.append(test_device)
1328 atft_manager.CheckProvisionStatus = MagicMock()
1329 atft_manager.CheckProvisionStatus.side_effect = self.MockSetFuseVbootSuccess
1334 atft_manager.Reboot(mock_target, timeout, mock_success, mock_fail)
1339 atft_manager._HandleRebootCallbacks()
1354 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
1356 atft_manager.ProcessProductAttributesFile(test_content)
1357 self.assertEqual(self.TEST_NAME, atft_manager.product_info.product_name)
1358 self.assertEqual(self.TEST_ID, atft_manager.product_info.product_id)
1360 atft_manager.product_info.product_attributes)
1362 atft_manager.product_info.vboot_key)
1374 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
1377 atft_manager.ProcessProductAttributesFile(test_content)
1388 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
1391 atft_manager.ProcessProductAttributesFile(test_content)
1404 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
1407 atft_manager.ProcessProductAttributesFile(test_content)
1419 atft_manager = atftman.AtftManager(self.FastbootDeviceTemplate,
1422 atft_manager.ProcessProductAttributesFile(test_content)