def test_member_not_found(self, ctx,
mock_member_repository: MemberRepository,
member_manager: MemberManager):
# Given that the database cannot find the specified member.
mock_member_repository.update_member = MagicMock(
side_effect=MemberNotFoundError)
with raises(MemberNotFoundError):
member_manager.new_membership(ctx, TEST_USERNAME, 1, 'card')
def test_invalid_duration(self, ctx, mock_member_repository: MagicMock,
mock_membership_repository: MagicMock,
member_manager: MemberManager):
# When...
with raises(IntMustBePositive):
member_manager.new_membership(ctx, TEST_USERNAME, -1,
'bank_cheque')
# Expect that the database has not been touched.
mock_member_repository.update_member.assert_not_called()
mock_membership_repository.create_membership.assert_not_called()
def test_no_price_for_duration(self, ctx,
mock_member_repository: MagicMock,
mock_membership_repository: MagicMock,
member_manager: MemberManager):
# When...
with raises(NoPriceAssignedToThatDuration):
member_manager.new_membership(ctx, TEST_USERNAME, 123456789,
'cash')
# Expect that the database has not been touched.
mock_member_repository.update_member.assert_not_called()
mock_membership_repository.create_membership.assert_not_called()
def test_not_found(self, ctx, mock_member_repository: MagicMock,
member_manager: MemberManager):
# Given...
mock_member_repository.search_member_by = MagicMock(return_value=([],
0))
# When...
with raises(MemberNotFoundError):
member_manager.get_by_username(ctx, TEST_USERNAME)
# Expect...
mock_member_repository.search_member_by.assert_called_once_with(
ctx, username=TEST_USERNAME)
def test_happy_path(self, ctx, mock_member_repository: MagicMock,
sample_member: Member, member_manager: MemberManager):
# Given...
mock_member_repository.search_member_by = MagicMock(
return_value=([sample_member], 1))
# When...
test_terms = 'somthing to serach'
test_offset = 42
test_limit = 99
result, count = member_manager.search(ctx,
limit=test_limit,
offset=test_offset,
room_number=TEST_ROOM_NUMBER,
terms=test_terms)
# Expect...
assert [sample_member] == result
# Make sure that all the parameters are passed to the DB.
mock_member_repository.search_member_by.assert_called_once_with(
ctx,
limit=test_limit,
offset=test_offset,
room_number=TEST_ROOM_NUMBER,
terms=test_terms)
def test_fail_to_create_transaction(
self, ctx, mock_membership_repository: MembershipRepository,
mock_member_repository: MemberRepository,
mock_money_repository: MoneyRepository, exception: Exception,
member_manager: MemberManager):
mock_member_repository.update_member = MagicMock()
mock_money_repository.add_member_payment_record = MagicMock(
side_effect=exception)
# When...
with raises(exception):
member_manager.new_membership(ctx,
TEST_USERNAME,
1,
'cash',
start_str=TEST_DATE1.isoformat())
# Expect...
mock_member_repository.update_member.assert_not_called()
def test_happy_path(self, ctx, mock_member_repository: MagicMock,
sample_member: Member, member_manager: MemberManager):
# Given...
mock_member_repository.search_member_by = MagicMock(
return_value=([sample_member], 1))
# When...
result = member_manager.get_by_username(ctx, TEST_USERNAME)
# Expect...
assert sample_member == result
mock_member_repository.search_member_by.assert_called_once_with(
ctx, username=TEST_USERNAME)
def test_invalid_offset(self, ctx, member_manager: MemberManager):
# When...
with raises(IntMustBePositive):
member_manager.search(ctx, limit=10, offset=-1)
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