diff options
author | Richard Purdie <richard.purdie@linuxfoundation.org> | 2019-12-20 17:53:00 +0000 |
---|---|---|
committer | Richard Purdie <richard.purdie@linuxfoundation.org> | 2020-01-22 15:56:39 +0000 |
commit | ca3993cc4b13d4e661228cee6fb9448adfd0a4ba (patch) | |
tree | 7bfdd2411a144e7acad6ee111880d7a9c0d391cc /bitbake/lib/bb/tests/support/httpserver.py | |
parent | 244141ef8cb01728cecf50b2d2702dbe0573ab0d (diff) | |
download | poky-3.1_M2.tar.gz |
bitbake: tests/fetch: Allow wget upgrade tests to run against a local server3.1_M2
Currently these tests rely upon multiple uptream webservers which may change
or be unavailable. Add local copies of the test data, copy the httpserver
from OE-Core (used for testing there) and run these tests against a local
server instead.
(Bitbake rev: d5a4a352723258b4d499d3a51f340109c4f36f60)
Signed-off-by: Richard Purdie <richard.purdie@linuxfoundation.org>
Diffstat (limited to 'bitbake/lib/bb/tests/support/httpserver.py')
-rw-r--r-- | bitbake/lib/bb/tests/support/httpserver.py | 65 |
1 files changed, 65 insertions, 0 deletions
diff --git a/bitbake/lib/bb/tests/support/httpserver.py b/bitbake/lib/bb/tests/support/httpserver.py new file mode 100644 index 0000000000..78f7660053 --- /dev/null +++ b/bitbake/lib/bb/tests/support/httpserver.py | |||
@@ -0,0 +1,65 @@ | |||
1 | # | ||
2 | # SPDX-License-Identifier: MIT | ||
3 | # | ||
4 | |||
5 | import http.server | ||
6 | import multiprocessing | ||
7 | import os | ||
8 | import traceback | ||
9 | import signal | ||
10 | import logging | ||
11 | from socketserver import ThreadingMixIn | ||
12 | |||
13 | class HTTPServer(ThreadingMixIn, http.server.HTTPServer): | ||
14 | |||
15 | def server_start(self, root_dir, logger): | ||
16 | os.chdir(root_dir) | ||
17 | self.serve_forever() | ||
18 | |||
19 | class HTTPRequestHandler(http.server.SimpleHTTPRequestHandler): | ||
20 | |||
21 | def log_message(self, format_str, *args): | ||
22 | pass | ||
23 | |||
24 | class HTTPService(object): | ||
25 | |||
26 | def __init__(self, root_dir, host='', port=0, logger=None): | ||
27 | self.root_dir = root_dir | ||
28 | self.host = host | ||
29 | self.port = port | ||
30 | if not logger: | ||
31 | logger = logging.getLogger() | ||
32 | self.logger = logger | ||
33 | |||
34 | def start(self): | ||
35 | print(self.root_dir) | ||
36 | if not os.path.exists(self.root_dir): | ||
37 | self.logger.info("Not starting HTTPService for directory %s which doesn't exist" % (self.root_dir)) | ||
38 | return | ||
39 | |||
40 | self.server = HTTPServer((self.host, self.port), HTTPRequestHandler) | ||
41 | if self.port == 0: | ||
42 | self.port = self.server.server_port | ||
43 | self.process = multiprocessing.Process(target=self.server.server_start, args=[self.root_dir, self.logger]) | ||
44 | |||
45 | # The signal handler from testimage.bbclass can cause deadlocks here | ||
46 | # if the HTTPServer is terminated before it can restore the standard | ||
47 | #signal behaviour | ||
48 | orig = signal.getsignal(signal.SIGTERM) | ||
49 | signal.signal(signal.SIGTERM, signal.SIG_DFL) | ||
50 | self.process.start() | ||
51 | signal.signal(signal.SIGTERM, orig) | ||
52 | |||
53 | if self.logger: | ||
54 | self.logger.info("Started HTTPService on %s:%s" % (self.host, self.port)) | ||
55 | |||
56 | |||
57 | def stop(self): | ||
58 | if hasattr(self, "server"): | ||
59 | self.server.server_close() | ||
60 | if hasattr(self, "process"): | ||
61 | self.process.terminate() | ||
62 | self.process.join() | ||
63 | if self.logger: | ||
64 | self.logger.info("Stopped HTTPService on %s:%s" % (self.host, self.port)) | ||
65 | |||