@@ -224,8 +224,11 @@ def setUp(self):
224224
225225 # Get the command object to test
226226 self .cmd = port .ListPort (self .app , self .namespace )
227-
228227 self .network .ports = mock .Mock (return_value = self ._ports )
228+ fake_router = network_fakes .FakeRouter .create_one_router ({
229+ 'id' : 'fake-router-id' ,
230+ })
231+ self .network .find_router = mock .Mock (return_value = fake_router )
229232
230233 def test_port_list_no_options (self ):
231234 arglist = []
@@ -239,6 +242,25 @@ def test_port_list_no_options(self):
239242 self .assertEqual (self .columns , columns )
240243 self .assertEqual (self .data , list (data ))
241244
245+ def test_port_list_router_opt (self ):
246+ arglist = [
247+ '--router' , 'fake-router-name' ,
248+ ]
249+
250+ verifylist = [
251+ ('router' , 'fake-router-name' )
252+ ]
253+
254+ parsed_args = self .check_parser (self .cmd , arglist , verifylist )
255+
256+ columns , data = self .cmd .take_action (parsed_args )
257+
258+ self .network .ports .assert_called_with (** {
259+ 'device_id' : 'fake-router-id'
260+ })
261+ self .assertEqual (self .columns , columns )
262+ self .assertEqual (self .data , list (data ))
263+
242264
243265class TestSetPort (TestPort ):
244266
0 commit comments