htsworkflow/frontend/samples/tests.py

   1 import datetime
   2 import unittest
   3
   4 try:
   5     import json
   6 except ImportError, e:
   7     import simplejson as json
   8
   9 from django.test import TestCase
  10
  11 from htsworkflow.frontend.samples.models import \
  12         Affiliation, \
  13         ExperimentType, \
  14         Species, \
  15         Library
  16
  17 from htsworkflow.frontend.samples.views import \
  18      library_dict, \
  19      library_json
  20
  21 from htsworkflow.frontend.auth import apidata
  22 from htsworkflow.util.conversion import unicode_or_none
  23
  24
  25 class LibraryTestCase(TestCase):
  26     def setUp(self):
  27         create_db(self)
  28
  29     def testOrganism(self):
  30         self.assertEquals(self.library_10001.organism(), 'human')
  31
  32     def testAffiliations(self):
  33         self.library_10001.affiliations.add(self.affiliation_alice)
  34         self.library_10002.affiliations.add(
  35                 self.affiliation_alice,
  36                 self.affiliation_bob
  37         )
  38         self.failUnless(len(self.library_10001.affiliations.all()), 1)
  39         self.failUnless(self.library_10001.affiliation(), 'Alice')
  40
  41         self.failUnless(len(self.library_10002.affiliations.all()), 2)
  42         self.failUnless(self.library_10001.affiliation(), 'Alice, Bob')
  43
  44
  45 class SampleWebTestCase(TestCase):
  46     """
  47     Test returning data from our database in rest like ways.
  48     (like returning json objects)
  49     """
  50     fixtures = ['test_samples.json']
  51
  52     def test_library_info(self):
  53
  54         for lib in Library.objects.all():
  55             lib_dict = library_dict(lib.id)
  56             url = '/samples/library/%s/json' % (lib.id,)
  57             lib_response = self.client.get(url, apidata)
  58             self.failUnlessEqual(lib_response.status_code, 200)
  59             lib_json = json.loads(lib_response.content)
  60             print lib_json
  61
  62             for d in [lib_dict, lib_json]:
  63                 # amplified_from_sample is a link to the library table,
  64                 # I want to use the "id" for the data lookups not
  65                 # the embedded primary key.
  66                 # It gets slightly confusing on how to implement sending the right id
  67                 # since amplified_from_sample can be null
  68                 #self.failUnlessEqual(d['amplified_from_sample'], lib.amplified_from_sample)
  69                 self.failUnlessEqual(d['antibody_id'], lib.antibody_id)
  70                 self.failUnlessEqual(d['cell_line_id'], lib.cell_line_id)
  71                 self.failUnlessEqual(d['cell_line'], unicode_or_none(lib.cell_line))
  72                 self.failUnlessEqual(d['experiment_type'], lib.experiment_type.name)
  73                 self.failUnlessEqual(d['experiment_type_id'], lib.experiment_type_id)
  74                 self.failUnlessEqual(d['gel_cut_size'], lib.gel_cut_size)
  75                 self.failUnlessEqual(d['hidden'], lib.hidden)
  76                 self.failUnlessEqual(d['id'], lib.id)
  77                 self.failUnlessEqual(d['insert_size'], lib.insert_size)
  78                 self.failUnlessEqual(d['library_name'], lib.library_name)
  79                 self.failUnlessEqual(d['library_species'], lib.library_species.scientific_name)
  80                 self.failUnlessEqual(d['library_species_id'], lib.library_species_id)
  81                 self.failUnlessEqual(d['library_type_id'], lib.library_type_id)
  82                 if lib.library_type_id is not None:
  83                     self.failUnlessEqual(d['library_type'], lib.library_type.name)
  84                 else:
  85                     self.failUnlessEqual(d['library_type'], None)
  86                     self.failUnlessEqual(d['made_for'], lib.made_for)
  87                     self.failUnlessEqual(d['made_by'], lib.made_by)
  88                     self.failUnlessEqual(d['notes'], lib.notes)
  89                     self.failUnlessEqual(d['replicate'], lib.replicate)
  90                     self.failUnlessEqual(d['stopping_point'], lib.stopping_point)
  91                     self.failUnlessEqual(d['successful_pM'], lib.successful_pM)
  92                     self.failUnlessEqual(d['undiluted_concentration'],
  93                                          unicode(lib.undiluted_concentration))
  94                 # some specific tests
  95                 if lib.id == '10981':
  96                     # test a case where there is no known status
  97                     lane_set = {u'status': u'Unknown', u'paired_end': True, u'read_length': 75, u'lane_number': 1, u'flowcell': u'303TUAAXX', u'status_code': None}
  98                     self.failUnlessEqual(len(d['lane_set']), 1)
  99                     self.failUnlessEqual(d['lane_set'][0], lane_set)
 100                 elif lib.id == '11016':
 101                     # test a case where there is a status
 102                     lane_set = {u'status': 'Good', u'paired_end': True, u'read_length': 75, u'lane_number': 5, u'flowcell': u'303TUAAXX', u'status_code': 2}
 103                     self.failUnlessEqual(len(d['lane_set']), 1)
 104                     self.failUnlessEqual(d['lane_set'][0], lane_set)
 105
 106     def test_invalid_library(self):
 107         """
 108         Make sure we get a 404 if we request an invalid library id
 109         """
 110         response = self.client.get('/samples/library/nottheone/json', apidata)
 111         self.failUnlessEqual(response.status_code, 404)
 112
 113
 114     def test_library_no_key(self):
 115         """
 116         Make sure we get a 302 if we're not logged in
 117         """
 118         response = self.client.get('/samples/library/10981/json')
 119         self.failUnlessEqual(response.status_code, 403)
 120         response = self.client.get('/samples/library/10981/json', apidata)
 121         self.failUnlessEqual(response.status_code, 200)
 122
 123 # The django test runner flushes the database between test suites not cases,
 124 # so to be more compatible with running via nose we flush the database tables
 125 # of interest before creating our sample data.
 126 def create_db(obj):
 127     Species.objects.all().delete()
 128     obj.species_human = Species(
 129         scientific_name = 'Homo Sapeins',
 130         common_name = 'human',
 131     )
 132     obj.species_human.save()
 133     obj.species_worm = Species(
 134         scientific_name = 'C. Elegans',
 135         common_name = 'worm',
 136     )
 137     obj.species_worm.save()
 138     obj.species_phix = Species(
 139         scientific_name = 'PhiX',
 140         common_name = 'PhiX'
 141     )
 142     obj.species_phix.save()
 143
 144     ExperimentType.objects.all().delete()
 145     obj.experiment_de_novo = ExperimentType(
 146         name = 'De Novo',
 147     )
 148     obj.experiment_de_novo.save()
 149     obj.experiment_chip_seq = ExperimentType(
 150         name = 'ChIP-Seq'
 151     )
 152     obj.experiment_chip_seq.save()
 153     obj.experiment_rna_seq = ExperimentType(
 154         name = 'RNA-Seq'
 155     )
 156     obj.experiment_rna_seq.save()
 157
 158     Affiliation.objects.all().delete()
 159     obj.affiliation_alice = Affiliation(
 160         name = 'Alice',
 161         contact = 'Lab Boss',
 162         email = 'alice@some.where.else.'
 163     )
 164     obj.affiliation_alice.save()
 165     obj.affiliation_bob = Affiliation(
 166         name = 'Bob',
 167         contact = 'Other Lab Boss',
 168         email = 'bob@some.where.else',
 169     )
 170     obj.affiliation_bob.save()
 171
 172     Library.objects.all().delete()
 173     obj.library_10001 = Library(
 174         id = "10001",
 175         library_name = 'C2C12 named poorly',
 176         library_species = obj.species_human,
 177         experiment_type = obj.experiment_rna_seq,
 178         creation_date = datetime.datetime.now(),
 179         made_for = 'scientist unit 2007',
 180         made_by = 'microfludics system 7321',
 181         stopping_point = '2A',
 182         undiluted_concentration = '5.01',
 183         hidden = False,
 184     )
 185     obj.library_10001.save()
 186     obj.library_10002 = Library(
 187         id = "10002",
 188         library_name = 'Worm named poorly',
 189         library_species = obj.species_human,
 190         experiment_type = obj.experiment_rna_seq,
 191         creation_date = datetime.datetime.now(),
 192         made_for = 'scientist unit 2007',
 193         made_by = 'microfludics system 7321',
 194         stopping_point = '2A',
 195         undiluted_concentration = '5.01',
 196         hidden = False,
 197     )
 198     obj.library_10002.save()
 199