mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-11-03 12:18:00 +00:00
90
py-scripts/event_breaker.py
Executable file
90
py-scripts/event_breaker.py
Executable file
@@ -0,0 +1,90 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
"""
|
||||||
|
This file is intended to expose concurrency
|
||||||
|
problems in the /events/ URL handler
|
||||||
|
"""
|
||||||
|
import sys
|
||||||
|
|
||||||
|
if sys.version_info[0] != 3:
|
||||||
|
print("This script requires Python 3")
|
||||||
|
exit(1)
|
||||||
|
|
||||||
|
if 'py-json' not in sys.path:
|
||||||
|
sys.path.append('../py-json')
|
||||||
|
|
||||||
|
import argparse
|
||||||
|
from LANforge.lfcli_base import LFCliBase
|
||||||
|
from realm import Realm
|
||||||
|
import datetime
|
||||||
|
from datetime import datetime
|
||||||
|
import time
|
||||||
|
import pprint
|
||||||
|
|
||||||
|
class EventBreaker(Realm):
|
||||||
|
def __init__(self, host, port,
|
||||||
|
duration=None,
|
||||||
|
_debug_on=False,
|
||||||
|
_exit_on_error=False,
|
||||||
|
_exit_on_fail=False):
|
||||||
|
super().__init__(host, port)
|
||||||
|
self.counter = 0
|
||||||
|
self.test_duration=duration
|
||||||
|
if (self.test_duration is None):
|
||||||
|
raise ValueError("run wants numeric run_duration_sec")
|
||||||
|
|
||||||
|
def create(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
|
||||||
|
now = datetime.now()
|
||||||
|
end_time = self.parse_time(self.test_duration) + now
|
||||||
|
while datetime.now() < end_time:
|
||||||
|
print ('.', end='')
|
||||||
|
response = self.json_get("/events/all")
|
||||||
|
if "events" not in response:
|
||||||
|
# pprint.pprint(response)
|
||||||
|
raise AssertionError("no events in response")
|
||||||
|
events = response["events"]
|
||||||
|
self.counter += 1
|
||||||
|
for record in events:
|
||||||
|
# pprint.pprint(record)
|
||||||
|
|
||||||
|
for k in record.keys():
|
||||||
|
if record[k] is None:
|
||||||
|
print ('.', end='')
|
||||||
|
continue
|
||||||
|
# pprint.pprint( record[k])
|
||||||
|
if "NA" == record[k]["event"] \
|
||||||
|
or "NA" == record[k]["name"] \
|
||||||
|
or "NA" == record[k]["type"] \
|
||||||
|
or "NA" == record[k]["priority"]:
|
||||||
|
print( "id[%s]"%k, end='')
|
||||||
|
print("counter %s"%self.counter)
|
||||||
|
|
||||||
|
|
||||||
|
def cleanup(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
parser = LFCliBase.create_bare_argparse(
|
||||||
|
prog='event_breaker.py',
|
||||||
|
formatter_class=argparse.RawTextHelpFormatter)
|
||||||
|
|
||||||
|
parser.add_argument("--test_duration", help='test duration', default="30s" )
|
||||||
|
# if optional_args is not None:
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
event_breaker = EventBreaker(host=args.mgr,
|
||||||
|
port=args.mgr_port,
|
||||||
|
duration=args.test_duration,
|
||||||
|
_debug_on=True,
|
||||||
|
_exit_on_error=True,
|
||||||
|
_exit_on_fail=True)
|
||||||
|
event_breaker.create()
|
||||||
|
event_breaker.run()
|
||||||
|
event_breaker.cleanup()
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
Reference in New Issue
Block a user