<s> import mock <EOL> from tests import test <EOL> from cloudferry . lib . base import exception <EOL> from cloudferry . lib . os . actions import verify_vms <EOL> class VerifyVmsTest ( test . TestCase ) : <EOL> def setUp ( self ) : <EOL> super ( VerifyVmsTest , self ) . setUp ( ) <EOL> self . fake_src_cloud = mock . Mock ( ) <EOL> self . fake_info = { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : <NUM_LIT:1> <EOL> } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT:int>' : '<STR_LIT>' <EOL> } , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ { <EOL> '<STR_LIT>' : <NUM_LIT:1> <EOL> } ] , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> } <EOL> } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : <NUM_LIT:1> <EOL> } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT:int>' : '<STR_LIT>' <EOL> } , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ { <EOL> '<STR_LIT>' : <NUM_LIT:1> <EOL> } ] , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> } <EOL> } <EOL> } <EOL> } <EOL> self . fake_compute = mock . Mock ( ) <EOL> self . fake_src_cloud . resources = { <EOL> '<STR_LIT>' : self . fake_compute <EOL> } <EOL> self . fake_init = { <EOL> '<STR_LIT>' : self . fake_src_cloud , <EOL> } <EOL> self . fake_dst_info = { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : <NUM_LIT:1> <EOL> } , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT:int>' : '<STR_LIT>' <EOL> } , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : <NUM_LIT:1> <EOL> } <EOL> } ] <EOL> } <EOL> } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : <NUM_LIT:1> <EOL> } , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT:int>' : '<STR_LIT>' <EOL> } , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : <NUM_LIT:1> <EOL> } <EOL> } ] <EOL> } <EOL> } <EOL> } <EOL> } <EOL> def test_verify_vms_no_exc ( self ) : <EOL> action = verify_vms . VerifyVms ( <EOL> self . fake_init , <EOL> cloud = '<STR_LIT>' ) <EOL> try : <EOL> action . run ( info = self . fake_dst_info , info_backup = self . fake_info ) <EOL> except exception . AbortMigrationError as e : <EOL> self . fail ( e ) <EOL> def test_verify_vms_exc ( self ) : <EOL> action = verify_vms . VerifyVms ( <EOL> self . fake_init , <EOL> cloud = '<STR_LIT>' ) <EOL> self . assertRaises ( exception . AbortMigrationError , action . run ) <EOL> def test_verify_vms_all ( self ) : <EOL> action = verify_vms . VerifyVms ( <EOL> self . fake_init , <EOL> cloud = '<STR_LIT>' ) <EOL> res = action . run ( info = self . fake_dst_info , info_backup = self . fake_info ) <EOL> self . assertEqual ( True , res ) </s>
