mirror of
https://github.com/theupdateframework/python-tuf
synced 2026-05-24 10:08:28 +00:00
3c80c5bcd0
Change the instances of "self.assertRaises" to "with self.assertRaises"
as using "self.assertRaises" can lead to long statements separated to
multiline expressions as pointed out by Jussi here:
https://github.com/theupdateframework/python-tuf/pull/1658#discussion_r741725382
On another hand "with self.assertRaises()" looks a lot better:
589ed9e0d4/tests/test_api.py (L131)
Signed-off-by: Martin Vrachev <mvrachev@vmware.com>
123 lines
3.3 KiB
Python
123 lines
3.3 KiB
Python
#!/usr/bin/env python
|
|
|
|
# Copyright 2020, TUF contributors
|
|
# SPDX-License-Identifier: MIT OR Apache-2.0
|
|
|
|
"""
|
|
<Program Name>
|
|
test_utils.py
|
|
|
|
<Author>
|
|
Martin Vrachev.
|
|
|
|
<Started>
|
|
October 21, 2020.
|
|
|
|
<Copyright>
|
|
See LICENSE-MIT OR LICENSE for licensing information.
|
|
|
|
<Purpose>
|
|
Provide tests for some of the functions in utils.py module.
|
|
"""
|
|
|
|
import os
|
|
import logging
|
|
import unittest
|
|
import socket
|
|
import sys
|
|
|
|
import tuf.unittest_toolbox as unittest_toolbox
|
|
|
|
from tests import utils
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class TestServerProcess(unittest_toolbox.Modified_TestCase):
|
|
|
|
def tearDown(self):
|
|
# Make sure we are calling clean on existing attribute.
|
|
if hasattr(self, 'server_process_handler'):
|
|
self.server_process_handler.clean()
|
|
|
|
|
|
def can_connect(self):
|
|
try:
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
sock.connect(('localhost', self.server_process_handler.port))
|
|
return True
|
|
except:
|
|
return False
|
|
finally:
|
|
# The process will always enter in finally even we return.
|
|
if sock:
|
|
sock.close()
|
|
|
|
|
|
def test_simple_server_startup(self):
|
|
# Test normal case
|
|
self.server_process_handler = utils.TestServerProcess(log=logger)
|
|
|
|
# Make sure we can connect to the server
|
|
self.assertTrue(self.can_connect())
|
|
|
|
|
|
def test_simple_https_server_startup(self):
|
|
# Test normal case
|
|
good_cert_path = os.path.join('ssl_certs', 'ssl_cert.crt')
|
|
self.server_process_handler = utils.TestServerProcess(log=logger,
|
|
server='simple_https_server.py', extra_cmd_args=[good_cert_path])
|
|
|
|
# Make sure we can connect to the server
|
|
self.assertTrue(self.can_connect())
|
|
self.server_process_handler.clean()
|
|
|
|
# Test when no cert file is provided
|
|
self.server_process_handler = utils.TestServerProcess(log=logger,
|
|
server='simple_https_server.py')
|
|
|
|
# Make sure we can connect to the server
|
|
self.assertTrue(self.can_connect())
|
|
self.server_process_handler.clean()
|
|
|
|
# Test with a non existing cert file.
|
|
non_existing_cert_path = os.path.join('ssl_certs', 'non_existing.crt')
|
|
self.server_process_handler = utils.TestServerProcess(log=logger,
|
|
server='simple_https_server.py',
|
|
extra_cmd_args=[non_existing_cert_path])
|
|
|
|
# Make sure we can connect to the server
|
|
self.assertTrue(self.can_connect())
|
|
|
|
|
|
def test_slow_retrieval_server_startup(self):
|
|
# Test normal case
|
|
self.server_process_handler = utils.TestServerProcess(log=logger,
|
|
server='slow_retrieval_server.py')
|
|
|
|
# Make sure we can connect to the server
|
|
self.assertTrue(self.can_connect())
|
|
|
|
|
|
def test_cleanup(self):
|
|
# Test normal case
|
|
self.server_process_handler = utils.TestServerProcess(log=logger,
|
|
server='simple_server.py')
|
|
|
|
self.server_process_handler.clean()
|
|
|
|
# Check if the process has successfully been killed.
|
|
self.assertFalse(self.server_process_handler.is_process_running())
|
|
|
|
|
|
def test_server_exit_before_timeout(self):
|
|
with self.assertRaises(utils.TestServerProcessError):
|
|
utils.TestServerProcess(logger, server='non_existing_server.py')
|
|
|
|
# Test starting a server which immediately exits."
|
|
with self.assertRaises(utils.TestServerProcessError):
|
|
utils.TestServerProcess(logger, server='fast_server_exit.py')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
utils.configure_test_logging(sys.argv)
|
|
unittest.main()
|