def test_insert(self):
self.db_create.well()
well = DataStructure()
well.UWI = 'UWI for this well'
# Should raise this error since we need to get the structure from the database
self.assertRaises(TypeError, self.db.insert)
well = self.db.get_data_structure('Well')
well.UWI = 'UWI for this well'
self.db.insert(well)
self.assertEqual(well.ID, 1)
well = self.db.get_data_structure('Well')
well.UWI = 'Different UWI for this well'
self.db.insert(well)
self.assertEqual(well.ID, 2)
well = self.db.select('Well', ID=1)
self.assertEqual(well[0].ID, 1)
self.assertEqual(well[0].UWI, 'UWI for this well')
well = self.db.get_data_structure('Well')
well.UWI = 'Next Well UWI'
well.ID = 10
self.db.insert(well)
well = self.db.select('Well', ID=1)
self.assertEqual(well[0].ID, 1)
self.assertEqual(well[0].UWI, 'UWI for this well')
well = self.db.select('Well', ID=10)
self.assertEqual(well[0].ID, 10)
self.assertEqual(well[0].UWI, 'Next Well UWI')
well = self.db.get_data_structure('Well')
well.UWI = 'Just One More'
self.db.insert(well)
self.assertEqual(well.ID, 11)
well = self.db.get_data_structure('Well')
well.BadAttr = 'Just another value'
self.assertRaises(AppError, self.db.insert,well)
# if the ID is None,Blank,or zero we shold still be able to insert a record
well = self.db.get_data_structure('Well')
well.ID = None
well.UWI = 'Just One More'
self.db.insert(well)
self.assertEqual(well.ID, 12)
well.ID = 0
self.db.insert(well)
self.assertEqual(well.ID, 13)
well.ID = ''
self.db.insert(well)
self.assertEqual(well.ID, 14)
def test_insert(self):
self.db_create.well()
well = DataStructure()
well.UWI = 'UWI for this well'
# Should raise this error since we need to get the structure from the database
self.assertRaises(TypeError, self.db.insert)
well = self.db.get_data_structure('Well')
well.UWI = 'UWI for this well'
self.db.insert(well)
self.assertEqual(well.ID, 1)
well = self.db.get_data_structure('Well')
well.UWI = 'Different UWI for this well'
self.db.insert(well)
self.assertEqual(well.ID, 2)
well = self.db.select('Well', ID=1)
self.assertEqual(well[0].ID, 1)
self.assertEqual(well[0].UWI, 'UWI for this well')
well = self.db.get_data_structure('Well')
well.UWI = 'Next Well UWI'
well.ID = 10
self.db.insert(well)
well = self.db.select('Well', ID=1)
self.assertEqual(well[0].ID, 1)
self.assertEqual(well[0].UWI, 'UWI for this well')
well = self.db.select('Well', ID=10)
self.assertEqual(well[0].ID, 10)
self.assertEqual(well[0].UWI, 'Next Well UWI')
well = self.db.get_data_structure('Well')
well.UWI = 'Just One More'
self.db.insert(well)
self.assertEqual(well.ID, 11)
well = self.db.get_data_structure('Well')
well.BadAttr = 'Just another value'
self.assertRaises(AppError, self.db.insert, well)
# if the ID is None,Blank,or zero we shold still be able to insert a record
well = self.db.get_data_structure('Well')
well.ID = None
well.UWI = 'Just One More'
self.db.insert(well)
self.assertEqual(well.ID, 12)
well.ID = 0
self.db.insert(well)
self.assertEqual(well.ID, 13)
well.ID = ''
self.db.insert(well)
self.assertEqual(well.ID, 14)
