def test_file_delete(self):
print("case: test_file_delete")
file_object = open(self.crashCtrlHandler.CORE_FILE_PATH + "test1.core.gz", 'w')
file_object.write(self.text)
file_object.close()
self.crashCtrlHandler.update_crash_file_table(None)
for key in CrashFileStatusInfo.get_keys():
crashFileStatusInfo = CrashFileStatusInfo(key)
crashFileStatusInfo.read()
self.crashCtrlHandler.update_pending_file_idx_list(key, self.crashCtrlHandler.CONTROL_DELETEFILE)
crashFileStatusInfo = CrashFileStatusInfo(key)
crashFileStatusInfo.read()
self.assertNotEquals(crashFileStatusInfo, None)
def test_add_crash_file(self):
print("case: test_add_crash_file")
self.crashCtrlHandler.update_crash_file_table(None)
if self.crashCtrlHandler.is_upload_process_alive():
self.crashCtrlHandler.stop_upload_process()
for fileName in self.nameList:
file_object = open(self.crashCtrlHandler.CORE_FILE_PATH + fileName, 'w')
file_object.write(self.text)
file_object.close()
time.sleep(self.ONE_SECOND)
self.crashCtrlHandler.update_crash_file_table(None)
for key in CrashFileStatusInfo.get_keys():
crashFileStatsInfo = CrashFileStatusInfo(key)
crashFileStatsInfo.read()
self.assertNotEquals(len(os.listdir(CrashFileCtrlHandler.CORE_FILE_PATH)), 20)
def test_tftp_fail_status(self):
print("case:test_tftp_fail_status")
cfg_msg = config()
rpdCtrl = cfg_msg.RpdCtrl
rpdServerCtrlCfg = rpdCtrl.CrashDataServerCtrl
rpdServerCtrlCfg.DestPath = "/./"
rpdServerCtrlCfg.Protocol = CrashFileCtrlHandler.PROTOCOL_TFTP
self.crashCtrlHandler.save_crash_data_server(rpdServerCtrlCfg)
if self.crashCtrlHandler.is_upload_process_alive():
self.crashCtrlHandler.stop_upload_process()
time.sleep(self.ONE_SECOND)
self.assertEquals(self.crashCtrlHandler.is_upload_process_alive(), False)
os.system("rm -rf /tmp/crash/*")
for fileName in self.nameList1:
file_object = open(self.crashCtrlHandler.CORE_FILE_PATH + fileName, 'w')
file_object.write(self.text)
file_object.close()
self.crashCtrlHandler.update_crash_file_table(None)
cfg_msg = config()
rpdCtrl = cfg_msg.RpdCtrl
rpdServerCtrlCfg = rpdCtrl.CrashDataServerCtrl
rpdServerCtrlCfg.DestIpAddress = ""
rpdServerCtrlCfg.DestPath = "/./"
rpdServerCtrlCfg.Protocol = CrashFileCtrlHandler.PROTOCOL_TFTP
self.crashCtrlHandler.save_crash_data_server(rpdServerCtrlCfg)
print("step5: test upload=========\n")
cfg_msg = config()
rpdCtrl = cfg_msg.RpdCtrl
rpdServerCtrlCfg = rpdCtrl.CrashDataServerCtrl
rpdServerCtrlCfg.DestIpAddress = "128.0.0.1"
rpdServerCtrlCfg.DestPath = "/./"
rpdServerCtrlCfg.Protocol = CrashFileCtrlHandler.PROTOCOL_TFTP
self.crashCtrlHandler.save_crash_data_server(rpdServerCtrlCfg)
self.crashCtrlHandler.get_crash_data_server(rpdServerCtrlCfg)
for key in CrashFileStatusInfo.get_keys():
crashFileStatusInfo = CrashFileStatusInfo(key)
crashFileStatusInfo.read()
self.crashCtrlHandler.update_pending_file_idx_list(key, self.crashCtrlHandler.CONTROL_UPLOAD)
time.sleep(self.FIVE_SECOND)
for key in CrashFileStatusInfo.get_keys():
crashFileStatusInfo = CrashFileStatusInfo(key)
crashFileStatusInfo.read()
self.assertNotEquals(crashFileStatusInfo.fileStatus, self.crashCtrlHandler.STATUS_AVAILFORUPLOAD)
def test_file_ctrl_set(self):
print("case: test_file_ctrl_set")
CrashFileCtrlHandler.delete_core_file("")
os.system("rm -rf /tmp/crash/*")
if self.crashCtrlHandler.is_upload_process_alive():
self.crashCtrlHandler.stop_upload_process()
time.sleep(self.ONE_SECOND)
file_object = open(self.crashCtrlHandler.CORE_FILE_PATH + "test1.core.gz", 'w')
file_object.write(self.text)
file_object.close()
os.mkdir("/tmp/crash/test")
CrashFileCtrlHandler.delete_core_file("test")
self.crashCtrlHandler.update_crash_file_table(None)
for key in CrashFileStatusInfo.get_keys():
crashFileStatusInfo = CrashFileStatusInfo(key)
crashFileStatusInfo.read()
self.crashCtrlHandler.update_pending_file_idx_list(key, self.crashCtrlHandler.CONTROL_OTHER)
crashFileStatusInfo.read()
self.assertEquals(crashFileStatusInfo.fileControl, self.crashCtrlHandler.CONTROL_OTHER)
self.assertEquals(crashFileStatusInfo.fileStatus, self.crashCtrlHandler.STATUS_AVAILFORUPLOAD)
def test_del_crash_file(self):
print("case: test_del_crash_file")
self.crashCtrlHandler.check_file_valid("111.log")
self.crashCtrlHandler.check_file_valid("111.core.gz")
self.crashCtrlHandler.update_waiting_upload_list(True, None)
self.crashCtrlHandler.update_crash_file_table(None)
delFileName = ""
for fileName in self.nameList:
delFileName = fileName
file_object = open(self.crashCtrlHandler.CORE_FILE_PATH + fileName, 'w')
file_object.write(self.text)
file_object.close()
self.crashCtrlHandler.update_crash_file_table(None)
os.system("rm -rf /tmp/crash/%s" % delFileName)
self.crashCtrlHandler.update_crash_file_table(None)
for key in CrashFileStatusInfo.get_keys():
crashFileStatsInfo = CrashFileStatusInfo(key)
crashFileStatsInfo.read()
self.assertNotEquals(len(os.listdir(CrashFileCtrlHandler.CORE_FILE_PATH)), 20)
def test_http_ctrl_hal_msg(self):
print("case: test_http_ctrl_hal_msg")
if not self.tftp_pid:
self.tftp_pid = subprocess.Popen(self.start_tftp_cmd, executable='bash', shell=True)
if self.crashCtrlHandler.is_upload_process_alive():
self.crashCtrlHandler.stop_upload_process()
time.sleep(self.ONE_SECOND)
self.assertEquals(self.crashCtrlHandler.is_upload_process_alive(), False)
os.system("rm -rf /tmp/crash/*")
for fileName in self.nameList:
file_object = open(self.crashCtrlHandler.CORE_FILE_PATH + fileName, 'w')
file_object.write(self.text)
file_object.close()
self.crashCtrlHandler.update_crash_file_table(None)
for key in CrashFileStatusInfo.get_keys():
crashFileStatusInfo = CrashFileStatusInfo(key)
crashFileStatusInfo.read()
rcp_msg = t_RcpMessage()
rcp_msg.RcpMessageType = t_RcpMessage.RPD_CONFIGURATION
rcp_msg.RpdDataMessage.RpdDataOperation = \
t_RpdDataMessage.RPD_CFG_WRITE
cfg_msg = config()
rpdCtrl = cfg_msg.RpdCtrl
dataFileCtrl = rpdCtrl.CrashDataFileCtrl.add()
dataFileCtrl.Index = crashFileStatusInfo.index
dataFileCtrl.FileControl = CrashFileCtrlHandler.CONTROL_UPLOAD
rpdCtrl.CrashDataServerCtrl.DestIpAddress = ""
rpdCtrl.CrashDataServerCtrl.DestPath = "/tmp/crash/"
rpdCtrl.CrashDataServerCtrl.Protocol = CrashFileCtrlHandler.PROTOCOL_HTTP
rcp_msg.RpdDataMessage.RpdData.CopyFrom(cfg_msg)
cfg_payload = rcp_msg.SerializeToString()
rdpCtrlMsg = HalMessage("HalConfig", SrcClientID="testRpdCtrlMsg",
SeqNum=325,
CfgMsgType=HalConfigMsg.MsgTypeRpdCtrl,
CfgMsgPayload=cfg_payload)
self.halClient.recvCfgMsgCb(rdpCtrlMsg)
def test_file_upload(self):
print("case: test_file_upload")
self.crashCtrlHandler = self.halClient.crashFileCtrlHandler
self.crashCtrlHandler.TFTP_PORT = self.TFTP_PORT
time.sleep(1)
self.assertEquals(self.crashCtrlHandler.is_upload_process_alive(), False)
cfg_msg = config()
rpdCtrl = cfg_msg.RpdCtrl
rpdServerCtrlCfg = rpdCtrl.CrashDataServerCtrl
rpdServerCtrlCfg.DestIpAddress = "127.0.0.1"
rpdServerCtrlCfg.DestPath = "/./"
rpdServerCtrlCfg.Protocol = CrashFileCtrlHandler.PROTOCOL_TFTP
self.crashCtrlHandler.save_crash_data_server(rpdServerCtrlCfg)
os.system("rm -rf /tmp/crash/*")
for fileName in self.nameList:
file_object = open(self.crashCtrlHandler.CORE_FILE_PATH + fileName, 'w')
file_object.write(self.text)
file_object.close()
break
self.crashCtrlHandler.update_crash_file_table(None)
print("step1: test upload=========\n")
for key in CrashFileStatusInfo.get_keys():
crashFileStatusInfo = CrashFileStatusInfo(key)
crashFileStatusInfo.read()
self.crashCtrlHandler.update_pending_file_idx_list(key, self.crashCtrlHandler.CONTROL_UPLOAD)
self.crashCtrlHandler.update_pending_file_idx_list(key, self.crashCtrlHandler.CONTROL_UPLOAD)
break
time.sleep(self.ONE_SECOND)
self.crashCtrlHandler.update_waiting_upload_list(True, 1)
self.crashCtrlHandler.waitingUploadList.append(1)
self.crashCtrlHandler.update_waiting_upload_list(False, 1)
for key in CrashFileStatusInfo.get_keys():
crashFileStatusInfo = CrashFileStatusInfo(key)
crashFileStatusInfo.read()
time.sleep(self.ONE_SECOND)
time.sleep(self.ONE_SECOND)
print("step2: test cancel upload")
for key in CrashFileStatusInfo.get_keys():
crashFileStatusInfo = CrashFileStatusInfo(key)
crashFileStatusInfo.read()
self.crashCtrlHandler.update_pending_file_idx_list(key, self.crashCtrlHandler.CONTROL_CANCELUPLOAD)
crashFileStatusInfo = CrashFileStatusInfo(key)
crashFileStatusInfo.read()
self.assertEquals(crashFileStatusInfo.fileStatus, self.crashCtrlHandler.STATUS_UPLOADCANCELLED)
self.crashCtrlHandler.waitingUploadList.append(crashFileStatusInfo.index)
crashFileStatusInfo.read()
self.crashCtrlHandler.update_waiting_upload_list(crashFileStatusInfo.index, self.crashCtrlHandler.CONTROL_CANCELUPLOAD)
self.crashCtrlHandler.update_waiting_upload_list(crashFileStatusInfo.index, self.crashCtrlHandler.CONTROL_DELETEFILE)
time.sleep(self.ONE_SECOND)
print("step3: test upload and delete")
for key in CrashFileStatusInfo.get_keys():
crashFileStatusInfo = CrashFileStatusInfo(key)
crashFileStatusInfo.read()
self.crashCtrlHandler.update_pending_file_idx_list(key, self.crashCtrlHandler.CONTROL_UPLOADANDDELETE)
crashFileStatusInfo = CrashFileStatusInfo(key)
crashFileStatusInfo.read()
time.sleep(self.ONE_SECOND)