aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--tests/dcaegen2/prh-testcases/resources/simulator/AAI.py38
1 files changed, 29 insertions, 9 deletions
diff --git a/tests/dcaegen2/prh-testcases/resources/simulator/AAI.py b/tests/dcaegen2/prh-testcases/resources/simulator/AAI.py
index 91a93de5..bd76c616 100644
--- a/tests/dcaegen2/prh-testcases/resources/simulator/AAI.py
+++ b/tests/dcaegen2/prh-testcases/resources/simulator/AAI.py
@@ -1,4 +1,5 @@
import logging
+import json
import sys
import re
import time
@@ -16,22 +17,29 @@ logging.basicConfig(
logger = logging.getLogger('AAI-simulator-logger')
pnfs = 'Empty'
-pnf_entry = 'Empty'
+pnf_entry = {}
+
+
+def _mark_response_as_http_ok(http_endpoint):
+ logger.info('Execution status 200')
+ httpServerLib.header_200_and_json(http_endpoint)
+
class AAISetup(BaseHTTPRequestHandler):
def do_PUT(self):
logger.info('AAI SIM Setup Put execution')
- if re.search('/set_pnf', self.path):
+ if re.search('/set_pnf$', self.path):
global pnfs
- content_length = int(self.headers['Content-Length'])
+ content_length = self._get_content_length()
pnfs = self.rfile.read(content_length)
- logger.info('Execution status 200')
- httpServerLib.header_200_and_json(self)
+ _mark_response_as_http_ok(self)
if re.search('/set_pnf_entry',self.path):
- logger.info('Execution status 200')
- httpServerLib.header_200_and_json(self)
+ global pnf_entry
+ content_length = self._get_content_length()
+ pnf_entry = json.loads(self.rfile.read(content_length))
+ _mark_response_as_http_ok(self)
return
@@ -40,14 +48,26 @@ class AAISetup(BaseHTTPRequestHandler):
if re.search('/reset', self.path):
global pnfs
pnfs = 'Empty'
- logger.info('Execution status 200')
- httpServerLib.header_200_and_json(self)
+ _mark_response_as_http_ok(self)
return
+ def _get_content_length(self):
+ return int(self.headers['Content-Length'])
+
class AAIHandler(BaseHTTPRequestHandler):
+ def do_GET(self):
+ logger.info('AAI SIM Get execution')
+ if re.search('/get_pnf_entry', self.path):
+ _mark_response_as_http_ok(self)
+ body = json.dumps(pnf_entry)
+ self.wfile.write(body.encode())
+
+ return
+
+
def do_PATCH(self):
logger.info('AAI SIM Patch execution')
pnfs_name = '/aai/v12/network/pnfs/pnf/' + pnfs.decode()