def test_post_empty(self):
self.assertEqual(BankBalance.get_price(1), 40000)
self.assertEqual(BankBalance.get_price(2), 20000)
self.assertEqual(CreditCheckedDate.get_price(1), 30000)
self.assertEqual(CreditCheckedDate.get_price(2), 2000)
self.client.force_login(User.objects.create_user(self.username))
response = self.client.post(reverse('moneybook:now_bank'))
self.assertEqual(response.status_code, 200)
response_json = json.loads(response.content.decode())
self.assertEqual(response_json['balance'],
54054 - (40000 + 20000 - 30000 - 2000))
self.assertEqual(BankBalance.get_price(1), 40000)
self.assertEqual(BankBalance.get_price(2), 20000)
self.assertEqual(CreditCheckedDate.get_price(1), 30000)
self.assertEqual(CreditCheckedDate.get_price(2), 2000)
def test_post_guest(self):
self.assertEqual(BankBalance.get_price(1), 40000)
self.assertEqual(BankBalance.get_price(2), 20000)
self.assertEqual(CreditCheckedDate.get_price(1), 30000)
self.assertEqual(CreditCheckedDate.get_price(2), 2000)
response = self.client.post(reverse('moneybook:now_bank'), {
'bank-1': 50000,
'bank-2': 10000,
'credit-1': 20000,
'credit-2': 3000
})
self.assertEqual(response.status_code, 403)
self.assertEqual(BankBalance.get_price(1), 40000)
self.assertEqual(BankBalance.get_price(2), 20000)
self.assertEqual(CreditCheckedDate.get_price(1), 30000)
self.assertEqual(CreditCheckedDate.get_price(2), 2000)
def get(self, request, *args, **kwargs):
now = datetime.now()
# 全データ
all_data = Data.get_all_data()
# 現在銀行
banks = BankBalance.get_all()
# クレカ確認日
credit_checked_date = CreditCheckedDate.get_all()
today = date.today()
for c in credit_checked_date:
# 日付が過ぎていたらpriceを0にする
if c.date <= today:
c.price = 0
# 銀行残高
all_bank_data = Data.get_bank_data(all_data)
checked_bank_data = Data.get_checked_data(all_bank_data)
bank_written = Data.get_income_sum(
checked_bank_data) - Data.get_outgo_sum(checked_bank_data)
context = {
'year': now.year,
'banks': banks,
'credit_checked_date': credit_checked_date,
'bank_written': bank_written,
}
return render(request, "_several_checked_date.html", context)
def test_post_str_credit(self):
self.assertEqual(BankBalance.get_price(1), 40000)
self.assertEqual(BankBalance.get_price(2), 20000)
self.assertEqual(CreditCheckedDate.get_price(1), 30000)
self.assertEqual(CreditCheckedDate.get_price(2), 2000)
self.client.force_login(User.objects.create_user(self.username))
response = self.client.post(reverse('moneybook:now_bank'), {
'bank-1': 50000,
'bank-2': 10000,
'credit-1': 'a',
'credit-2': 3000
})
self.assertEqual(response.status_code, 400)
self.assertEqual(BankBalance.get_price(1), 40000)
self.assertEqual(BankBalance.get_price(2), 20000)
self.assertEqual(CreditCheckedDate.get_price(1), 30000)
self.assertEqual(CreditCheckedDate.get_price(2), 2000)
def post(self, request, *args, **kwargs):
written_bank_data = Data.get_checked_data(
Data.get_bank_data(Data.get_all_data()))
bank_sum = 0
bb = BankBalance.get_all()
cc = CreditCheckedDate.get_all()
# フォーマットチェック
try:
for b in bb:
key = "bank-" + str(b.pk)
if key in request.POST:
int(request.POST.get(key))
for c in cc:
key = "credit-" + str(c.pk)
if key in request.POST:
int(request.POST.get(key))
except ValueError:
return HttpResponseBadRequest(
json.dumps({"message": "invalid parameter"}))
# 更新と計算
for b in bb:
key = "bank-" + str(b.pk)
if key in request.POST:
value = int(request.POST.get(key))
BankBalance.set(b.pk, value)
bank_sum += BankBalance.get_price(b.pk)
for c in cc:
key = "credit-" + str(c.pk)
if key in request.POST:
value = int(request.POST.get(key))
CreditCheckedDate.set_price(c.pk, value)
bank_sum -= CreditCheckedDate.get_price(c.pk)
return HttpResponse(
json.dumps({
"balance":
Data.get_income_sum(written_bank_data) -
Data.get_outgo_sum(written_bank_data) - bank_sum
}))
def test_set(self):
BankBalance.set(2, 10001)
data = BankBalance.get_all()
self.assertEqual(data[0].name, "三井住友")
self.assertEqual(data[0].price, 10001)
def test_get_price_invalid_pk(self):
self.assertEqual(BankBalance.get_price(10000), 0)
def test_get_price(self):
self.assertEqual(BankBalance.get_price(1), 40000)
self.assertEqual(BankBalance.get_price(2), 20000)
def test_get_all(self):
data = BankBalance.get_all()
self.assertEqual(data[0].name, "三井住友")
self.assertEqual(data[0].price, 20000)
self.assertEqual(data[1].name, "みずほ")
self.assertEqual(data[1].price, 40000)