thermokarst/ccdb-api
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add collection-species to data import and tests

Browse source
This commit is contained in:
Matthew Ryan Dillon 2016-06-14 21:19:02 -07:00
parent 1b5cf2ebea
commit 067448943e
3 changed files with 73 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

19
ccdb/species/tests/test_models.py
View file

@ -1,8 +1,8 @@
from django.test import TestCase
from django.db import IntegrityError, transaction
from ..models import Species
from ..factories import SpeciesFactory
from ..models import Species, CollectionSpecies
from ..factories import SpeciesFactory, CollectionSpeciesFactory
class SpeciesTestCase(TestCase):
@ -17,3 +17,18 @@ class SpeciesTestCase(TestCase):
SpeciesFactory(common_name=s1.common_name, species=s1.species)
s3 = SpeciesFactory()
self.assertTrue(isinstance(s3, Species))
class CollectionSpeciesTestCase(TestCase):
def test_creation(self):
c = CollectionSpeciesFactory()
self.assertTrue(isinstance(c, CollectionSpecies))
label = "{} {}".format(c.collection, c.species)
self.assertEqual(c.__str__(), label)
def test_uniqueness(self):
c1 = CollectionSpeciesFactory()
with transaction.atomic(), self.assertRaises(IntegrityError):
CollectionSpeciesFactory(collection=c1.collection, species=c1.species)
c3 = CollectionSpeciesFactory()
self.assertTrue(isinstance(c3, CollectionSpecies))