def _initialize_orchestration(self):
if self._should_enable_faucetizer:
self._initialize_faucetizer()
self._faucetizer.reload_structural_config()
if self._gauge_config_file:
self._faucetizer.reload_and_flush_gauge_config(self._gauge_config_file)
if self._segments_vlans_file:
self._faucetizer.reload_segments_to_vlans(self._segments_vlans_file)
process_device_placement = self._faucetizer.process_device_placement
process_device_behavior = self._faucetizer.process_device_behavior
get_vlan_from_segment = self._faucetizer.get_vlan_from_segment
else:
process_device_placement = None
process_device_behavior = None
get_vlan_from_segment = None
self._port_state_manager = PortStateManager(
process_device_placement, process_device_behavior, get_vlan_from_segment,
self._config.orchestration.sequester_config.segment)
sequester_segment, grpc_server_port = self._calculate_sequester_config()
if sequester_segment:
self._device_report_server = DeviceReportServer(
self._port_state_manager.handle_testing_result, grpc_server_port)
self._attempt_authenticator_initialise()
self._process_static_device_placement()
self._process_static_device_behavior()
if self._faucetizer:
self._faucetizer.flush_behavioral_config(force=True)
def setUp(self):
"""Setup fixture for each test method"""
self._client = DeviceReportClient(server_port=self._SERVER_PORT)
self._server = DeviceReportServer(
self._process_result, self._SERVER_ADDRESS, self._SERVER_PORT)
self._server.start()
def setUp(self):
"""setup fixture for each test method"""
channel = grpc.insecure_channel(f'{self.SERVER_ADDRESS}:{self.SERVER_PORT}')
self._client = DeviceReportStub(channel)
self._server = DeviceReportServer(
self._process_devices_state, self.SERVER_ADDRESS, self.SERVER_PORT)
self._server.start()
def _create_device_report_handler(self):
service_address, service_port = self._get_testing_service_config()
if service_address:
unauth_vlan = self._config.orchestration.unauthenticated_vlan
self._logger.info('Connecting report client to %s:%s',
service_address, service_port)
return DeviceReportClient(self._handle_device_result,
service_address, service_port,
unauth_vlan)
self._logger.info('Starting device report server on %s', service_port)
return DeviceReportServer(self._handle_device_result, service_port)
def setUp(self):
"""Setup fixture for each test method"""
self._client = DeviceReportClient(server_port=self._SERVER_PORT2)
def mock_function(_, __, ___):
for event in self._mock_port_events:
yield event
with patch.object(DeviceReportServicer, 'GetPortState', side_effect=mock_function,
autospec=True):
self._server = DeviceReportServer(
self._process_result, self._SERVER_ADDRESS, self._SERVER_PORT2)
self._server.start()