def test_update(self):
self.dbu.create_some_test_well_royalty_masters()
# change all types of attributes, read another record and then read the record again to make sure
# the changes were made.
well = self.db.select('WellRoyaltyMaster', ID=2)
well[0].WellEvent = 'Changed'
well[0].LeaseID = 100
well[0].CommencementDate = '2016-02-01 00:00:00'
well[0].WellType = None
self.db.update(well[0])
well = self.db.select('WellRoyaltyMaster', ID=1)
self.assertEqual(well[0].ID, 1)
self.assertEqual(well[0].WellEvent, 'SKWI111062705025W300')
well = self.db.select('WellRoyaltyMaster', ID=2)
self.assertEqual(well[0].ID, 2)
self.assertEqual(well[0].WellEvent, 'Changed')
self.assertEqual(well[0].CommencementDate, datetime(2016, 2, 1, 0, 0))
self.assertEqual(well[0].WellType, None)
ds = DataStructure()
self.assertRaises(AttributeError, self.db.update, ds)
ds._table_name = 'WellRoyaltyMaster'
self.assertRaises(AttributeError, self.db.update, ds)
ds.ID = 100
self.assertRaises(AppError, self.db.update, ds)
def test_update(self):
self.dbu.create_some_test_well_royalty_masters()
# change all types of attributes, read another record and then read the record again to make sure
# the changes were made.
well = self.db.select("WellRoyaltyMaster", ID=2)
well[0].WellEvent = "Changed"
well[0].LeaseID = 100
well[0].CommencementDate = "2016-02-01 00:00:00"
well[0].WellType = None
self.db.update(well[0])
well = self.db.select("WellRoyaltyMaster", ID=1)
self.assertEqual(well[0].ID, 1)
self.assertEqual(well[0].WellEvent, "SKWI111062705025W300")
well = self.db.select("WellRoyaltyMaster", ID=2)
self.assertEqual(well[0].ID, 2)
self.assertEqual(well[0].WellEvent, "Changed")
self.assertEqual(well[0].CommencementDate, datetime(2016, 2, 1, 0, 0))
self.assertEqual(well[0].WellType, None)
ds = DataStructure()
self.assertRaises(AttributeError, self.db.update, ds)
ds._table_name = "WellRoyaltyMaster"
self.assertRaises(AttributeError, self.db.update, ds)
ds.ID = 100
self.assertRaises(AppError, self.db.update, ds)
def get_data_structure(self, table_name):
""" This method must be called to create a valid database data structure. """
ds = DataStructure()
ds._table_name = table_name
return ds
def get_data_structure(self, table_name):
""" This method must be called to create a valid database data structure. """
ds = DataStructure()
ds._table_name = table_name
return ds
