def test_obj_to_dict(self):
ds = DataStructure()
utils = Utils()
ds.ID = 123
ds.Name = "My Name"
result = utils.obj_to_dict(ds, dict())
self.assertEqual(result['ID'],123)
self.assertEqual(result['Name'],'My Name')
result = utils.obj_to_dict(ds)
self.assertEqual(result['ID'],123)
self.assertEqual(result['Name'],'My Name')
def test_link_update(self):
db = config.get_database()
dbu = DatabaseUtilities()
db_create = DatabaseCreate()
utils = Utils()
dbu.delete_all_tables()
db_create.linktab()
# Test Insert
data = dict()
data["ID"] = '0'
data["TabName"] = 'Lease'
data["AttrName"] = 'ID'
data["LinkName"] = 'Lease'
json_to_browser = json.dumps(data)
resp = self.myapp.post('/data/updateLinkRow.json',
data=json_to_browser)
self.assertEqual(resp.status_code, 200)
data = utils.json_decode(resp)
self.assertEqual(data["StatusCode"], 0)
result = db.select('LinkTab', TabName='Lease', AttrName='ID')
self.assertEqual(len(result), 1)
# Test Update
data = utils.obj_to_dict(result[0])
data['LinkName'] = 'Changed'
json_to_browser = json.dumps(data)
resp = self.myapp.post('/data/updateLinkRow.json',
data=json_to_browser)
self.assertEqual(resp.status_code, 200)
data = utils.json_decode(resp)
self.assertEqual(data["StatusCode"], 0)
result = db.select('LinkTab', TabName='Lease', AttrName='ID')
self.assertEqual(len(result), 1)
self.assertEqual(result[0].LinkName, 'Changed')
def test_get_link_row(self):
#setup stuff
db = config.get_database()
dbu = DatabaseUtilities()
db_create = DatabaseCreate()
utils = Utils()
dbu.delete_all_tables()
db_create.linktab()
linktab = db.get_data_structure('LinkTab')
linktab.TabName = 'Well'
linktab.AttrName = 'ID'
json_to_browser = json.dumps(utils.obj_to_dict(linktab))
print('json_to_browser', json_to_browser)
# Data should not be found but there should not be an error
resp = self.myapp.post('/data/getLinkRow.json', data=json_to_browser)
self.assertEqual(resp.status_code, 200)
data = utils.json_decode(resp)
self.assertEqual(data["LinkName"], '')
linktab.LinkName = 'Well'
linktab.BaseTab = 0
linktab.ShowAttrs = 'ID,UWI'
db.insert(linktab)
# Data should be found
resp = self.myapp.post('/data/getLinkRow.json', data=json_to_browser)
data = utils.json_decode(resp)
self.assertEqual(resp.status_code, 200)
self.assertEqual(data["LinkName"], 'Well')
self.assertEqual(data["ShowAttrs"], 'ID,UWI')
print('status:', resp.status)
print('status code:', resp.status_code)
print(resp)
print(resp.data)
def test_get_link_row(self):
#setup stuff
db = config.get_database()
dbu = DatabaseUtilities()
db_create = DatabaseCreate()
utils = Utils()
dbu.delete_all_tables()
db_create.linktab()
linktab = db.get_data_structure('LinkTab')
linktab.TabName = 'Well'
linktab.AttrName = 'ID'
json_to_browser = json.dumps(utils.obj_to_dict(linktab))
print('json_to_browser', json_to_browser)
# Data should not be found but there should not be an error
resp = self.myapp.post('/data/getLinkRow.json',data=json_to_browser)
self.assertEqual(resp.status_code, 200)
data = utils.json_decode(resp)
self.assertEqual(data["LinkName"],'')
linktab.LinkName = 'Well'
linktab.BaseTab = 0
linktab.ShowAttrs = 'ID,UWI'
db.insert(linktab)
# Data should be found
resp = self.myapp.post('/data/getLinkRow.json',data=json_to_browser)
data = utils.json_decode(resp)
self.assertEqual(resp.status_code, 200)
self.assertEqual(data["LinkName"],'Well')
self.assertEqual(data["ShowAttrs"],'ID,UWI')
print('status:',resp.status)
print('status code:',resp.status_code)
print(resp)
print(resp.data)
def test_link_update(self):
db = config.get_database()
dbu = DatabaseUtilities()
db_create = DatabaseCreate()
utils = Utils()
dbu.delete_all_tables()
db_create.linktab()
# Test Insert
data = dict()
data["ID"] = '0'
data["TabName"] = 'Lease'
data["AttrName"] = 'ID'
data["LinkName"] = 'Lease'
json_to_browser = json.dumps(data)
resp = self.myapp.post('/data/updateLinkRow.json',data=json_to_browser)
self.assertEqual(resp.status_code, 200)
data = utils.json_decode(resp)
self.assertEqual(data["StatusCode"],0)
result = db.select('LinkTab', TabName='Lease', AttrName='ID')
self.assertEqual(len(result), 1)
# Test Update
data = utils.obj_to_dict(result[0])
data['LinkName'] = 'Changed'
json_to_browser = json.dumps(data)
resp = self.myapp.post('/data/updateLinkRow.json',data=json_to_browser)
self.assertEqual(resp.status_code, 200)
data = utils.json_decode(resp)
self.assertEqual(data["StatusCode"],0)
result = db.select('LinkTab', TabName='Lease', AttrName='ID')
self.assertEqual(len(result), 1)
self.assertEqual(result[0].LinkName,'Changed')
