diff options
Diffstat (limited to 'test/mocks/datafilecollector-testharness/mr-sim/mr-sim.py')
| -rw-r--r-- | test/mocks/datafilecollector-testharness/mr-sim/mr-sim.py | 289 |
1 files changed, 289 insertions, 0 deletions
diff --git a/test/mocks/datafilecollector-testharness/mr-sim/mr-sim.py b/test/mocks/datafilecollector-testharness/mr-sim/mr-sim.py new file mode 100644 index 000000000..c37ae698a --- /dev/null +++ b/test/mocks/datafilecollector-testharness/mr-sim/mr-sim.py @@ -0,0 +1,289 @@ +import argparse +import os +from werkzeug import secure_filename +from flask import Flask, render_template, request +from time import sleep +import sys +import json +from flask import Flask +app = Flask(__name__) + +DEFAULT_IP = "localhost" + + +@app.route( + "/events/unauthenticated.VES_NOTIFICATION_OUTPUT/OpenDcae-c12/C12", + methods=['GET']) +def MR_reply(): + global mr_counter + global mr_replies + + mr_counter = mr_counter + 1 + print("MR receiver counter: " + str(mr_counter)) + + if mr_replies[mr_counter].sleepMs != 0: + sleep(mr_replies[mr_counter].sleepMs / 1000.0) + print("Sleeping: " + str(mr_replies[mr_counter].sleepMs) + " ms") + + if mr_replies[mr_counter].replytype == 0: + #print (str(mr_replies[mr_counter].jsonreply)) + print("Regular reply") + response = app.response_class( + response=mr_replies[mr_counter].jsonreply, + status=200, + mimetype='application/json') + + return response + + if mr_replies[mr_counter].replytype == 2: + + print("error: 404") + response = app.response_class( + response="", + status=404, + mimetype='application/json') + + return response + + if mr_replies[mr_counter].replytype == 1: + print("do nothing, sink request") + return + + +class Reply: + """An instance of the reply event, which can be configured to behave in a certain way + (delay, error code, reply body""" + + def to_json(self): + return self.jsonreply + + def __init__( + self, + ip=DEFAULT_IP, + file="1MB.tar.gz", + sleepMs=0, + replyType=0, + port=1022, + type="ftps"): + self.sleepMs = sleepMs + self.ip = ip + self.file = file + self.port = port + self.replytype = replyType # 0 for reply, 1 timeout, 2 deny + self.user = "onap" + self.passwd = "pano" + self.type = type + self.jsonreply = str.encode(""" + [{ + "event": { + "commonEventHeader": { + "startEpochMicrosec": 8745745764578, + "eventId": "FileReady_1797490e-10ae-4d48-9ea7-3d7d790b25e1", + "timeZoneOffset": "UTC+05.30", + "internalHeaderFields": { + "collectorTimeStamp": "Tue, 09 18 2018 10:56:52 UTC" + }, + "priority": "Normal", + "version": "4.0.1", + "reportingEntityName": "otenb5309", + "sequence": 0, + "domain": "notification", + "lastEpochMicrosec": 8745745764578, + "eventName": "Noti_RnNode-Ericsson_FileReady", + "vesEventListenerVersion": "7.0.1", + "sourceName": "oteNB5309" + }, + "notificationFields": { + "notificationFieldsVersion": "2.0", + "changeType": "FileReady", + "changeIdentifier": "PM_MEAS_FILES", + "arrayOfNamedHashMap": [ + { + "name": \"""" + + self.file + + """", + "hashMap": { + "fileFormatType": "org.3GPP.32.435#measCollec", + "location": \"""" + + self.type + + """://""" + + self.user + + """:""" + + self.passwd + + """@""" + + self.ip + + """:""" + + str(self.port) + + """/""" + + self.file + + """", + "fileFormatVersion": "V10", + "compression": "gzip" + } + } + ] + } + } + }] + """) + + +def replyFactory( + ip=DEFAULT_IP, + file="1MB.tar.gz", + factoryport=1022, + count=1, + factorytype="ftps"): + aggregatedReply = "" + # first item does not require . + aggregatedReply = Reply(ip, file, port=factoryport).to_json() + for i in range(count - 1): + aggregatedReply = aggregatedReply + b", " + \ + Reply(ip, file, port=factoryport, type=factorytype).to_json() + #print(b"aggregated reply: " + aggregatedReply) + return b"[" + aggregatedReply + b"]" + + +def prepareMrRespArrSftp(): + global mr_replies + + for i in range(400): # prepare 400 regular replies + mr_replies.append( + Reply( + port=1022, + ip="localhost", + type="sftp", + file="1MB.tar.gz")) + #mr_replies[0] is not used + + +def prepareMrRespArrFtps(): + global mr_replies + + for i in range(400): + mr_replies.append( + Reply( + port=21, + ip="localhost", + type="ftps", + file="1MB.tar.gz")) + + +def tc1(): + prepareMrRespArrSftp() + # no mutation needed in this TC + + +def tc2(): + global mr_replies + + for i in range(7): + mr_replies.append( + Reply( + port=1022, + ip="localhost", + type="sftp", + file="1MB.tar.gz")) + + # inserting and empty reply message + mr_replies[1].jsonreply = b"" + mr_replies[2].jsonreply = b"" + + # inserting a 404 error and delay + mr_replies[3].replytype = 2 + mr_replies[3].sleepMs = 2000 + + # inserting and empty reply message + mr_replies[4].jsonreply = b"" + + # sink the message + mr_replies[5].replytype = 1 + + # reply with one proper file finally + mr_replies[6] = Reply( + port=1022, + ip="localhost", + type="sftp", + file="1MB.tar.gz") + + +def tc3(): + prepareMrRespArrFtps() + + +def tc4(): + global mr_replies + + for i in range(7): + mr_replies.append( + Reply( + port=21, + ip="localhost", + type="ftps", + file="1MB.tar.gz")) + + # inserting and empty reply message + mr_replies[1].jsonreply = b"" + mr_replies[2].jsonreply = b"" + + # inserting a 404 error and delay + mr_replies[3].replytype = 2 + mr_replies[3].sleepMs = 2000 + + # inserting and empty reply message + mr_replies[4].jsonreply = b"" + + # sink the message + mr_replies[5].replytype = 1 + + # reply with one proper file finally + mr_replies[6] = Reply( + port=21, + ip="localhost", + type="fftp", + file="1MB.tar.gz") + + +if __name__ == "__main__": + mr_replies = [] + mr_counter = 0 # counting hits reaching MR instance + DR_block_single_req = 0 + + parser = argparse.ArgumentParser() + parser.add_argument( + '--tc1', + action='store_true', + help='TC1: reply all queries with 1-1 files using SFTP') + parser.add_argument( + '--tc2', + action='store_true', + help='TC2: Reply according to error scenarios, then return 1 file finally for SFTP ---NOTE: updated keys required') + parser.add_argument( + '--tc3', + action='store_true', + help='TC3: reply all queries with 1-1 files using FTPS') + parser.add_argument( + '--tc4', + action='store_true', + help='TC4: Reply according to error scenarios, then return 1 file finally for FTPS ---NOTE: updated keys required') + + args = parser.parse_args() + + if args.tc1: + print("TC: #1") + tc1() + elif args.tc2: + print("TC: #2") + tc2() + elif args.tc3: + print("TC: #3") + tc3() + elif args.tc4: + print("TC: #4") + tc4() + + else: + print("No TC was defined") + print("use --help for usage info") + sys.exit() + app.run(port=2222) |