def test_port_not_found(self, ctx, mock_port_repository: PortRepository,
port_manager: PortManager):
# Given...
mock_port_repository.delete_port = MagicMock(
side_effect=PortNotFoundError)
# When...
with raises(PortNotFoundError):
port_manager.delete(ctx, 'portID')
def test_happy_path(self, ctx, mock_port_repository: PortRepository,
port_manager: PortManager):
# Given...
mock_port_repository.delete_port = MagicMock()
# When...
port_manager.delete(ctx, 'portID')
# Expect...
mock_port_repository.delete_port.assert_called_once_with(ctx, 'portID')
def test_happy_path(self, ctx, mock_port_repository: PortRepository,
sample_mutation_req: MutationRequest,
port_manager: PortManager):
# Given...
mock_port_repository.create_port = MagicMock()
# When...
port_manager.create(ctx, sample_mutation_req)
# Expect...
args = asdict(sample_mutation_req)
mock_port_repository.create_port.assert_called_once_with(ctx, **args)
def test_unknown_room(self, ctx, mock_port_repository: PortRepository,
sample_mutation_req: MutationRequest,
port_manager: PortManager):
# Given...
mock_port_repository.create_port = MagicMock(
side_effect=RoomNotFoundError)
# When...
with raises(RoomNotFoundError):
port_manager.create(ctx, sample_mutation_req)
# Expect..
mock_port_repository.create_port.assert_called_once()
def test_invalid_mutation_request(self, ctx,
mock_port_repository: PortRepository,
field: str, value,
sample_mutation_req: MutationRequest,
port_manager: PortManager):
# Given...
mock_port_repository.create_port = MagicMock()
req = MutationRequest(**{
**asdict(sample_mutation_req),
**{
field: value
}
})
# When...
with raises(ValueError):
port_manager.create(ctx, req)
# Expect...
mock_port_repository.create_port.assert_not_called()
def test_happy_path(self, ctx, mock_port_repository: PortRepository,
sample_port: Port, port_manager: PortManager):
mock_port_repository.search_port_by = MagicMock(
return_value=([sample_port], 1))
result, count = port_manager.search(ctx,
limit=DEFAULT_LIMIT,
offset=DEFAULT_OFFSET,
port_id='port',
switch_id='switch',
room_number='room',
terms='terms')
assert [sample_port] == result
assert 1 == count
mock_port_repository.search_port_by.assert_called_once_with(
ctx,
limit=DEFAULT_LIMIT,
offset=DEFAULT_OFFSET,
port_id='port',
switch_id='switch',
room_number='room',
terms='terms')
def init(testing=True):
"""
Initialize and wire together the dependency of the application.
"""
if testing:
configuration = TEST_CONFIGURATION
else:
configuration = CONFIGURATION
Database.init_db(configuration.DATABASE, testing=testing)
# Repositories:
ping_repository = PingSQLRepository()
member_sql_repository = MemberSQLRepository()
network_object_sql_repository = NetworkObjectSQLRepository()
device_sql_repository = DeviceSQLRepository()
room_sql_repository = RoomSQLRepository()
elk_repository = ElasticSearchRepository(configuration)
money_repository = MoneySQLRepository()
switch_network_manager = SwitchSNMPNetworkManager()
account_sql_repository = AccountSQLRepository()
product_sql_repository = ProductSQLRepository()
payment_method_sql_repository = PaymentMethodSQLRepository()
transaction_sql_repository = TransactionSQLRepository()
account_type_sql_repository = AccountTypeSQLRepository()
# Managers
health_manager = HealthManager(ping_repository)
switch_manager = SwitchManager(
switch_repository=network_object_sql_repository, )
port_manager = PortManager(port_repository=network_object_sql_repository, )
device_manager = DeviceManager(
device_repository=device_sql_repository,
member_repository=member_sql_repository,
room_repository=room_sql_repository,
vlan_repository=network_object_sql_repository,
ip_allocator=device_sql_repository,
)
member_manager = MemberManager(
member_repository=member_sql_repository,
money_repository=money_repository,
membership_repository=member_sql_repository,
logs_repository=elk_repository,
configuration=configuration,
)
room_manager = RoomManager(room_repository=room_sql_repository, )
account_manager = AccountManager(
account_repository=account_sql_repository,
member_repository=member_sql_repository,
)
product_manager = ProductManager(
product_repository=product_sql_repository, )
payment_method_manager = PaymentMethodManager(
payment_method_repository=payment_method_sql_repository)
transaction_manager = TransactionManager(
transaction_repository=transaction_sql_repository, )
account_type_manager = AccountTypeManager(
account_type_repository=account_type_sql_repository)
# HTTP Handlers:
health_handler = HealthHandler(health_manager)
transaction_handler = TransactionHandler(transaction_manager)
member_handler = MemberHandler(member_manager)
device_handler = DeviceHandler(device_manager)
room_handler = RoomHandler(room_manager)
switch_handler = SwitchHandler(switch_manager)
port_handler = PortHandler(port_manager, switch_manager,
switch_network_manager)
temporary_account_handler = TemporaryAccountHandler()
account_type_handler = AccountTypeHandler(account_type_manager)
payment_method_handler = PaymentMethodHandler(payment_method_manager)
account_handler = AccountHandler(account_manager)
product_handler = ProductHandler(product_manager)
# Connexion will use this function to authenticate and fetch the information of the user.
if os.environ.get('TOKENINFO_FUNC') is None:
os.environ[
'TOKENINFO_FUNC'] = 'src.interface_adapter.http_api.auth.token_info'
app = connexion.FlaskApp(__name__, specification_dir='openapi')
app.add_api(
'swagger.yaml',
# resolver=RestyResolver('src.interface_adapter.http_api'),
resolver=ADHResolver({
'health': health_handler,
'transaction': transaction_handler,
'member': member_handler,
'device': device_handler,
'room': room_handler,
'switch': switch_handler,
'port': port_handler,
'temporary_account': temporary_account_handler,
'account_type': account_type_handler,
'payment_method': payment_method_handler,
'account': account_handler,
'product': product_handler,
}),
validate_responses=True,
strict_validation=True,
pythonic_params=True,
auth_all_paths=True,
)
app.app.config.update(configuration.API_CONF)
return app
def test_invalid_limit(self, ctx, port_manager: PortManager):
with raises(IntMustBePositive):
port_manager.search(ctx, limit=-1)
def port_manager(mock_port_repository, ):
return PortManager(port_repository=mock_port_repository)