def setUp(self):
super().setUp()
self.amc_service = ActiveModeControllerService(
SessionManager(self.engine), )
DBInitializer(SessionManager(self.engine)).initialize()
grant_states = {
x.name: x.id
for x in self.session.query(DBGrantState).all()
}
cbsd_states = {
x.name: x.id
for x in self.session.query(DBCbsdState).all()
}
request_states = {
x.name: x.id
for x in self.session.query(DBRequestState).all()
}
self.unregistered = cbsd_states[CbsdStates.UNREGISTERED.value]
self.registered = cbsd_states[CbsdStates.REGISTERED.value]
self.idle = grant_states[GrantStates.IDLE.value]
self.granted = grant_states[GrantStates.GRANTED.value]
self.authorized = grant_states[GrantStates.AUTHORIZED.value]
self.pending = request_states[RequestStates.PENDING.value]
self.processed = request_states[RequestStates.PROCESSED.value]
def run():
"""
Top-level function for radio controller
"""
logger.info("Starting grpc server")
config = get_config()
scheduler = BackgroundScheduler()
metricsd_client = get_metricsd_client()
scheduler.add_job(
process_metrics,
args=[metricsd_client, config.SERVICE_HOSTNAME, "radio_controller"],
trigger=IntervalTrigger(
seconds=config.METRICS_PROCESSING_INTERVAL_SEC, ),
max_instances=1,
name="metrics_processing_job",
)
scheduler.start()
logger.info(f"grpc port is: {config.GRPC_PORT}")
db_engine = create_engine(
url=config.SQLALCHEMY_DB_URI,
encoding=config.SQLALCHEMY_DB_ENCODING,
echo=config.SQLALCHEMY_ECHO,
future=config.SQLALCHEMY_FUTURE,
pool_size=config.SQLALCHEMY_ENGINE_POOL_SIZE,
max_overflow=config.SQLALCHEMY_ENGINE_MAX_OVERFLOW,
)
session_manager = SessionManager(db_engine)
server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
add_RadioControllerServicer_to_server(
RadioControllerService(session_manager=session_manager),
server,
)
add_ActiveModeControllerServicer_to_server(
ActiveModeControllerService(session_manager=session_manager),
server,
)
add_DPServiceServicer_to_server(
DPService(
session_manager=session_manager,
now_func=datetime.now,
fluentd_client=FluentdClient(),
),
server,
)
server.add_insecure_port(f"[::]:{config.GRPC_PORT}")
server.start()
logger.info(f"GRPC Server started on port {config.GRPC_PORT}")
def handle_sigterm(*_):
logger.info("Received shutdown signal")
all_rpcs_done_event = server.stop(30)
all_rpcs_done_event.wait(30)
logger.info("Shut down gracefully")
signal(SIGTERM, handle_sigterm)
server.wait_for_termination()
def run():
"""
Top-level function for radio controller
"""
logger.info("Starting grpc server")
config = get_config()
logger.info(f"grpc port is: {config.GRPC_PORT}")
db_engine = create_engine(
url=config.SQLALCHEMY_DB_URI,
encoding=config.SQLALCHEMY_DB_ENCODING,
echo=config.SQLALCHEMY_ECHO,
future=config.SQLALCHEMY_FUTURE,
pool_size=config.SQLALCHEMY_ENGINE_POOL_SIZE,
max_overflow=config.SQLALCHEMY_ENGINE_MAX_OVERFLOW,
)
session_manager = SessionManager(db_engine)
server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
add_RadioControllerServicer_to_server(
RadioControllerService(session_manager=session_manager),
server,
)
add_ActiveModeControllerServicer_to_server(
ActiveModeControllerService(session_manager=session_manager),
server,
)
add_DPServiceServicer_to_server(
DPService(
session_manager=session_manager,
now_func=datetime.now,
fluentd_client=FluentdClient(),
),
server,
)
server.add_insecure_port(f"[::]:{config.GRPC_PORT}")
server.start()
logger.info(f"GRPC Server started on port {config.GRPC_PORT}")
def handle_sigterm(*_):
logger.info("Received shutdown signal")
all_rpcs_done_event = server.stop(30)
all_rpcs_done_event.wait(30)
logger.info("Shut down gracefully")
signal(SIGTERM, handle_sigterm)
server.wait_for_termination()
class ActiveModeControllerTestCase(LocalDBTestCase):
def setUp(self):
super().setUp()
self.amc_service = ActiveModeControllerService(
SessionManager(self.engine), )
DBInitializer(SessionManager(self.engine)).initialize()
grant_states = {
x.name: x.id
for x in self.session.query(DBGrantState).all()
}
cbsd_states = {
x.name: x.id
for x in self.session.query(DBCbsdState).all()
}
request_states = {
x.name: x.id
for x in self.session.query(DBRequestState).all()
}
self.unregistered = cbsd_states[CbsdStates.UNREGISTERED.value]
self.registered = cbsd_states[CbsdStates.REGISTERED.value]
self.idle = grant_states[GrantStates.IDLE.value]
self.granted = grant_states[GrantStates.GRANTED.value]
self.authorized = grant_states[GrantStates.AUTHORIZED.value]
self.pending = request_states[RequestStates.PENDING.value]
self.processed = request_states[RequestStates.PROCESSED.value]
def test_get_basic_state(self):
cbsd = self._prepare_base_cbsd().build()
self.session.add(cbsd)
self.session.commit()
config = self._prepare_base_active_mode_config().build()
expected = State(active_mode_configs=[config])
actual = self.amc_service.GetState(GetStateRequest(), None)
self.assertEqual(expected, actual)
def test_get_state_with_grants(self):
cbsd = self._prepare_base_cbsd(). \
with_grant("idle_grant", self.idle, 1, 2). \
with_grant("granted_grant", self.granted, 3). \
with_grant("authorized_grant", self.authorized, 5, 6). \
build()
self.session.add(cbsd)
self.session.commit()
config = self._prepare_base_active_mode_config(). \
with_grant("granted_grant", Granted, 3, 0). \
with_grant("authorized_grant", Authorized, 5, 6). \
build()
expected = State(active_mode_configs=[config])
actual = self.amc_service.GetState(GetStateRequest(), None)
self.assertEqual(expected, actual)
def test_get_state_with_channels(self):
cbsd = self._prepare_base_cbsd(). \
with_channel(1, 2, 3, 4). \
with_channel(5, 6). \
build()
self.session.add(cbsd)
self.session.commit()
config = self._prepare_base_active_mode_config(). \
with_channel(1, 2, 3, 4). \
with_channel(5, 6). \
build()
expected = State(active_mode_configs=[config])
actual = self.amc_service.GetState(GetStateRequest(), None)
self.assertEqual(expected, actual)
def test_get_state_with_last_seen(self):
cbsd = self._prepare_base_cbsd().\
with_last_seen(1).\
build()
self.session.add(cbsd)
self.session.commit()
config = self._prepare_base_active_mode_config(). \
with_last_seen(1).\
build()
expected = State(active_mode_configs=[config])
actual = self.amc_service.GetState(GetStateRequest(), None)
self.assertEqual(expected, actual)
def test_get_state_with_requests(self):
cbsd = self._prepare_base_cbsd().\
with_request(self.processed, '{"key1":"value1"}').\
with_request(self.pending, '{"key2":"value2"}').\
build()
self.session.add(cbsd)
self.session.commit()
config = self._prepare_base_active_mode_config().\
with_pending_request('{"key2":"value2"}').\
build()
expected = State(active_mode_configs=[config])
actual = self.amc_service.GetState(GetStateRequest(), None)
self.assertEqual(expected, actual)
def test_get_state_with_multiple_cbsds(self):
some_cbsd = DBCbsdBuilder().\
with_state(self.unregistered).\
with_registration('some').\
with_eirp_capabilities(0, 10, 20, 1).\
with_active_mode_config(self.registered).\
build()
other_cbsd = DBCbsdBuilder(). \
with_state(self.registered). \
with_registration('other'). \
with_eirp_capabilities(5, 15, 25, 3). \
with_active_mode_config(self.registered). \
build()
self.session.add_all([some_cbsd, other_cbsd])
self.session.commit()
some_config = ActiveModeConfigBuilder().\
with_state(Unregistered).\
with_registration('some').\
with_eirp_capabilities(0, 10, 20, 1).\
with_desired_state(Registered).\
build()
other_config = ActiveModeConfigBuilder(). \
with_state(Registered). \
with_registration('other'). \
with_eirp_capabilities(5, 15, 25, 3). \
with_desired_state(Registered). \
build()
expected = State(active_mode_configs=[some_config, other_config])
actual = self.amc_service.GetState(GetStateRequest(), None)
self.assertEqual(expected, actual)
def test_not_get_state_without_active_mode_config(self):
cbsd = DBCbsdBuilder(). \
with_state(self.unregistered). \
with_registration('some'). \
with_eirp_capabilities(0, 10, 20, 1). \
build()
self.session.add(cbsd)
self.session.commit()
expected = State()
actual = self.amc_service.GetState(GetStateRequest(), None)
self.assertEqual(expected, actual)
def test_not_get_state_without_registration_params(self):
cbsd = DBCbsdBuilder(). \
with_state(self.unregistered). \
with_eirp_capabilities(0, 10, 20, 1). \
with_active_mode_config(self.registered).\
build()
self.session.add(cbsd)
self.session.commit()
expected = State()
actual = self.amc_service.GetState(GetStateRequest(), None)
self.assertEqual(expected, actual)
def test_not_get_state_without_eirp_capabilities(self):
cbsd = DBCbsdBuilder(). \
with_state(self.unregistered). \
with_registration('some'). \
with_active_mode_config(self.registered). \
build()
self.session.add(cbsd)
self.session.commit()
expected = State()
actual = self.amc_service.GetState(GetStateRequest(), None)
self.assertEqual(expected, actual)
def _prepare_base_cbsd(self) -> DBCbsdBuilder:
return DBCbsdBuilder(). \
with_state(self.unregistered). \
with_registration('some'). \
with_eirp_capabilities(0, 10, 20, 1). \
with_active_mode_config(self.registered)
@staticmethod
def _prepare_base_active_mode_config() -> ActiveModeConfigBuilder:
return ActiveModeConfigBuilder(). \
with_state(Unregistered). \
with_registration('some'). \
with_eirp_capabilities(0, 10, 20, 1). \
with_desired_state(Registered)