from django.test import TestCase
from django.core.exceptions import ValidationError
from Transaction.models import Transaction, TvaType, DescriptionType

# Create your tests here.
class TestTransaction(TestCase):
	def setUp(self):
		
		self.description = DescriptionType()
		self.description.year = 2015
		self.description.name = 'test'
		self.description.quotity = 0.25
		self.description.save()


		self.tvatype = TvaType()
		self.tvatype.year = 2015
		self.tvatype.description = self.description
		self.tvatype.percent = 0.21
		self.tvatype.save()

		self.Transaction = Transaction()
		self.Transaction.description = self.description
		self.Transaction.tvaType = self.tvatype

	def testCheckAmountsTva(self):
		self.Transaction.amountHTva = 100
		self.Transaction.checkAmounts()

		self.assertEqual(self.Transaction.amountTva, 121)

	def testCheckAmountHtva(self):
		self.Transaction.amountTva = 121
		self.Transaction.checkAmounts()

		self.assertEqual(self.Transaction.amountHTva, 100)

	def testCheckAmountRaise(self):
		with self.assertRaises(Exception):
			self.Transaction.checkAmounts()

	def tearDown(self):
		pass