def test_delete_culture(self):
"""Ensure culture is successfully deleted."""
add_culture('Pholiota', 'nameko', 'JP', 'PNJP001', 1)
with self.client:
response = self.client.delete(
'/api/v1/users/1/cultures/PNJP001',
data=json.dumps(dict()),
content_type='application/json',
)
data = json.loads(response.data.decode())
self.assertEqual(response.status_code, 200)
self.assertIn('PNJP001 was deleted.', data['message'])
self.assertIn('success', data['status'])
def test_add_culture(self):
culture = add_culture('Grifola', 'frondosa', 'UK', 'GFUK001', 1)
self.assertTrue(culture.id)
self.assertEqual(culture.genus, 'Grifola')
self.assertEqual(culture.species, 'frondosa')
self.assertEqual(culture.strain, 'UK')
self.assertEqual(culture.culture_id, 'GFUK001')
self.assertEqual(culture.user_id, 1)
def test_add_culture_duplicate_culture_id(self):
culture = add_culture('Hypsizygus', 'tesselatus', 'RL', 'HTRL001', 1)
duplicate_culture = Culture(genus='test',
species='testing',
strain='testy',
culture_id='HTRL001',
user_id=2)
db.session.add(duplicate_culture)
self.assertRaises(IntegrityError, db.session.commit)
def test_all_cultures(self):
"""Ensure get all cultures behaves correctly."""
add_culture('Pholiota', 'nameko', 'JP', 'PNJP001', 1)
add_culture('Hypsizygus', 'tesselatus', 'RL', 'HTRL001', 1)
with self.client:
response = self.client.get('/api/v1/users/1/cultures')
data = json.loads(response.data.decode())
self.assertEqual(response.status_code, 200)
self.assertEqual(len(data['data']['cultures']), 2)
self.assertIn('Pholiota', data['data']['cultures'][0]['genus'])
self.assertIn('nameko', data['data']['cultures'][0]['species'])
self.assertIn('JP', data['data']['cultures'][0]['strain'])
self.assertEqual(data['data']['cultures'][0]['user_id'], 1)
self.assertIn('PNJP001', data['data']['cultures'][0]['culture_id'])
self.assertIn('Hypsizygus', data['data']['cultures'][1]['genus'])
self.assertIn('tesselatus', data['data']['cultures'][1]['species'])
self.assertIn('RL', data['data']['cultures'][1]['strain'])
self.assertEqual(data['data']['cultures'][1]['user_id'], 1)
self.assertIn('HTRL001', data['data']['cultures'][1]['culture_id'])
self.assertIn('success', data['status'])
def test_single_culture(self):
"""Ensure get single culture behaves correctly."""
culture = add_culture('Pholiota', 'nameko', 'JP', 'PNJP001', 1)
with self.client:
response = self.client.get(
f'/api/v1/users/1/cultures/{culture.culture_id}')
data = json.loads(response.data.decode())
self.assertEqual(response.status_code, 200)
self.assertIn('Pholiota', data['data']['genus'])
self.assertIn('nameko', data['data']['species'])
self.assertIn('JP', data['data']['strain'])
self.assertIn('PNJP001', data['data']['culture_id'])
self.assertEqual(data['data']['user_id'], 1)
self.assertIn('success', data['status'])