@@ -36,6 +36,53 @@ def setUp(self):
3636 self .volumes_mock .reset_mock ()
3737
3838
39+ class TestTransferAccept (TestTransfer ):
40+
41+ columns = (
42+ 'id' ,
43+ 'name' ,
44+ 'volume_id' ,
45+ )
46+
47+ def setUp (self ):
48+ super (TestTransferAccept , self ).setUp ()
49+
50+ self .volume_transfer = (
51+ transfer_fakes .FakeTransfer .create_one_transfer ())
52+ self .data = (
53+ self .volume_transfer .id ,
54+ self .volume_transfer .name ,
55+ self .volume_transfer .volume_id ,
56+ )
57+
58+ self .transfer_mock .get .return_value = self .volume_transfer
59+ self .transfer_mock .accept .return_value = self .volume_transfer
60+
61+ # Get the command object to test
62+ self .cmd = volume_transfer_request .AcceptTransferRequest (
63+ self .app , None )
64+
65+ def test_transfer_accept (self ):
66+ arglist = [
67+ self .volume_transfer .id ,
68+ 'auth_key' ,
69+ ]
70+ verifylist = [
71+ ('transfer_request' , self .volume_transfer .id ),
72+ ('auth_key' , 'auth_key' ),
73+ ]
74+ parsed_args = self .check_parser (self .cmd , arglist , verifylist )
75+
76+ columns , data = self .cmd .take_action (parsed_args )
77+
78+ self .transfer_mock .get .assert_called_once_with (
79+ self .volume_transfer .id )
80+ self .transfer_mock .accept .assert_called_once_with (
81+ self .volume_transfer .id , 'auth_key' )
82+ self .assertEqual (self .columns , columns )
83+ self .assertEqual (self .data , data )
84+
85+
3986class TestTransferCreate (TestTransfer ):
4087
4188 volume = transfer_fakes .FakeVolume .create_one_volume ()
@@ -52,7 +99,10 @@ def setUp(self):
5299 super (TestTransferCreate , self ).setUp ()
53100
54101 self .volume_transfer = transfer_fakes .FakeTransfer .create_one_transfer (
55- attrs = {'volume_id' : self .volume .id })
102+ attrs = {'volume_id' : self .volume .id ,
103+ 'auth_key' : 'key' ,
104+ 'created_at' : 'time' }
105+ )
56106 self .data = (
57107 self .volume_transfer .auth_key ,
58108 self .volume_transfer .created_at ,
@@ -266,3 +316,49 @@ def test_transfer_list_with_argument(self):
266316 detailed = True ,
267317 search_opts = {'all_tenants' : 1 }
268318 )
319+
320+
321+ class TestTransferShow (TestTransfer ):
322+
323+ columns = (
324+ 'created_at' ,
325+ 'id' ,
326+ 'name' ,
327+ 'volume_id' ,
328+ )
329+
330+ def setUp (self ):
331+ super (TestTransferShow , self ).setUp ()
332+
333+ self .volume_transfer = (
334+ transfer_fakes .FakeTransfer .create_one_transfer (
335+ attrs = {'created_at' : 'time' })
336+ )
337+ self .data = (
338+ self .volume_transfer .created_at ,
339+ self .volume_transfer .id ,
340+ self .volume_transfer .name ,
341+ self .volume_transfer .volume_id ,
342+ )
343+
344+ self .transfer_mock .get .return_value = self .volume_transfer
345+
346+ # Get the command object to test
347+ self .cmd = volume_transfer_request .ShowTransferRequest (
348+ self .app , None )
349+
350+ def test_transfer_show (self ):
351+ arglist = [
352+ self .volume_transfer .id ,
353+ ]
354+ verifylist = [
355+ ('transfer_request' , self .volume_transfer .id ),
356+ ]
357+ parsed_args = self .check_parser (self .cmd , arglist , verifylist )
358+
359+ columns , data = self .cmd .take_action (parsed_args )
360+
361+ self .transfer_mock .get .assert_called_once_with (
362+ self .volume_transfer .id )
363+ self .assertEqual (self .columns , columns )
364+ self .assertEqual (self .data , data )
0 commit comments