def test_cbsd_register_updates_active_mode_config_if_desired_state_is_unregistered(self):
cbsd = self._build_cbsd(SOME_SERIAL_NUMBER)
active_mode_config = DBActiveModeConfig(
cbsd=cbsd,
desired_state_id=self.cbsd_states[CbsdStates.UNREGISTERED.value],
)
self.session.add_all([cbsd, active_mode_config])
self.session.commit()
request = self._build_request(**test_cbsd_dict)
self.dp_service.CBSDRegister(request, None)
self.assertEqual(1, self._get_active_cbsd_count(request))
def _create_or_update_active_mode_config(self, session: Session, cbsd: DBCbsd) -> DBActiveModeConfig:
registered_state = session.query(DBCbsdState). \
filter(DBCbsdState.name == CbsdStates.REGISTERED.value).first()
active_mode_config = session.query(DBActiveModeConfig). \
filter(DBActiveModeConfig.cbsd_id == cbsd.id).first()
if active_mode_config:
active_mode_config.desired_state = registered_state
return None
active_mode_config = DBActiveModeConfig(
desired_state=registered_state,
cbsd=cbsd,
)
session.add(active_mode_config)
return active_mode_config
def with_active_mode_config(self, desired_state_id: int) -> DBCbsdBuilder:
config = DBActiveModeConfig(
desired_state_id=desired_state_id,
)
self.cbsd.active_mode_config.append(config)
return self