def test_should_confirm_receive_billing_confirm_receive(self):
naruto = User(uid=str(uuid.uuid4()))
billing = Billing(title='Gas Station',
description='Yesterday',
value=Decimal(10),
work_date=yesterday)
naruto.confirm_receive_billing(billing)
self.assertIsInstance(billing.receive_date, datetime)
def test_should_add_billing(self):
naruto = User(uid=str(uuid.uuid4()))
billing = Billing(title='Gas Station',
description='Yesterday',
value=Decimal(10),
work_date=yesterday)
naruto.add_billing(billing)
self.assertEqual(1, len(naruto.billing))
def test_should_save_billing_update_billing_when_exists(self):
yesterday = datetime.now() + timedelta(days=1)
billing_table = BillingTable(uid=str(uuid.uuid4()),
title='Gas Station',
description='Yesterday',
value=Decimal(10),
work_date=yesterday)
self._session.add(billing_table)
billing = Billing(title='Gas Station',
description='Yesterday',
value=Decimal(10),
work_date=yesterday,
uid=billing_table.uid)
naruto = User()
self._user_repository.save_billing(naruto, billing)
result = len(self._session.query(BillingTable).all())
expected = 1
self.assertEqual(expected, result)
def get_by_uid(self, uid):
user_table = self._uow.session.query(UserTable).filter(
UserTable.uid == uid).first()
if not user_table:
return
return User(uid=user_table.uid)
def test_should_get_billing_by_uid_return_none_when_user_not_exists(self):
user_uid = str(uuid.uuid4())
billing_uid = str(uuid.uuid4())
user = User(user_uid)
billing = self._user_repository.get_billing_by_uid(user, billing_uid)
self.assertIsNone(billing)
def test_should_is_valid_false_when_billing_not_exists(self):
self._user_repository.get_billing_by_uid.return_value = None
self._user_repository.get_by_uid.return_value = User()
command = RemoveBillingCommand(user_uid=str(uuid.uuid4()),
billing_uid=str(uuid.uuid4()))
self._use_case.execute(command)
self.assertFalse(self._use_case.is_valid)
def test_should_is_valid_false_when_billing_not_exists(self):
user_repository = Mock()
user_repository.get_by_uid.return_value = User()
user_repository.get_billing_by_uid.return_value = None
command = ConfirmReceiveBillingCommand(user_uid=str(uuid.uuid4()),
billing_uid=str(uuid.uuid4()))
use_case = ConfirmReceiveBilling(user_repository)
use_case.execute(command)
self.assertFalse(use_case.is_valid)
def test_should_is_valid_true_when_cancel_receive_billing(self):
self._user_repository.get_by_uid.return_value = User()
self._user_repository.get_billing_by_uid.return_value = Billing(title='Gas Station',
description='Yesterday',
value=Decimal(10),
work_date=datetime.now())
command = CancelReceiveBillingCommand(user_uid=str(uuid.uuid4()),
billing_uid=str(uuid.uuid4()))
self._use_case.execute(command)
self.assertTrue(self._use_case.is_valid)
def test_should_is_valid_false_when_billing_not_exists(self):
user_repository = Mock()
user_repository.get_by_uid.return_value = User()
user_repository.get_billing_by_uid.return_value = None
command = UpdateBillingCommand(user_uid=str(uuid.uuid4()),
billing_uid=str(uuid.uuid4()),
title='Gas Station',
description='Yesterday',
value=Decimal(10),
work_date=datetime.now())
use_case = UpdateBilling(user_repository)
use_case.execute(command)
self.assertFalse(use_case.is_valid)
def test_should_is_valid_false_when_billing_added_not_is_valid(self):
user_repository = Mock()
user_repository.get_by_uid.return_value = User(uid=str(uuid.uuid4()))
yesterday = datetime.now() + timedelta(days=1)
command = AddBillingCommand(user_uid=str(uuid.uuid4()),
title='',
description='Yesterday',
value=Decimal(10),
work_date=yesterday)
use_case = AddBilling(user_repository)
use_case.execute(command)
self.assertFalse(use_case.is_valid)
def test_should_get_billing_by_uid_return_billing_when_user_exists(self):
user_uid = str(uuid.uuid4())
billing_uid = str(uuid.uuid4())
billing_table = BillingTable(uid=billing_uid,
title='',
description='',
value=Decimal(10),
work_date=datetime.now(),
user_uid=user_uid)
self._session.add(billing_table)
user = User(user_uid)
billing = self._user_repository.get_billing_by_uid(user, billing_uid)
self.assertIsInstance(billing, Billing)
