def test_get_user_transfers_should_not_return_other_user_transfer(self):
transfer = Transfer(name="transfer from user",
amount=123,
date=datetime.date(2018, 11, 20),
from_user=self.other_user,
to_user=self.other_user)
transfer.save()
fetched_transfers = Transfer.get_user_transfers(
2018, 11, self.fetching_user)
self.assertListEqual([t.id for t in fetched_transfers], [])
def test_get_user_transfers_should_not_return_transfer_outside_year_requested(
self):
transfer_before = Transfer(name="transfer from user",
amount=123,
date=datetime.date(2017, 11, 20),
from_user=self.fetching_user,
to_user=self.fetching_user)
transfer_before.save()
transfer_after = Transfer(name="transfer from user",
amount=123,
date=datetime.date(2019, 11, 20),
from_user=self.fetching_user,
to_user=self.fetching_user)
transfer_after.save()
fetched_transfers = Transfer.get_user_transfers(
2018, 11, self.fetching_user)
self.assertListEqual([t.id for t in fetched_transfers], [])
def resolve_transfers(self, info, **kwargs):
year = kwargs.get('year')
month = kwargs.get('month')
return Transfer.get_user_transfers(year, month, info.context.user)