def setUp(self):
# Create and record some data
self.result = Result.objects.all()[0]
self.pathogenicity = Pathogenicity(name='pathogenic')
self.pathogenicity.save()
self.parental_result = ParentalResult(name='heterozygous')
self.parental_result.save()
self.category = AssessmentCategory(name='other')
self.category.save()
super(AssessmentResourceTestCase, self).setUp()
def setUp(self):
super(DeleteTestCase, self).setUp()
# Immediately validates and creates a sample
management.call_command('samples', 'queue', startworkers=True)
# Create and record some data that will be used to create knowledge
# capture assessments later on.
self.pathogenicity = Pathogenicity(name='pathogenic')
self.pathogenicity.save()
self.parental_result = ParentalResult(name='heterozygous')
self.parental_result.save()
self.category = AssessmentCategory(name='other')
self.category.save()
self.user = User.objects.all()[0]
def test_put(self):
assessment = Assessment(sample_result=self.result,
user=self.user,
assessment_category=self.category,
sanger_requested=False,
pathogenicity=self.pathogenicity,
father_result=self.parental_result,
mother_result=self.parental_result)
assessment.save()
# Create a new category to change our assessment
new_category = AssessmentCategory(name='dangerous')
new_category.save()
assessment_obj = {
'sample_result': assessment.sample_result.id,
'user': assessment.user.id,
'assessment_category': new_category.id,
'pathogenicity': assessment.pathogenicity.id,
'father_result': assessment.father_result.id,
'mother_result': assessment.mother_result.id,
'sanger_requested': assessment.sanger_requested,
}
# Try to change the assessment object's category field
response = self.client.put('/api/assessments/{0}/'.format(
assessment.id),
data=json.dumps(assessment_obj),
content_type='application/json',
HTTP_ACCEPT='application/json')
self.assertEqual(response.status_code, codes.ok)
# Confirm that the category changed
assessment = Assessment.objects.get(id=assessment.id)
self.assertEqual(assessment.assessment_category.name, 'dangerous')
# Make sure an incomplete form causes an error
response = self.client.put(
'/api/assessments/{0}/'.format(assessment.id),
data=json.dumps({'user': assessment.user.id}),
content_type='application/json',
HTTP_ACCEPT='application/json')
self.assertEqual(response.status_code, codes.unprocessable_entity)
