def test_create_fail_if_command_name_empty(client):
error_message = "Field may not be empty."
with pytest.raises(ValidationError, match=error_message):
custom_command_processor(
user_id=user_id,
team_id=team_id,
channel_id="1",
command_name="",
)
def test_create_fail_if_number_of_items_to_select_is_too_high(client):
error_message = "Selecting more than 50 at once is prohibited."
with pytest.raises(ValidationError, match=error_message):
custom_command_processor(
user_id=user_id,
team_id=team_id,
channel_id="1",
command_name="sqd",
number_of_items_to_select=51,
)
def test_create_fail_if_number_of_items_to_select_is_0(client):
error_message = "Must select at least 1 item."
with pytest.raises(ValidationError, match=error_message):
custom_command_processor(
user_id=user_id,
team_id=team_id,
channel_id="1",
command_name="sqd",
number_of_items_to_select=0,
)
def test_custom_only_active_users_error(command_with_no_active_users):
error_message = "No active users to select found."
with pytest.raises(MissingElementError, match=error_message):
custom_command_processor(
command_name=command_with_no_active_users.name,
additional_text="",
number_of_items_to_select=1,
channel_id=command_with_no_active_users.channel_id,
team_id=team_id,
user_id=user_id,
)
def test_custom_command_self_exclude_error(command_for_self_exclude_error):
error_message = "Pick list contains only the user using the command.*selfExclude.*True.*" # noqa E501
with pytest.raises(BadRequestError, match=error_message):
custom_command_processor(
command_name=command_for_self_exclude_error.name,
additional_text="",
number_of_items_to_select=1,
channel_id=command_for_self_exclude_error.channel_id,
team_id=team_id,
user_id=user_id,
)
def workflow_step_execute_processor(
*,
command_name: str,
team_id: str,
user_id: str,
channel_id: str,
**kwargs,
) -> dict[str, any]:
custom_command_response = custom_command_processor(
command_name=command_name,
channel_id=channel_id,
team_id=team_id,
user_id=user_id,
should_update_weight_list=True,
**kwargs,
)
outputs = {}
for index, selected_item in enumerate(
custom_command_response.get("selected_items")):
outputs[create_select_item_name(index)] = selected_item
outputs[OutputVariable.SELECTION_MESSAGE.value] = (
custom_command_response.get("message").content
if custom_command_response.get("message") else "")
return {
"outputs": outputs,
**custom_command_response,
}
def resubmit_command_and_delete_message_processor(
wheel_ts: str = None, *, ts: str, **kwargs
) -> dict[str, any]:
delete_message_processor(ts=ts, **kwargs)
if wheel_ts:
delete_message_processor(ts=wheel_ts, **kwargs)
return custom_command_processor(**kwargs, should_update_weight_list=True)
def test_custom_command_only_active_users(command_for_active_users):
response = custom_command_processor(
command_name=command_for_active_users.name,
additional_text="",
number_of_items_to_select=1,
channel_id=command_for_active_users.channel_id,
team_id=team_id,
user_id=user_id,
)
message = response.get("message")
assert f"<@{user_id}|name>" in message.content
selected_items = response.get("selected_items")
assert selected_items == [f"<@{user_id}|name>"]
def test_custom_command_self_exclude(command_for_self_exclude):
response = custom_command_processor(
command_name=command_for_self_exclude.name,
additional_text="",
number_of_items_to_select=1,
channel_id=command_for_self_exclude.channel_id,
team_id=team_id,
user_id=user_id,
)
message = response.get("message")
assert "choose 2" in message.content
selected_items = response.get("selected_items")
assert selected_items == ["2"]
def test_custom_command(basic_command):
response = custom_command_processor(
command_name=basic_command.name,
additional_text="",
number_of_items_to_select=1,
channel_id=basic_command.channel_id,
team_id=team_id,
user_id=user_id,
)
message = response.get("message")
assert f"Hey ! <@{user_id}> choose " in message.content
assert basic_command.label in message.content
assert message.visibility == MessageVisibility.NORMAL
assert message.as_attachment is False
def test_custom_command_with_no_wheel(basic_command):
response = custom_command_processor(
command_name=basic_command.name,
additional_text="",
number_of_items_to_select=1,
channel_id=basic_command.channel_id,
team_id=team_id,
user_id=user_id,
with_wheel=False,
)
with_wheel = response.get("with_wheel")
assert with_wheel is False
gif_frames = response.get("gif_frames")
assert gif_frames is None
def test_custom_command_with_wheel(basic_command):
response = custom_command_processor(
command_name=basic_command.name,
additional_text="",
number_of_items_to_select=1,
channel_id=basic_command.channel_id,
team_id=team_id,
user_id=user_id,
with_wheel=True,
)
with_wheel = response.get("with_wheel")
assert with_wheel is True
gif_frames = response.get("gif_frames")
assert gif_frames.shape == (NB_FRAMES, WHEEL_HEIGHT,
WHEEL_WIDTH + LEGEND_WIDTH, 3)
def test_custom_command_multi_select(
additional_text,
number_of_items_to_select,
expected_message,
expected_items,
basic_command,
set_seed,
):
response = custom_command_processor(
command_name=basic_command.name,
additional_text=additional_text,
number_of_items_to_select=number_of_items_to_select,
channel_id=basic_command.channel_id,
team_id=team_id,
user_id=user_id,
)
message = response.get("message")
assert expected_message in message.content
selected_items = response.get("selected_items")
assert selected_items == expected_items