class TestServer(unittest.TestCase): def setUp(self): args = parse_args() self.server = Server((args.a, args.p)) def test_create_response_success(self): self.assertEqual( self.server.create_response({ 'action': 'presence', 'time': time.time() }), {'response': 200}) def test_create_response_error(self): self.assertEqual( self.server.create_response({ 'action': 'random', 'time': time.time() }), { 'response': 400, 'error': 'Не верный запрос.' }) def tearDown(self): self.server.close()
from server.server import Server import time if __name__ == "__main__": server = Server() server.listen() server.close()