coin/hardware_provisioning/tests.py

from datetime import datetime, timedelta, date

from django.test import TestCase
from django.utils import timezone

from coin.members.models import Member
from .models import Item, ItemType, Loan


def localize(naive_dt):
    if not timezone.is_naive(naive_dt):
        raise ValueError('Expecting a naive datetime')
    else:
        return timezone.make_aware(naive_dt, timezone.get_current_timezone())


class HardwareModelsFactoryMixin:
    def get_item_type(self, **kwargs):
        params = {'name': 'Foos'}
        params.update(**kwargs)
        item_type, _ = ItemType.objects.get_or_create(**kwargs)
        return item_type

    def get_item(self, **kwargs):
        params = {
            'type': self.get_item_type(),
            'designation': 'Test item',
        }
        params.update(**kwargs)
        item, _ = Item.objects.get_or_create(**params)
        return item


class HardwareLoaningTestCase(HardwareModelsFactoryMixin, TestCase):
    def setUp(self):
        self.member = Member.objects.create(username='jdoe')
        self.item = self.get_item()

    def test_running_(self):
        loan_start_date = localize(datetime(2011, 1, 14, 12, 0, 0))
        loan = Loan.objects.create(
            item=self.item, user=self.member,
            loan_date=loan_start_date)

        self.assertEqual(Loan.objects.running().count(), 1)
        self.assertEqual(Loan.objects.finished().count(), 0)
        loan.item.give_back()
        self.assertEqual(Loan.objects.running().count(), 0)
        self.assertEqual(Loan.objects.finished().count(), 1)


class ItemTestCase(HardwareModelsFactoryMixin, TestCase):
    def setUp(self):
        self.member = Member.objects.create(username='jdoe')

        self.free_item = self.get_item(designation='free')
        self.deployed_item = self.get_item(
            designation='deployed', deployed=True)
        self.borrowed_item = self.get_item(designation='borrowed')

    def test_queryset_methods(self):
        self.assertEqual(Item.objects.borrowed().count(), 0)
        self.assertEqual(Item.objects.deployed().count(), 1)
        self.assertEqual(Item.objects.available().count(), 2)
        self.assertEqual(Item.objects.unavailable().count(), 1)

        Loan.objects.create(
            item=self.borrowed_item, user=self.member,
            loan_date=localize(datetime(2011, 1, 14, 12, 0, 0)))

        self.assertEqual(Item.objects.borrowed().count(), 1)
        self.assertEqual(Item.objects.deployed().count(), 1)
        self.assertEqual(Item.objects.available().count(), 1)
        self.assertEqual(Item.objects.unavailable().count(), 2)