-
Notifications
You must be signed in to change notification settings - Fork 0
/
loadtest.py
executable file
·67 lines (47 loc) · 1.47 KB
/
loadtest.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
#! /usr/bin/env python
import sys
import logging
import logging.config
import time
import asyncore
from loadtest.Configuration import configuration
from loadtest.EventHandler import EventHandler
def initialize_logger():
global logger
print "Reading logging config from logging.ini..."
logging.config.fileConfig("logging.ini")
logger = logging.getLogger("chattest.loadtest")
logger.info("Load tester logger initialized!")
def initialize_configuration():
global config
global configfilename
configfilename = "loadtest.conf"
config = configuration()
logger.info("Reading configuration...")
ret = config.read(configfilename)
config.dump()
return ret
def save_configuration():
logger.info("Saving configuration to " + configfilename + "...")
configfile = open(configfilename, "w")
config.write(configfile)
configfile.close()
def main():
global clients
initialize_logger()
if(not initialize_configuration()):
logger.warn("Reading configuration failed")
save_configuration()
clients = []
for i in range(config.get_users_amount()):
clients.append(EventHandler(i+1))
try:
logger.info("Entering event loop...")
asyncore.loop(1)
except KeyboardInterrupt, err:
for i in range(config.get_users_amount()):
clients[i].disconnect()
save_configuration()
logger.info("Load tester terminated successfully")
if __name__ == "__main__":
main()