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)