def test_transfer_to_inn_not_enough_money(self):
data = {
'user_from': self.user,
'inn': '1234567890',
'amount': 5555
}
self.assertEquals(transfer_to_inn(data), 'У пользователя недостаточно средств')
def test_transfer_to_inn_invalid_inn(self):
data = {
'user_from': self.user,
'inn': '0987654321',
'amount': 10
}
self.assertEquals(transfer_to_inn(data), 'Нет пользователей с таким ИНН')
def test_transfer_to_inn_negative_amount(self):
data = {
'user_from': self.user,
'inn': '1234567890',
'amount': -3
}
self.assertEquals(transfer_to_inn(data), 'Введите положительную сумму')
def test_transfer_to_inn_one_inn(self):
data = {
'user_from': self.user,
'inn': '1234567890',
'amount': 10
}
self.assertEquals(transfer_to_inn(data), '10 рублей переведенно одному пользователю')
def test_transfer_to_inn_multiple_inn(self):
self.target_user2 = User.objects.create(username='******')
self.target_user2.profile.inn = '1234567890'
self.target_user2.profile.save()
data = {
'user_from': self.user,
'inn': '1234567890',
'amount': 10
}
self.assertEquals(transfer_to_inn(data), '2 пользователям переведенно по 5.0 рублей')
def home(request):
context = {}
if request.method == 'GET':
transfer_form = TransferForm()
context['transfer_form'] = transfer_form
return render(request, 'core/home.html', context)
else:
transfer_form = TransferForm(request.POST)
if transfer_form.is_valid():
data = transfer_form.cleaned_data
response = transfer_to_inn(data)
context['response'] = response
context['transfer_form'] = transfer_form
return render(request, 'core/home.html', context)