mirror of https://github.com/progval/irctest.git
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
import os
|
|
import shutil
|
|
import socket
|
|
import tempfile
|
|
import subprocess
|
|
|
|
from .optional_extensions import NotImplementedByController
|
|
|
|
class _BaseController:
|
|
pass
|
|
|
|
class DirectoryBasedController(_BaseController):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.directory = None
|
|
self.proc = None
|
|
|
|
def kill_proc(self):
|
|
self.proc.terminate()
|
|
try:
|
|
self.proc.wait(5)
|
|
except subprocess.TimeoutExpired:
|
|
self.proc.kill()
|
|
self.proc = None
|
|
def kill(self):
|
|
if self.proc:
|
|
self.kill_proc()
|
|
if self.directory:
|
|
shutil.rmtree(self.directory)
|
|
def open_file(self, name, mode='a'):
|
|
assert self.directory
|
|
if os.sep in name:
|
|
dir_ = os.path.join(self.directory, os.path.dirname(name))
|
|
if not os.path.isdir(dir_):
|
|
os.makedirs(dir_)
|
|
assert os.path.isdir(dir_)
|
|
return open(os.path.join(self.directory, name), mode)
|
|
def create_config(self):
|
|
self.directory = tempfile.mkdtemp()
|
|
|
|
class BaseClientController(_BaseController):
|
|
def run(self, hostname, port, auth):
|
|
raise NotImplementedError()
|
|
|
|
class BaseServerController(_BaseController):
|
|
def run(self, hostname, port, start_wait):
|
|
raise NotImplementedError()
|
|
def registerUser(self, case, username):
|
|
raise NotImplementedByController('registration')
|
|
|