def test_can_handle_multiple_records(self, mock: MagicMock):
mock.return_value.fetchall.return_value = [
create_productline_db_record('record1'),
create_productline_db_record('record2'),
]
records = productlines.get_all()
self.assertEqual(['record1', 'record2'], list(map(lambda p: p.productline_id, records)))
def list_productlines():
productlines = _productlines.get_all()
return flask.jsonify({
'productlines': {
'type': 'FeatureCollection',
'features': [p.serialize() for p in productlines],
},
})
def test_gracefully_handles_db_errors(self, mock: MagicMock):
mock.side_effect = helpers.create_database_error()
with self.assertRaises(DatabaseError):
productlines.get_all()
def test_can_handle_empty_recordset(self, mock: MagicMock):
mock.return_value.fetchall.return_value = []
self.assertEqual([], productlines.get_all())
def test_assigns_correct_stop_date(self, mock):
mock.return_value.fetchall.return_value = [create_productline_db_record()]
record = productlines.get_all().pop()
self.assertEqual(DATE_STOP, record.stop_on)
def test_assigns_correct_spatial_filter_id(self, mock):
mock.return_value.fetchall.return_value = [create_productline_db_record()]
record = productlines.get_all().pop()
self.assertEqual('test-spatial-filter-id', record.spatial_filter_id)
def test_assigns_correct_owner(self, mock):
mock.return_value.fetchall.return_value = [create_productline_db_record()]
record = productlines.get_all().pop()
self.assertEqual('test-productline-owner', record.owned_by)
def test_assigns_correct_name(self, mock):
mock.return_value.fetchall.return_value = [create_productline_db_record()]
record = productlines.get_all().pop()
self.assertEqual('test-name', record.name)
def test_assigns_correct_max_cloud_cover(self, mock):
mock.return_value.fetchall.return_value = [create_productline_db_record()]
record = productlines.get_all().pop()
self.assertEqual(42, record.max_cloud_cover)
def test_assigns_correct_created_on(self, mock):
mock.return_value.fetchall.return_value = [create_productline_db_record()]
record = productlines.get_all().pop()
self.assertEqual(datetime.utcnow().date(), record.created_on.date())
def test_assigns_correct_bbox(self, mock):
mock.return_value.fetchall.return_value = [create_productline_db_record()]
record = productlines.get_all().pop()
self.assertEqual({"type": "Polygon", "coordinates": [[[0, 0], [0, 30], [30, 30], [30, 0], [0, 0]]]},
record.bbox)
def test_assigns_correct_algorithm_name(self, mock):
mock.return_value.fetchall.return_value = [create_productline_db_record()]
records = productlines.get_all()
self.assertEqual('test-algo-name', records[0].algorithm_name)
def test_returns_list_of_productlines(self, mock: MagicMock):
mock.return_value.fetchall.return_value = [create_productline_db_record()]
records = productlines.get_all()
self.assertIsInstance(records, list)
self.assertIsInstance(records[0], productlines.ProductLine)
