def test_server(self):
af_inet = AF_INET
sock_stream = SOCK_STREAM
sol_socket = SOL_SOCKET
so_reuseaddr = SO_REUSEADDR
with mock.patch("tcp_server.socket") as mock_socket, \
mock.patch("tcp_server.ClientThread", spec=True) as mock_ClientThread:
host = 'localhost'
port = 5555
tcp_server = TcpServer(host=host, port=port)
self.assertEqual(tcp_server.host, host)
self.assertEqual(tcp_server.port, port)
self.assertIsNone(tcp_server._socket)
self.assertFalse(tcp_server._runnning)
def accept_stub():
ca = [("conn1", "addr1"), ("conn2", "addr2"), ("conn3", "addr3")]
for i in range(len(ca)):
if i == 2:
tcp_server.stop()
yield ca[i]
s_socket = mock.Mock(name="s_sock")
s_socket.accept.side_effect = accept_stub()
mock_socket.socket.return_value = s_socket
mock_socket.AF_INET = af_inet
mock_socket.SOCK_STREAM = sock_stream
mock_socket.SOL_SOCKET = sol_socket
mock_socket.SO_REUSEADDR = so_reuseaddr
tcp_server.run()
mock_socket.socket.assert_called_once_with(af_inet, sock_stream)
tcp_server._socket.setsockopt.assert_called_once_with(sol_socket, so_reuseaddr, 1)
tcp_server._socket.bind.assert_called_once_with((host, port))
tcp_server._socket.listen.assert_called_once_with(5)
self.assertFalse(tcp_server._runnning)
self.assertEqual(tcp_server._socket.accept.call_count, 3)
self.assertEqual(mock_ClientThread.call_count, 3)
inst_client_thread = mock_ClientThread("stub", "stub")
self.assertEqual(inst_client_thread.start.call_count, 3)
tcp_server._socket.close.assert_called_once()
}
################################################################################
# Main #
################################################################################
script_name = os.path.basename(sys.argv[0])
if len(sys.argv) != 2:
print_usage(script_name)
sys.exit(-1)
service_name = sys.argv[1]
if service_name not in service_table:
print_usage(script_name)
sys.exit(-1)
try:
init_logger(log_file_name, log_max_bytes, log_backup_count)
logger = logging.getLogger()
logger.info(f'SW version: {sw_version}')
tcp_server = TcpServer("svc-server")
tcp_server.run(port, ClientHandlerFactory(service_table[service_name]))
except Exception as ex:
logger.error(traceback.format_exc())
finally:
logger.info('Server stopped.')
logging.shutdown()
"""
PyPepperChecker module
"""
from tcp_server import TcpServer
from network import Network
OUTPUT_SIZE = 3
INPUT_SIZE = 172 * 229 * 3
print("Building neural network...")
network = Network(INPUT_SIZE, OUTPUT_SIZE)
print "Processing images samples..."
network.process_images()
print "Training network..."
network.train()
print("Running TCP server...")
tcp_server = TcpServer(network)
tcp_server.run()
