def setUp(self):
self.headers = [
u'Code', u'Name', u'City', u'State', u'State Name', u'Latitude',
u'Longitude', u'Country', u'Country Name', u'Global Region',
u'WAC', u'Notes'
]
self.row = [
u'AAA', u'Anaa', u'Anaa', u'', u'', u'-17.351700', u'-145.497800',
u'PF', u'French Polynesia', u'Australasia', u'823', u''
]
# setting a required field to empty string
self.invalid_row = [
u'', u'Anaa', u'Anaa', u'', u'', u'-17.351700', u'-145.497800',
u'PF', u'French Polynesia', u'Australasia', u'823', u''
]
provider_type = DiioAirportType()
self.provider_map = provider_type.map
self.collection_name = provider_type.collection_name
self.mongo_connection = mongomock.MongoClient()
self.valid_obj = AirportRecord(self.headers, self.provider_map,
self.collection_name, 300,
self.mongo_connection)
self.invalid_obj = AirportRecord(None, self.provider_map,
self.collection_name, None,
self.mongo_connection)
class TestGritsAirportRecord(unittest.TestCase):
def setUp(self):
self.headers = [u'Code', u'Name', u'City', u'State', u'State Name',
u'Latitude', u'Longitude', u'Country', u'Country Name',
u'Global Region', u'WAC', u'Notes']
self.row = [u'AAA', u'Anaa', u'Anaa', u'', u'', u'-17.351700',
u'-145.497800', u'PF', u'French Polynesia', u'Australasia',
u'823', u'']
# setting a required field to empty string
self.invalid_row = [u'', u'Anaa', u'Anaa', u'', u'', u'-17.351700',
u'-145.497800', u'PF', u'French Polynesia', u'Australasia',
u'823', u'']
provider_type = DiioAirportType()
self.provider_map = provider_type.map
self.collection_name = provider_type.collection_name
self.mongo_connection = mongomock.MongoClient()
self.valid_obj = AirportRecord(self.headers, self.provider_map, self.collection_name, 300, self.mongo_connection)
self.invalid_obj = AirportRecord(None, self.provider_map, self.collection_name, None, self.mongo_connection)
def test_create_invalid_obj(self):
self.assertRaises(InvalidRecordProperty, self.invalid_obj.create, self.row)
def test_validate_invalid_obj(self):
self.assertEquals(False, self.invalid_obj.validate())
def test_create_valid_obj_invalid_row(self):
self.valid_obj.create(self.invalid_row)
self.assertEquals(False, self.valid_obj.validate())
def test_create_valid_obj_is_valid(self):
self.valid_obj.create(self.row)
self.assertEquals(True, self.valid_obj.validate())
def setUp(self):
self.headers = [u'Code', u'Name', u'City', u'State', u'State Name',
u'Latitude', u'Longitude', u'Country', u'Country Name',
u'Global Region', u'WAC', u'Notes']
self.row = [u'AAA', u'Anaa', u'Anaa', u'', u'', u'-17.351700',
u'-145.497800', u'PF', u'French Polynesia', u'Australasia',
u'823', u'']
# setting a required field to empty string
self.invalid_row = [u'', u'Anaa', u'Anaa', u'', u'', u'-17.351700',
u'-145.497800', u'PF', u'French Polynesia', u'Australasia',
u'823', u'']
provider_type = DiioAirportType()
self.provider_map = provider_type.map
self.collection_name = provider_type.collection_name
self.mongo_connection = mongomock.MongoClient()
self.valid_obj = AirportRecord(self.headers, self.provider_map, self.collection_name, 300, self.mongo_connection)
self.invalid_obj = AirportRecord(None, self.provider_map, self.collection_name, None, self.mongo_connection)
class TestGritsAirportRecord(unittest.TestCase):
def setUp(self):
self.headers = [
u'Code', u'Name', u'City', u'State', u'State Name', u'Latitude',
u'Longitude', u'Country', u'Country Name', u'Global Region',
u'WAC', u'Notes'
]
self.row = [
u'AAA', u'Anaa', u'Anaa', u'', u'', u'-17.351700', u'-145.497800',
u'PF', u'French Polynesia', u'Australasia', u'823', u''
]
# setting a required field to empty string
self.invalid_row = [
u'', u'Anaa', u'Anaa', u'', u'', u'-17.351700', u'-145.497800',
u'PF', u'French Polynesia', u'Australasia', u'823', u''
]
provider_type = DiioAirportType()
self.provider_map = provider_type.map
self.collection_name = provider_type.collection_name
self.mongo_connection = mongomock.MongoClient()
self.valid_obj = AirportRecord(self.headers, self.provider_map,
self.collection_name, 300,
self.mongo_connection)
self.invalid_obj = AirportRecord(None, self.provider_map,
self.collection_name, None,
self.mongo_connection)
def test_create_invalid_obj(self):
self.assertRaises(InvalidRecordProperty, self.invalid_obj.create,
self.row)
def test_validate_invalid_obj(self):
self.assertEquals(False, self.invalid_obj.validate())
def test_create_valid_obj_invalid_row(self):
self.valid_obj.create(self.invalid_row)
self.assertEquals(False, self.valid_obj.validate())
def test_create_valid_obj_is_valid(self):
self.valid_obj.create(self.row)
self.assertEquals(True, self.valid_obj.validate())
