async def on_message(message):
if message.author.bot:
return
if Help().is_help(message.content):
await message.channel.send(Help().get_help_mes(message.content))
elif TeamSplitter().is_team_command(message.content):
mes = TeamSplitter().create_teams(client, message)
await message.channel.send(mes)
def test_modify_user_list(message, user_names, expect):
# modify_user_listは内部でリストをsetに変換しているので順番が入れ替わる
assert set(TeamSplitter().modify_user_list(user_names, message)) == set(expect)
def test_normal_result(teams, expect):
assert TeamSplitter().normal_result(teams) == expect
def test_lol_result(teams, expect):
assert TeamSplitter().lol_result(teams) == expect
def test_create_team_headder(team_num, team_size, expect):
assert TeamSplitter().create_team_headder(team_num, team_size) == expect
def test_set_team_size(message, expect):
assert TeamSplitter().set_team_size(message) == expect
def test_create_teams(mocker, vc, message, expect):
mocker.patch("lib.util.Util.GetAuthorVChannel", return_value=vc)
mocker.patch("random.sample", side_effect=rand_return_value)
# modify_user_listは内部でsetを使っているので、利用すると順番が入れ替わる
mocker.patch("lib.team_splitter.TeamSplitter.modify_user_list", side_effect=modify_user_list_mock)
assert TeamSplitter().create_teams(mock.Mock(), message) == expect
def test_is_team_command(message, expect):
assert TeamSplitter().is_team_command(message) == expect
def test_lol_team(users, expect):
assert TeamSplitter().make_lol_team(users) == expect
def test_split_list(users, team_num, expect):
assert TeamSplitter().split_list(users, team_num) == expect
import pytest
import re
from unittest import mock
from tests.create_mock import CreateMock
from lib.team_splitter import TeamSplitter
@pytest.mark.parametrize("message, expect", [
(CreateMock().message("/team -num 1" ), 1),
(CreateMock().message("/team -num 10"), 10),
(CreateMock().message("/team -num 1a"), 1),
(CreateMock().message("/team-num 3" ), 3),
(CreateMock().message("" ), TeamSplitter().DEFAULT_TEAM_NUM),
(CreateMock().message("/team -num a1"), TeamSplitter().DEFAULT_TEAM_NUM),
])
def test_set_team_num(message, expect):
assert TeamSplitter().set_team_num(message) == expect
@pytest.mark.parametrize("message, expect", [
(CreateMock().message("/team -size 1" ), 1),
(CreateMock().message("/team -size 10"), 10),
(CreateMock().message("/team -size 1a"), 1),
(CreateMock().message("/team-size 3" ), 3),
(CreateMock().message("" ), TeamSplitter().DEFAULT_TEAM_SIZE),
(CreateMock().message("/team -size a1"), TeamSplitter().DEFAULT_TEAM_SIZE),
])
def test_set_team_size(message, expect):
assert TeamSplitter().set_team_size(message) == expect
@pytest.mark.parametrize("message, user_names, expect", [
(
CreateMock().message("/team"),