mirror of https://github.com/dnomd343/ProxyC
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
3.2 KiB
93 lines
3.2 KiB
#!/usr/bin/python
|
|
# -*- coding:utf-8 -*-
|
|
|
|
import os
|
|
import sys
|
|
import time
|
|
import signal
|
|
import subprocess
|
|
|
|
import Check as Checker
|
|
import ProxyTester as Tester
|
|
|
|
testConfig = {
|
|
'bind': '::1',
|
|
'addr': '::1',
|
|
'port': 12345,
|
|
'passwd': 'dnomd343',
|
|
'host': 'local.343.re',
|
|
'path': '/test',
|
|
'service': 'dnomd343',
|
|
'file': '/tmp/proxycTest.json',
|
|
'id': '1f7aa040-94d8-4b53-ae85-af6946d550bb',
|
|
'cert': '/etc/ssl/certs/343.re/fullchain.pem',
|
|
'key': '/etc/ssl/certs/343.re/privkey.pem',
|
|
}
|
|
|
|
def testBuild(config: dict): # load file and start process
|
|
if config['filePath'] is not None:
|
|
with open(config['filePath'], 'w') as fileObject: # save file
|
|
fileObject.write(config['fileContent'])
|
|
if config['startCommand'] is None:
|
|
return None
|
|
return subprocess.Popen( # start process
|
|
config['startCommand'],
|
|
env = config['envVar'],
|
|
stdout = subprocess.DEVNULL,
|
|
stderr = subprocess.DEVNULL,
|
|
preexec_fn = os.setpgrp # new process group
|
|
)
|
|
|
|
def testDestroy(config: dict, process): # remove file and kill process
|
|
if process is not None and process.poll() is None: # still alive
|
|
os.killpg(os.getpgid(process.pid), signal.SIGTERM) # kill process group
|
|
while process.poll() is None: # wait for exit
|
|
process.terminate() # SIGTERM
|
|
time.sleep(0.2)
|
|
if config['filePath'] is not None:
|
|
os.remove(config['filePath']) # remove file
|
|
|
|
def testObject(option: dict) -> None: # test target object
|
|
aiderProcess = None
|
|
serverProcess = testBuild(option['server']) # start server process
|
|
if option['aider'] is not None:
|
|
aiderProcess = testBuild(option['aider']) # start aider process
|
|
|
|
checkResult = Checker.proxyTest({ # http check
|
|
'check': ['http'],
|
|
'info': option['proxy']
|
|
}, startDelay = 0.5)
|
|
print(option['caption'], end=' -> ')
|
|
if not checkResult['success']: # client build error
|
|
print('\n--------------------------------------------------------------------------------------------------------------------------------')
|
|
print(option)
|
|
print('--------------------------------------------------------------------------------------------------------------------------------\n')
|
|
testDestroy(option['server'], serverProcess) # destroy and exit
|
|
if option['aider'] is not None:
|
|
testDestroy(option['aider'], aiderProcess)
|
|
raise Exception('check error')
|
|
delay = checkResult['check']['http']['delay'] # get http delay
|
|
if delay > 0:
|
|
print(format(delay, '.2f') + 'ms')
|
|
else:
|
|
print('ERROR')
|
|
|
|
testDestroy(option['server'], serverProcess) # destroy server process
|
|
if option['aider'] is not None:
|
|
testDestroy(option['aider'], aiderProcess) # destroy aider process
|
|
|
|
if len(sys.argv) == 1: # no param
|
|
print('Unknown test type')
|
|
sys.exit(1)
|
|
testList = Tester.test(sys.argv[1].lower().strip(), testConfig) # get test list
|
|
|
|
if len(sys.argv) == 2: # test all
|
|
for i in range(0, len(testList)):
|
|
print(str(i), end = ': ')
|
|
testObject(testList[i])
|
|
sys.exit(0)
|
|
|
|
if len(sys.argv) == 3: # test target index
|
|
testObject(testList[int(sys.argv[2])])
|
|
else:
|
|
print('Too many options')
|
|
|