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)