blob: 3e6e419545ae9ee166350287408338b576350c3e (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
from contextlib import contextmanager
from .ftp_server import SimpleFtpServer
from .http_server import SimpleHttpServer
@contextmanager
def create_file_server(file_server_type):
if file_server_type == "FTP":
server = SimpleFtpServer()
elif file_server_type == "HTTP":
server = SimpleHttpServer()
else:
assert False
try:
yield server
finally:
server.stop()
|