@@ -118,8 +118,10 @@ def test_list_project(self):
118118
119119class TestShowIPAvailability (TestIPAvailability ):
120120
121+ _network = network_fakes .FakeNetwork .create_one_network ()
121122 _ip_availability = \
122- network_fakes .FakeIPAvailability .create_one_ip_availability ()
123+ network_fakes .FakeIPAvailability .create_one_ip_availability (
124+ attrs = {'network_id' : _network .id })
123125
124126 columns = (
125127 'network_id' ,
@@ -144,6 +146,8 @@ def setUp(self):
144146
145147 self .network .find_network_ip_availability = mock .Mock (
146148 return_value = self ._ip_availability )
149+ self .network .find_network = mock .Mock (
150+ return_value = self ._network )
147151
148152 # Get the command object to test
149153 self .cmd = ip_availability .ShowIPAvailability (
@@ -166,8 +170,10 @@ def test_show_all_options(self):
166170 parsed_args = self .check_parser (self .cmd , arglist , verifylist )
167171 columns , data = self .cmd .take_action (parsed_args )
168172 self .network .find_network_ip_availability .assert_called_once_with (
173+ self ._ip_availability .network_id ,
174+ ignore_missing = False )
175+ self .network .find_network .assert_called_once_with (
169176 self ._ip_availability .network_name ,
170177 ignore_missing = False )
171-
172178 self .assertEqual (self .columns , columns )
173179 self .assertEqual (self .data , data )
0 commit comments