TST: Coverage and adding new model tests

This commit is contained in:
Matthew Ryan Dillon 2016-08-21 15:50:29 -07:00
parent 5c69e774f9
commit 31714215d0
10 changed files with 65 additions and 11 deletions

View file

@ -1,8 +1,9 @@
from django.test import TestCase
from django.db import IntegrityError, transaction
from ..models import Species, CollectionSpecies
from ..factories import SpeciesFactory, CollectionSpeciesFactory
from ..models import Species, TrapSpecies, CollectionSpecies
from ..factories import (SpeciesFactory, TrapSpeciesFactory,
CollectionSpeciesFactory)
class SpeciesTestCase(TestCase):
@ -19,6 +20,14 @@ class SpeciesTestCase(TestCase):
self.assertTrue(isinstance(s3, Species))
class TrapSpeciesTestCase(TestCase):
def test_creation(self):
t = TrapSpeciesFactory()
self.assertTrue(isinstance(t, TrapSpecies))
label = "{} {}".format(t.collection_trap, t.species)
self.assertEqual(t.__str__(), label)
class CollectionSpeciesTestCase(TestCase):
def test_creation(self):
c = CollectionSpeciesFactory()
@ -29,6 +38,7 @@ class CollectionSpeciesTestCase(TestCase):
def test_uniqueness(self):
c1 = CollectionSpeciesFactory()
with transaction.atomic(), self.assertRaises(IntegrityError):
CollectionSpeciesFactory(collection=c1.collection, species=c1.species)
CollectionSpeciesFactory(collection=c1.collection,
species=c1.species)
c3 = CollectionSpeciesFactory()
self.assertTrue(isinstance(c3, CollectionSpecies))