def setUp(self):
self.client = TestClient()
self.client.name = "Azamat Tokhtaev"
self.client.save()
self.service = TestService()
self.service.name = "NEO Service"
self.service.save()
self.storage = DatabaseTransactionStorage()
class TestDatabaseStorage(TestCase):
client = None
service = None
storage = None
@classmethod
def setUpClass(cls):
from django.conf import settings
settings.INSTALLED_APPS+=('ledger.tests',)
from django.core.management import call_command
from django.db.models import loading
loading.cache.loaded = False
call_command('syncdb', verbosity=0)
def setUp(self):
self.client = TestClient()
self.client.name = "Azamat Tokhtaev"
self.client.save()
self.service = TestService()
self.service.name = "NEO Service"
self.service.save()
self.storage = DatabaseTransactionStorage()
def test_save_transaction_to_db(self):
deposit_transaction = DepositTransaction()
deposit_transaction.agent_from = self.client
deposit_transaction.agent_to = self.service
deposit_transaction.amount = 200
deposit_transaction.batch_id = 'custom_batch_id'
self.storage.save_transaction(deposit_transaction)
db_transaction = Transaction.objects.get(agent_from_id=self.client.pk, agent_from_content_type=ContentType.objects.get_for_model(self.client))
self.assertEqual(db_transaction.agent_from, self.client)
self.assertEqual(db_transaction.agent_to, self.service)
self.assertEqual(1, len(self.storage.get_transactions_from(self.client)))
self.assertEqual(1, len(self.storage.get_transactions_to(self.service)))
def test_get_deposit_transactions(self):
deposit_transaction = DepositTransaction()
deposit_transaction.agent_from = self.client
deposit_transaction.agent_to = self.service
deposit_transaction.amount = 1000
deposit_transaction.batch_id = 'custom_batch_id'
self.storage.save_transaction(deposit_transaction)
transaction = self.storage.filter(self.storage.get_transactions_from(self.client), TRANSACTION_DEPOSIT)[0]
self.assertEqual(1000, transaction.amount)
def test_sum(self):
deposit_transaction = DepositTransaction()
deposit_transaction.agent_from = self.client
deposit_transaction.agent_to = self.service
deposit_transaction.amount = 1000
deposit_transaction.batch_id = 'custom_batch_id'
self.storage.save_transaction(deposit_transaction)
deposit_transaction = DepositTransaction()
deposit_transaction.agent_from = self.client
deposit_transaction.agent_to = self.service
deposit_transaction.amount = 200
deposit_transaction.batch_id = 'custom_batch_id'
self.storage.save_transaction(deposit_transaction)
sum_amount = self.storage.sum(self.storage.filter(self.storage.get_transactions_from(self.client), TRANSACTION_DEPOSIT))
self.assertEqual(1200, sum_amount)
def test_transaction_reason(self):
credit_transaction = CreditTransaction()
credit_transaction.agent_from = self.client
credit_transaction.agent_to = self.service
credit_transaction.amount = 200
credit_transaction.batch_id = 'custom_batch_id'
reason = TestReason()
reason.description = "Here is long and funny description object for the payment"
reason.save()
credit_transaction.reason = reason
self.storage.save_transaction(credit_transaction)
transaction = self.storage.filter(self.storage.get_transactions_from(self.client), TRANSACTION_CREDIT)[0]
self.assertEqual(reason, transaction.reason)