def test_get_extract_by_odes(self):
db = Mock()
db.fetchone.return_value = ('123', 'XYZ', '456', [1, 2, 3, 4], '7', 8,
'Oakland', 85921881, None)
odes = data.ODES('7')
extract = data.get_extract(db, odes=odes)
self.assertEqual(db.mock_calls[0][0], 'execute')
self.assertEqual(db.mock_calls[0][1][1], ('7', ))
self.assertEqual(
db.mock_calls[0][1][0], '''
SELECT id, name, envelope_id, envelope_bbox, odes_id, user_id, wof_name, wof_id, created
FROM extracts WHERE odes_id = %s
''')
self.assertEqual(extract.id, '123')
self.assertEqual(extract.name, 'XYZ')
self.assertEqual(id(extract.odes), id(odes))
self.assertEqual(extract.user_id, 8)
self.assertEqual(extract.envelope.id, '456')
self.assertEqual(extract.envelope.bbox, [1, 2, 3, 4])
self.assertEqual(extract.wof.id, 85921881)
self.assertEqual(extract.wof.name, 'Oakland')
with self.assertRaises(AssertionError) as bad_id:
odes2 = data.ODES(7)
def test_set_extract(self):
db, name = Mock(), str(uuid4())
envelope = data.Envelope('xyz',
[-122.26447, 37.79724, -122.24825, 37.81230])
wof = data.WoF(85921881, 'Oakland')
odes = data.ODES('4')
extract = data.Extract('123', name, envelope, odes, 5, None, wof)
data.set_extract(db, extract)
self.assertEqual(db.mock_calls[0][0], 'execute')
self.assertEqual(
db.mock_calls[0][1][1],
(name, 'xyz', [-122.26447, 37.79724, -122.24825, 37.8123
], '4', 5, 'Oakland', 85921881, '123'))
self.assertEqual(
db.mock_calls[0][1][0], '''
UPDATE extracts
SET name = %s, envelope_id = %s, envelope_bbox = %s, odes_id = %s,
user_id = %s, wof_name = %s, wof_id = %s
WHERE id = %s
''')