@@ -182,6 +182,8 @@ class TestGenericHardwareManager(test_base.BaseTestCase):
182182 def setUp (self ):
183183 super (TestGenericHardwareManager , self ).setUp ()
184184 self .hardware = hardware .GenericHardwareManager ()
185+ self .node = {'uuid' : 'dda135fb-732d-4742-8e72-df8f3199d244' ,
186+ 'driver_internal_info' : {}}
185187
186188 @mock .patch ('os.listdir' )
187189 @mock .patch ('os.path.exists' )
@@ -409,7 +411,7 @@ def test_erase_block_device_ata_success(self, mocked_execute):
409411
410412 block_device = hardware .BlockDevice ('/dev/sda' , 'big' , 1073741824 ,
411413 True )
412- self .hardware .erase_block_device (block_device )
414+ self .hardware .erase_block_device (self . node , block_device )
413415 mocked_execute .assert_has_calls ([
414416 mock .call ('hdparm' , '-I' , '/dev/sda' ),
415417 mock .call ('hdparm' , '--user-master' , 'u' , '--security-set-pass' ,
@@ -422,6 +424,8 @@ def test_erase_block_device_ata_success(self, mocked_execute):
422424 @mock .patch .object (utils , 'execute' )
423425 def test_erase_block_device_nosecurity_shred (self , mocked_execute ):
424426 hdparm_output = HDPARM_INFO_TEMPLATE .split ('\n Security:' )[0 ]
427+ info = self .node .get ('driver_internal_info' )
428+ info ['agent_erase_devices_iterations' ] = 2
425429
426430 mocked_execute .side_effect = [
427431 (hdparm_output , '' ),
@@ -430,11 +434,11 @@ def test_erase_block_device_nosecurity_shred(self, mocked_execute):
430434
431435 block_device = hardware .BlockDevice ('/dev/sda' , 'big' , 1073741824 ,
432436 True )
433- self .hardware .erase_block_device (block_device )
437+ self .hardware .erase_block_device (self . node , block_device )
434438 mocked_execute .assert_has_calls ([
435439 mock .call ('hdparm' , '-I' , '/dev/sda' ),
436440 mock .call ('shred' , '--force' , '--zero' , '--verbose' ,
437- '--iterations' , '1' , '/dev/sda' )
441+ '--iterations' , 2 , '/dev/sda' )
438442 ])
439443
440444 @mock .patch .object (utils , 'execute' )
@@ -453,11 +457,11 @@ def test_erase_block_device_notsupported_shred(self, mocked_execute):
453457
454458 block_device = hardware .BlockDevice ('/dev/sda' , 'big' , 1073741824 ,
455459 True )
456- self .hardware .erase_block_device (block_device )
460+ self .hardware .erase_block_device (self . node , block_device )
457461 mocked_execute .assert_has_calls ([
458462 mock .call ('hdparm' , '-I' , '/dev/sda' ),
459463 mock .call ('shred' , '--force' , '--zero' , '--verbose' ,
460- '--iterations' , '1' , '/dev/sda' )
464+ '--iterations' , 1 , '/dev/sda' )
461465 ])
462466
463467 @mock .patch .object (hardware .GenericHardwareManager ,
@@ -466,7 +470,7 @@ def test_erase_block_device_virtual_media(self, vm_mock):
466470 vm_mock .return_value = True
467471 block_device = hardware .BlockDevice ('/dev/sda' , 'big' , 1073741824 ,
468472 True )
469- self .hardware .erase_block_device (block_device )
473+ self .hardware .erase_block_device (self . node , block_device )
470474 vm_mock .assert_called_once_with (self .hardware , block_device )
471475
472476 @mock .patch .object (os , 'readlink' , autospec = True )
@@ -512,20 +516,20 @@ def test_erase_block_device_shred_fail_oserror(self, mocked_execute):
512516 mocked_execute .side_effect = OSError
513517 block_device = hardware .BlockDevice ('/dev/sda' , 'big' , 1073741824 ,
514518 True )
515- res = self .hardware ._shred_block_device (block_device )
519+ res = self .hardware ._shred_block_device (self . node , block_device )
516520 self .assertFalse (res )
517521 mocked_execute .assert_called_once_with ('shred' , '--force' , '--zero' ,
518- '--verbose' , '--iterations' , '1' , '/dev/sda' )
522+ '--verbose' , '--iterations' , 1 , '/dev/sda' )
519523
520524 @mock .patch .object (utils , 'execute' )
521525 def test_erase_block_device_shred_fail_processerror (self , mocked_execute ):
522526 mocked_execute .side_effect = processutils .ProcessExecutionError
523527 block_device = hardware .BlockDevice ('/dev/sda' , 'big' , 1073741824 ,
524528 True )
525- res = self .hardware ._shred_block_device (block_device )
529+ res = self .hardware ._shred_block_device (self . node , block_device )
526530 self .assertFalse (res )
527531 mocked_execute .assert_called_once_with ('shred' , '--force' , '--zero' ,
528- '--verbose' , '--iterations' , '1' , '/dev/sda' )
532+ '--verbose' , '--iterations' , 1 , '/dev/sda' )
529533
530534 @mock .patch .object (utils , 'execute' )
531535 def test_erase_block_device_ata_security_enabled (self , mocked_execute ):
@@ -544,7 +548,7 @@ def test_erase_block_device_ata_security_enabled(self, mocked_execute):
544548 True )
545549 self .assertRaises (errors .BlockDeviceEraseError ,
546550 self .hardware .erase_block_device ,
547- block_device )
551+ self . node , block_device )
548552
549553 @mock .patch .object (utils , 'execute' )
550554 def test_erase_block_device_ata_frozen (self , mocked_execute ):
@@ -563,7 +567,7 @@ def test_erase_block_device_ata_frozen(self, mocked_execute):
563567 True )
564568 self .assertRaises (errors .BlockDeviceEraseError ,
565569 self .hardware .erase_block_device ,
566- block_device )
570+ self . node , block_device )
567571
568572 @mock .patch .object (utils , 'execute' )
569573 def test_erase_block_device_ata_failed (self , mocked_execute ):
@@ -594,7 +598,7 @@ def test_erase_block_device_ata_failed(self, mocked_execute):
594598 True )
595599 self .assertRaises (errors .BlockDeviceEraseError ,
596600 self .hardware .erase_block_device ,
597- block_device )
601+ self . node , block_device )
598602
599603 def test_normal_vs_enhanced_security_erase (self ):
600604 @mock .patch .object (utils , 'execute' )
@@ -617,7 +621,7 @@ def test_security_erase_option(test_case,
617621
618622 block_device = hardware .BlockDevice ('/dev/sda' , 'big' , 1073741824 ,
619623 True )
620- test_case .hardware .erase_block_device (block_device )
624+ test_case .hardware .erase_block_device (self . node , block_device )
621625 mocked_execute .assert_any_call ('hdparm' , '--user-master' , 'u' ,
622626 expected_option ,
623627 'NULL' , '/dev/sda' )
0 commit comments