def test_remove_files(self):
self.tmp_dir = tempfile.mkdtemp()
self.assertEqual(len(os.listdir(self.tmp_dir)), 0)
file = os.path.join(self.tmp_dir, 'file_to_be_removed.txt')
with open(file, 'w') as f:
f.write('this file will be removed')
self.assertEqual(len(os.listdir(self.tmp_dir)), 1)
Server.remove_files(self.tmp_dir)
self.assertEqual(len(os.listdir(self.tmp_dir)), 0)
shutil.rmtree(self.tmp_dir)
def test_split_parent_and_reply_messages(self):
archive_index = ArchiveMessageIndex(MessageDao())
messages = archive_index.update('test_data')
parents, replies = Server.split_parent_and_reply_messages(messages)
self.assertEqual(len(parents), 2)
self.assertEqual(len(replies), 6)
expected_parents = [
'[PATCH v2 1/2] Input: i8042 - Prevent intermixing i8042 commands',
'[PATCH v2 0/4] kselftests/arm64: add PAuth tests'
]
expected_replies = [
'Re: [PATCH] Remove final reference to superfluous smp_commence().',
'[PATCH v2 1/3] dmaengine: add dma_get_channel_caps()',
'[PATCH v2 1/4] kselftests/arm64: add a basic Pointer Authentication test',
'[PATCH v2 2/4] kselftests/arm64: add nop checks for PAuth tests',
'[PATCH v2 3/4] kselftests/arm64: add PAuth test for whether exec() changes keys',
'[PATCH v2 4/4] kselftests/arm64: add PAuth tests for single threaded consistency and key uniqueness'
]
def compare_message_subject(messages: List[Message],
subjects: List[str]):
for message, subject in zip(messages, subjects):
self.assertEqual(message.subject, subject)
compare_message_subject(parents, expected_parents)
compare_message_subject(replies, expected_replies)
def test_server_upload_across_batches(self, mock_upload_comments,
mock_upload_messages,
mock_fill_message_directory):
archive_index = ArchiveMessageIndex(MessageDao())
messages = archive_index.update('test_data')
first_batch = messages[0:6]
second_batch = messages[6:]
mock_fill_message_directory.return_value = ''
# declaring mock objects here because I want to use the ArchiveMessageIndex functionality to build the test data
with mock.patch.object(ArchiveMessageIndex,
'update') as mock_update, mock.patch.object(
MessageDao, 'get') as mock_get:
mock_update.side_effect = [first_batch, second_batch]
mock_get.side_effect = [None, None, messages[6], messages[7]]
server = Server()
server.update_convert_upload()
mock_upload_messages.assert_called_with(
[messages[2].id, messages[3].id])
mock_upload_comments.assert_called_with(set())
server.update_convert_upload()
mock_upload_messages.assert_called_with(
[messages[6].id, messages[7].id])
mock_upload_comments.assert_called_with(set())
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()
import os
from main import Server
conf = os.path.join(os.path.dirname(__file__), "settings.conf")
if __name__ == '__main__':
import cherrypy
cherrypy.quickstart(Server(), config=conf)
def setUp(self):
args = parse_args()
self.server = Server((args.a, args.p))
def create_app(loop):
return Server(loop=loop)
from main import Server
from handlers.config import read_config
if __name__ == '__main__':
server = Server(read_config("/etc/httpd.conf"))
server.server_init()
async def get_application(self, loop):
self.server = Server(loop=loop)
return self.server.app
from main import Server
server = Server()
while (True):
server.send(raw_input())
def server_start_stop_command(self):
"""
Start the server
:return: None
"""
if self.ui.server.server_is_on:
if not self.player_2_is_connected or tkinter.messagebox.askquestion(
self.ui.translation.get_translation(
"server_dialog_server_stop_title"),
self.ui.translation.get_translation(
"server_dialog_server_stop_message")
) == tkinter.messagebox.YES:
if self.ui.server.stop_server():
self.server_port_spin_box.configure(
state=tkinter.tix.NORMAL, increment=1)
self.server_start_stop.configure(
text=self.ui.translation.get_translation(
"server_configuration_start"))
self.server_state_label.configure(
text=self.ui.translation.get_translation(
"server_state_stopped"),
fg="#ee2e31")
else:
try:
port = int(self.server_port_spin_box.get())
except ValueError:
tkinter.messagebox.showerror(
self.ui.translation.get_translation(
"server_configuration_dialog_port_error_title"),
self.ui.translation.get_translation(
"server_configuration_dialog_port_error_message"))
else:
if 3000 <= port <= 3020:
self.ui.server = Server.Server(
port=port,
max_clients_connected=1,
on_message_function=self.server_on_message,
on_client_connect_function=self.
server_on_client_connect_function,
on_client_disconnect_function=self.
server_on_client_disconnect_function)
result = self.ui.server.start_server()
if result[0]:
self.server_port_spin_box.configure(state="readonly",
increment=0)
self.server_start_stop.configure(
text=self.ui.translation.get_translation(
"server_configuration_stop"))
self.server_state_label.configure(
text=self.ui.translation.get_translation(
"server_state_started"),
fg="#78bc61")
else:
error_name = errno.errorcode[result[1]]
if self.ui.translation.translation_exist(
SERVER_ERROR_DIALOG_KEY_FORMAT.format(
error_name)):
tkinter.messagebox.showerror(
self.ui.translation.get_translation(
"server_start_error_dialog_title").format(
result[1]),
self.ui.translation.get_translation(
SERVER_ERROR_DIALOG_KEY_FORMAT.format(
error_name)))
else:
tkinter.messagebox.showerror(
self.ui.translation.get_translation(
"server_start_error_dialog_title").format(
result[1]),
self.ui.translation.get_translation(
"server_start_error_unknown_dialog_message"
).format(result[1], error_name,
result[2].strerror))
else:
tkinter.messagebox.showerror(
self.ui.translation.get_translation(
"server_configuration_dialog_port_error_title"),
self.ui.translation.get_translation(
"server_configuration_dialog_port_error_message"))
from main import Server from dotenv import load_dotenv load_dotenv() app = Server().config_app()