def report(args):
service = AdService()
reports = service.report(
client=args.client,
start=args.start,
end=args.end,
)
print_report(reports)
def add(args):
service = AdService()
service.add(
name=args.name,
client=args.client,
start=args.start,
end=args.end,
investment=int(args.investment * 100),
)
def test_client_is_str(self, ad_input):
service = AdService()
with pytest.raises(SchemaError) as exception:
service.add(
name=ad_input.name,
client=31,
start=ad_input.start,
end=ad_input.end,
investment=ad_input.investment,
)
assert 'client must be str' in [err for err in exception.value.errors]
def test_end_is_formated_str(self, ad_input):
service = AdService()
with pytest.raises(SchemaError) as exception:
service.add(
name=ad_input.name,
client=ad_input.client,
start=ad_input.start,
end=31,
investment=ad_input.investment,
)
print(exception.value.autos)
assert 'end date must be str' in [
err for err in exception.value.errors
]
with pytest.raises(SchemaError) as exception:
service.add(
name=ad_input.name,
client=ad_input.client,
start=ad_input.start,
end='',
investment=ad_input.investment,
)
assert 'end date must be in the format dd-mm-yyyy' in [
err for err in exception.value.errors
]
def test_investment_is_positive_int(self, ad_input):
service = AdService()
with pytest.raises(SchemaError) as exception:
service.add(
name=ad_input.name,
client=ad_input.client,
start=ad_input.start,
end=ad_input.end,
investment='500',
)
assert 'investment must be int' in [
err for err in exception.value.errors
]
with pytest.raises(SchemaError) as exception:
service.add(
name=ad_input.name,
client=ad_input.client,
start=ad_input.start,
end=ad_input.end,
investment=-500,
)
assert 'investment must be positive' in [
err for err in exception.value.errors
]
def test_report_service(self, report_input, report_dto, mocker):
report_input_dto = mocker.patch(
'src.services.adservice.ReportInputDTO')
adreg_mock = mocker.patch('src.services.adservice.AdReg')
reports = [report_dto for i in range(5)]
adreg_mock().create_report.return_value = reports
service = AdService()
result = service.report(
client=report_input.client,
start='5-4-2021',
end='7-5-2021',
)
report_input_dto.assert_called_once_with(
client=report_input.client,
start=date(2021, 4, 5),
end=date(2021, 5, 7),
)
adreg_mock().create_report.assert_called_once_with(report_input_dto())
assert result == reports
def test_passes_correct_values(self, ad_input, mocker):
mocker.patch('src.services.adservice.AdReg')
ad_input_dto = mocker.patch('src.services.adservice.AdInputDTO')
service = AdService()
service.add(
name=ad_input.name,
client=ad_input.client,
start='5-4-2021',
end='7-5-2021',
investment=ad_input.investment,
)
ad_input_dto.assert_called_once_with(
name=ad_input.name,
client=ad_input.client,
start=date(2021, 4, 5),
end=date(2021, 5, 7),
investment=ad_input.investment,
)
adreg = src.services.adservice.AdReg()
adreg.create_adentry.assert_called_once_with(ad_input_dto())
def test_add_without_args(self):
service = AdService()
with pytest.raises(SchemaError):
service.add()