Пример #1
0
def test_whitelistbot_directmsgs() -> None:
    bot = WhiteListBot(master=None, whitelist=[], whitelist_directmsgs=True)
    r1 = m.RoomRef2(roomType=m.RoomType.PUBLIC,
                    roomParticipant=True,
                    roomName='room1')
    r2 = m.RoomRef2(roomType=m.RoomType.DIRECT, roomParticipant=True)

    assert bot.is_applicable(r1) is False
    assert bot.is_applicable(r2) is True
Пример #2
0
def test_blacklistbot() -> None:
    bot = BlacklistBot(master=None, blacklist=['room1'])
    r1 = m.RoomRef2(roomType=m.RoomType.PUBLIC,
                    roomParticipant=True,
                    roomName='room1')
    r2 = m.RoomRef2(roomType=m.RoomType.PUBLIC,
                    roomParticipant=True,
                    roomName='room2')

    assert bot.is_applicable(r1) is False
    assert bot.is_applicable(r2) is True
Пример #3
0
def test_roomtypebot_public() -> None:
    bot = RoomTypeBot(master=None, enable_public_channel=True)

    r1 = m.RoomRef2(roomType=m.RoomType.PUBLIC,
                    roomParticipant=True,
                    roomName='room1')
    r2 = m.RoomRef2(roomType=m.RoomType.PRIVATE,
                    roomParticipant=True,
                    roomName='room2')
    r3 = m.RoomRef2(roomType=m.RoomType.DIRECT,
                    roomParticipant=True,
                    roomName='room3')

    assert bot.is_applicable(r1) is True
    assert bot.is_applicable(r2) is False
    assert bot.is_applicable(r3) is False
Пример #4
0
def test_roomtypebot_multiple() -> None:
    bot = RoomTypeBot(master=None,
                      enable_private_group=True,
                      enable_direct_message=True)

    r1 = m.RoomRef2(roomType=m.RoomType.PUBLIC,
                    roomParticipant=True,
                    roomName='room1')
    r2 = m.RoomRef2(roomType=m.RoomType.PRIVATE,
                    roomParticipant=True,
                    roomName='room2')
    r3 = m.RoomRef2(roomType=m.RoomType.DIRECT,
                    roomParticipant=True,
                    roomName='room3')

    assert bot.is_applicable(r1) is False
    assert bot.is_applicable(r2) is True
    assert bot.is_applicable(r3) is True