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
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
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
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