コード例 #1
0
def test_call__route_list_blocklisted_tags(context):
    update = MockTelegramUpdate.with_message(chat_id=14358,
                                             text="/list_blocklisted_tags")
    api = MockExportAPI()
    watcher = SubscriptionWatcher(api, context.bot)
    func = BlocklistFunctionality(watcher)
    list_tags = MockMethod("Listing blocklisted tags")
    func._list_blocklisted_tags = list_tags.call

    func.call(update, context)

    assert list_tags.called
    assert list_tags.args is not None
    assert list_tags.args[0] == 14358
    context.bot.send_message.assert_called()
    assert context.bot.send_message.call_args[1][
        'chat_id'] == update.message.chat_id
    assert context.bot.send_message.call_args[1][
        'text'] == "Listing blocklisted tags"
コード例 #2
0
def test_call__route_add_blocklisted_tag(context):
    update = MockTelegramUpdate.with_message(chat_id=14358,
                                             text="/add_blocklisted_tag test")
    api = MockExportAPI()
    watcher = SubscriptionWatcher(api, context.bot)
    func = BlocklistFunctionality(watcher)
    add_tag = MockMethod("Added to blocklist: test")
    func._add_to_blocklist = add_tag.call

    func.call(update, context)

    assert add_tag.called
    assert add_tag.args is not None
    assert add_tag.args[0] == 14358
    assert add_tag.args[1] == "test"
    context.bot.send_message.assert_called()
    assert context.bot.send_message.call_args[1][
        'chat_id'] == update.message.chat_id
    assert context.bot.send_message.call_args[1][
        'text'] == "Added to blocklist: test"
コード例 #3
0
def test_call__route_remove_blocklisted_tag(context):
    update = MockTelegramUpdate.with_message(
        chat_id=14358, text="/remove_blocklisted_tag example")
    api = MockExportAPI()
    watcher = SubscriptionWatcher(api, context.bot)
    func = BlocklistFunctionality(watcher)
    remove_tag = MockMethod("Removed from blocklist: example")
    func._remove_from_blocklist = remove_tag.call

    func.call(update, context)

    assert remove_tag.called
    assert remove_tag.args is not None
    assert remove_tag.args[0] == 14358
    assert remove_tag.args[1] == "example"
    context.bot.send_message.assert_called()
    assert context.bot.send_message.call_args[1][
        'chat_id'] == update.message.chat_id
    assert context.bot.send_message.call_args[1][
        'text'] == "Removed from blocklist: example"