def test_setAttrAndReset(self):
#check that the value is correct to start
err, unknownErr = pref.get(pref.ERROR_UNKNOWN)
self.assertEqual(
pref.getError(pref.ERROR_UNKNOWN).code, unknownErr.code)
self.assertEqual(pref.Success.code, err.code)
#set new value
err = pref.setAttr(pref.ERROR_UNKNOWN, "error")
self.assertEqual(pref.Success.code, err.code)
#check new value
err, unknownErr = pref.get(pref.ERROR_UNKNOWN)
self.assertEqual("error", unknownErr)
self.assertEqual(pref.Success.code, err.code)
#reset config
pref.resetConfig()
#check that the value was reset
err, unknownErr = pref.get(pref.ERROR_UNKNOWN)
self.assertEqual(
pref.getError(pref.ERROR_UNKNOWN).code, unknownErr.code)
self.assertEqual(pref.Success.code, err.code)
def test_setAttrInDict(self):
err = pref.setAttr(pref.ERROR_UNKNOWN, "error")
self.assertEqual(pref.Success.code, err.code)
err, unknownErr = pref.get(pref.ERROR_UNKNOWN)
self.assertEqual("error", unknownErr)
self.assertEqual(pref.Success.code, err.code)
def test_setAttr(self):
err = pref.setAttr(pref.CONFIG_LOGIN_ENDPOINT, "/login2")
self.assertEqual(pref.Success.code, err.code)
err, login = pref.get(pref.CONFIG_LOGIN_ENDPOINT)
self.assertEqual("/login2", login)
self.assertEqual(pref.Success.code, err.code)
def apiRequest():
'''
API Request /api
used for all operations between frontend and backend
ie
RUN_SCRIPT
MANAGE_SCRIPT
MANAGE_COMPUTERS
MANAGE_SCRIPT_LOGS
SCHEDULE_SCRIPT
GET_FILE
request:
{
op: <operation:string>
data: <opData:dict>
}
return:
{
Error:
}
'''
logger.info("api request: {}".format(request.json))
# if not("userID" in session):
# session.pop("userID", None)
#Makes sure user is logged in if not redirect to login
userID = None
try:
userID = session[pref.getNoCheck(pref.REQ_VAR_USER_ID)]
except:
return redirect(pref.getNoCheck("/#/"))
err = pref.Success
#names of request vars
bodyName = pref.getNoCheck(pref.REQ_VAR_BODY)
dataName = pref.getNoCheck(pref.REQ_VAR_DATA)
jsonOpName = pref.getNoCheck(pref.REQ_VAR_OP)
try:
opName = pref.CONFIG_OPERATIONS + ":" + request.json[bodyName][
jsonOpName]
data = request.json[bodyName][dataName]
except:
err = pref.getError(pref.ERROR_INVALID_REQUEST, args=(request.json))
logger.error(err)
returnValue = None
if (err == pref.Success):
err, op = pref.get(opName)
if (err == pref.Success):
apiFtn = OPS[op]
returnValue = apiFtn(data)
else:
logger.error(err)
returnValue = jsonify(Error=err.toJson(), data={})
return (returnValue, 200)
def loginRequest():
logger.info("login request")
#names of request vars
bodyName = pref.getNoCheck(pref.REQ_VAR_BODY)
dataName = pref.getNoCheck(pref.REQ_VAR_DATA)
jsonOpName = pref.getNoCheck(pref.REQ_VAR_OP)
opName = pref.CONFIG_LOGIN_OPERATION + ":" + request.json[bodyName][
jsonOpName]
data = request.json[bodyName][dataName]
returnValue = None
err, op = pref.get(opName)
if (err == pref.Success):
loginFtn = LOGIN_OPS[op]
returnValue = loginFtn(data)
else:
logger.error(err)
returnValue = jsonify(Error=err.toJson(), data={})
return (returnValue, 200)
def test_getNoCheckInvalidKey(self):
err, dne = pref.get("DNE")
self.assertEqual(None, dne)
def test_getInvalidKey(self):
err, dne = pref.get("DNE")
self.assertEqual(None, dne)
self.assertEqual(
pref.getError(pref.ERROR_ATTRIBUTE_NOT_FOUND).code, err.code)
def test_get(self):
err, login = pref.get(pref.CONFIG_LOGIN_ENDPOINT)
self.assertEqual("/login", login)
self.assertEqual(pref.Success.code, err.code)
def loginRequest():
logger.info("login request")
#names of request vars
bodyName = pref.getNoCheck(pref.REQ_VAR_BODY)
dataName = pref.getNoCheck(pref.REQ_VAR_DATA)
jsonOpName = pref.getNoCheck(pref.REQ_VAR_OP)
opName = pref.CONFIG_LOGIN_OPERATION + ":" + request.json[bodyName][
jsonOpName]
data = request.json[bodyName][dataName]
returnValue = None
err, op = pref.get(opName)
if (err == pref.Success):
loginFtn = LOGIN_OPS[op]
returnValue = loginFtn(data)
else:
logger.error(err)
returnValue = jsonify(Error=err.toJson(), data={})
return (returnValue, 200)
if __name__ == '__main__':
err, port = pref.get(pref.CONFIG_PORT)
if (err.code == 0):
app.run(debug=True, port=port)
else:
logger.critical(err)
