def test_channel_offline(self):
self.twitch.live_ch_list = {
'testch1': {
'status': u'test_status',
'game': u'test_game',
'url': u'test_url'
}
}
fake_sender = CommandSubmitter('test', 'fake_sender')
self.sublist = {fake_sender: ['testch1']}
self.checker.start()
threading.Event().wait(1)
self.twitch.live_ch_list = {}
self.checker.refresh()
threading.Event().wait(1)
# no message should be sent
ok_(len(interfaces.get('test').msg_queue[fake_sender]) == 0)
# simulate channel goes live
self.twitch.live_ch_list = {
'testch1': {
'status': u'test_status',
'game': u'test_game',
'url': u'test_url'
}
}
self.checker.refresh()
threading.Event().wait(1)
self.checker.stop()
ok_(
'testch1'
in ' '.join(interfaces.get('test').msg_queue[fake_sender]),
interfaces.get('test').msg_queue[fake_sender])
def test_channel_offline(self):
self.twitch.live_ch_list = {
'testch1': {
'status': u'test_status',
'game': u'test_game',
'url': u'test_url'
}
}
fake_sender = CommandSubmitter('test', 'fake_sender')
self.sublist = {fake_sender: ['testch1']}
self.checker.start()
threading.Event().wait(1)
self.twitch.live_ch_list = {}
self.checker.refresh()
threading.Event().wait(1)
# no message should be sent
ok_(len(interfaces.get('test').msg_queue[fake_sender]) == 0)
# simulate channel goes live
self.twitch.live_ch_list = {
'testch1': {
'status': u'test_status',
'game': u'test_game',
'url': u'test_url'
}
}
self.checker.refresh()
threading.Event().wait(1)
self.checker.stop()
ok_('testch1' in ' '.join(interfaces.get('test').msg_queue[fake_sender]),
interfaces.get('test').msg_queue[fake_sender])
def add_test(self, test_list):
for sender_code, cmd, cmd_chk, msg_chk in test_list:
sender = CommandSubmitter(sender_code, self.if_name)
interfaces.get(self.in_name).add_command(sender, cmd)
self.senders[self.total_test_count] = sender
self.cmd_checkers[self.total_test_count] = cmd_chk
self.msg_checkers[self.total_test_count] = msg_chk
self.total_test_count += 1
def test_list_channel(self):
self.service._twitch.set_live_channel_list({'testch2': {'display_name': 'TESTCH2'}})
self.test_subscribe_multi()
fake_sender = CommandSubmitter('test', 'fake_sender')
self.service._list_channel(True, fake_sender)
ok_(' '.join(interfaces.get('test').msg_queue[fake_sender]).count('LIVE') == 1)
check_msg = ' '.join(interfaces.get('test').msg_queue[fake_sender])
ok_('testch2' in check_msg.lower())
ok_('testch1' in check_msg.lower())
def test_list_channel(self):
self.service._twitch.set_live_channel_list(
{'testch2': {
'display_name': 'TESTCH2'
}})
self.test_subscribe_multi()
fake_sender = CommandSubmitter('test', 'fake_sender')
self.service._list_channel(True, fake_sender)
ok_(' '.join(interfaces.get('test').msg_queue[fake_sender]).count(
'LIVE') == 1)
check_msg = ' '.join(interfaces.get('test').msg_queue[fake_sender])
ok_('testch2' in check_msg.lower())
ok_('testch1' in check_msg.lower())
def test_subscribe_one(self):
try:
os.remove(self.service.SUB_FILE)
except OSError:
pass
self.service.start()
threading.Event().wait(.1)
self.service.stop()
self.service._twitch.set_exists_channel_list(['testch1', 'testch2'])
fake_sender = CommandSubmitter('test', 'fake_sender')
self.service._subscribe(['testch1'], fake_sender)
ok_(
'testch1'
in [x.lower() for x in self.service._sublist[fake_sender]],
'sublist = ' + str(self.service._sublist[fake_sender]))
ok_('not found' not in ' '.join(
interfaces.get('test').msg_queue[fake_sender]))
ok_(self.service._channel_sub_count['testch1'] == 1)
fake_sender2 = CommandSubmitter('test', 'fake_sender2')
self.service._subscribe(['testch1'], fake_sender2)
ok_(self.service._channel_sub_count['testch1'] == 2)
def test_print_help(self):
def cmd_process(args, sender):
pass
test_str = 'testtesttest'
lap = LinotArgParser('testcmd', self.parser, cmd_process)
lap.add_argument('-a', action='store_true', help=test_str)
lap.add_argument('-noshow',
action='store_true',
help=argparse.SUPPRESS)
lap.add_argument('-showme', action='store_true', help='')
# test -h and --help goes to print_help
args, unknown_args = self.parser.parse_known_args('testcmd -h'.split())
eq_(len(unknown_args), 0)
sender = CommandSubmitter('test', 'test_sender')
args.proc(args, sender)
msg_queue = interfaces.get('test').msg_queue
ok_('command list' in ' '.join(msg_queue[sender.code]))
args, unknown_args = self.parser.parse_known_args(
'testcmd --help'.split())
eq_(len(unknown_args), 0)
interfaces.get('test').reset()
args.proc(args, sender)
msg_queue = interfaces.get('test').msg_queue
ok_('command list' in ' '.join(msg_queue[sender.code]))
# add help
args, unknown_args = self.parser.parse_known_args('testcmd -h'.split())
interfaces.get('test').reset()
args.proc(args, sender)
msg_queue = interfaces.get('test').msg_queue
msg = ' '.join(msg_queue[sender.code])
ok_(test_str in msg, msg)
ok_('-nowshow' not in msg)
ok_('-showme' in msg)
# Test help suppress if sender not found (for coverage)
args, unknown_args = self.parser.parse_known_args('testcmd -h'.split())
interfaces.get('test').reset()
args.proc(args, None)
msg_queue = interfaces.get('test').msg_queue
msg = ' '.join(msg_queue[sender.code])
ok_(msg is '', msg)
def test_cmd_process(self):
test_sender = CommandSubmitter('test', 'mock')
self.service._cmd_process(None, test_sender)
test_if = interfaces.get('test')
# cmd_process arg=None indicates the msg contains no recognized command
# we should return something to let user know
ok_('Unknown' in test_if.msg_queue[test_sender.code][0])
test_if.reset()
self.service._cmd_process('test', test_sender)
ok_('not implemented' in test_if.msg_queue[test_sender.code][0])
def test_cmd_process(self):
test_sender = CommandSubmitter('test', 'mock')
self.service._cmd_process(None, test_sender)
test_if = interfaces.get('test')
# cmd_process arg=None indicates the msg contains no recognized command
# we should return something to let user know
ok_('Unknown' in test_if.msg_queue[test_sender.code][0])
test_if.reset()
self.service._cmd_process('test', test_sender)
ok_('not implemented' in test_if.msg_queue[test_sender.code][0])
def test_unknown_commands(self):
# test command server can properly handle 1 or multiple commands
def default_process(args, sender):
ok_(getattr(args, sender.code) is True)
default_process.called += 1
lap = LinotArgParser('testcmd', self.parser, default_process)
lap.add_argument('-a', action='store_true')
lap.add_argument('-b', action='store_true')
lap.add_argument('-c', action='store_true')
# Test 1 unknown command, command server respose to user
sender_a = CommandSubmitter('test', 'a')
fake_cmd = [
(sender_a, 'some_unknown_words'),
]
interfaces.get('test').reset()
interfaces.get('test').add_command_list(fake_cmd)
command_server.start(self.parser, ['test'])
default_process.called = 0
threading.Event().wait(.5)
command_server.stop()
ok_(default_process.called == 0)
ok_('Unknown' in ' '.join(interfaces.get('test').msg_queue[sender_a.code]))
# Test multiple unknown commands
sender_a = CommandSubmitter('test', 'a')
sender_u = CommandSubmitter('test', 'u')
fake_cmd = [
(sender_u, 'some_unknown_cmds'),
(sender_a, 'testcmd -a'),
(sender_u, 'some_unknown_cmds'),
]
interfaces.get('test').reset()
interfaces.get('test').add_command_list(fake_cmd)
command_server.start(self.parser, ['test'])
default_process.called = 0
threading.Event().wait(.5)
command_server.stop()
unknown_response = ' '.join(interfaces.get('test').msg_queue[sender_u.code])
ok_(unknown_response.count('Unknown') == 2)
ok_(default_process.called == 1)
def test_print_help(self):
def cmd_process(args, sender):
pass
test_str = 'testtesttest'
lap = LinotArgParser('testcmd', self.parser, cmd_process)
lap.add_argument('-a', action='store_true', help=test_str)
lap.add_argument('-noshow', action='store_true', help=argparse.SUPPRESS)
lap.add_argument('-showme', action='store_true', help='')
# test -h and --help goes to print_help
args, unknown_args = self.parser.parse_known_args('testcmd -h'.split())
eq_(len(unknown_args), 0)
sender = CommandSubmitter('test', 'test_sender')
args.proc(args, sender)
msg_queue = interfaces.get('test').msg_queue
ok_('command list' in ' '.join(msg_queue[sender.code]))
args, unknown_args = self.parser.parse_known_args('testcmd --help'.split())
eq_(len(unknown_args), 0)
interfaces.get('test').reset()
args.proc(args, sender)
msg_queue = interfaces.get('test').msg_queue
ok_('command list' in ' '.join(msg_queue[sender.code]))
# add help
args, unknown_args = self.parser.parse_known_args('testcmd -h'.split())
interfaces.get('test').reset()
args.proc(args, sender)
msg_queue = interfaces.get('test').msg_queue
msg = ' '.join(msg_queue[sender.code])
ok_(test_str in msg, msg)
ok_('-nowshow' not in msg)
ok_('-showme' in msg)
# Test help suppress if sender not found (for coverage)
args, unknown_args = self.parser.parse_known_args('testcmd -h'.split())
interfaces.get('test').reset()
args.proc(args, None)
msg_queue = interfaces.get('test').msg_queue
msg = ' '.join(msg_queue[sender.code])
ok_(msg is '', msg)
def test_subscribe_one_not_found(self):
self.service.start()
threading.Event().wait(.1)
self.service.stop()
self.service._twitch.set_exists_channel_list(['testch2', 'testch3'])
fake_sender = CommandSubmitter('test', 'fake_sender')
self.service._subscribe(['testch1'], fake_sender)
ok_('testch1' not in self.service._sublist[fake_sender],
'sublist = '+str(self.service._sublist[fake_sender]))
ok_('not found' in ' '.join(interfaces.get('test').msg_queue[fake_sender]))
def test_live_notification(self):
fake_sender = CommandSubmitter('test', 'fake_sender')
fake_sender2 = CommandSubmitter('test', 'fake_sender2')
self.sublist = {
fake_sender: ['testch1'],
fake_sender2: ['testch2'],
}
self.checker.start()
threading.Event().wait(1)
self.checker.refresh()
threading.Event().wait(1)
ok_(len(interfaces.get('test').msg_queue[fake_sender]) == 0)
ok_(len(interfaces.get('test').msg_queue[fake_sender2]) == 0)
self.twitch.live_ch_list = {
'TESTCH1': {
'status': u'test_status',
'game': u'test_game',
'url': u'test_url'
}
}
self.checker.refresh()
threading.Event().wait(1)
self.checker.stop()
ok_(
'TESTCH1'
in ' '.join(interfaces.get('test').msg_queue[fake_sender]),
interfaces.get('test').msg_queue[fake_sender])
ok_(
'TESTCH1'
not in ' '.join(interfaces.get('test').msg_queue[fake_sender2]),
interfaces.get('test').msg_queue[fake_sender2])
def test_live_notification(self):
fake_sender = CommandSubmitter('test', 'fake_sender')
fake_sender2 = CommandSubmitter('test', 'fake_sender2')
self.sublist = {
fake_sender: ['testch1'],
fake_sender2: ['testch2'],
}
self.checker.start()
threading.Event().wait(1)
self.checker.refresh()
threading.Event().wait(1)
ok_(len(interfaces.get('test').msg_queue[fake_sender]) == 0)
ok_(len(interfaces.get('test').msg_queue[fake_sender2]) == 0)
self.twitch.live_ch_list = {
'TESTCH1': {
'status': u'test_status',
'game': u'test_game',
'url': u'test_url'
}
}
self.checker.refresh()
threading.Event().wait(1)
self.checker.stop()
ok_('TESTCH1' in ' '.join(interfaces.get('test').msg_queue[fake_sender]),
interfaces.get('test').msg_queue[fake_sender])
ok_('TESTCH1' not in ' '.join(interfaces.get('test').msg_queue[fake_sender2]),
interfaces.get('test').msg_queue[fake_sender2])
def test_subscribe_one_not_found(self):
self.service.start()
threading.Event().wait(.1)
self.service.stop()
self.service._twitch.set_exists_channel_list(['testch2', 'testch3'])
fake_sender = CommandSubmitter('test', 'fake_sender')
self.service._subscribe(['testch1'], fake_sender)
ok_('testch1' not in self.service._sublist[fake_sender],
'sublist = ' + str(self.service._sublist[fake_sender]))
ok_('not found' in ' '.join(
interfaces.get('test').msg_queue[fake_sender]))
def test_unsub_all(self):
sender = CommandSubmitter('test', 'sender')
test_channels = ['testch1', 'testch2', 'testch3']
self.service._twitch.set_exists_channel_list(test_channels)
self.service.setup(self.parser)
self.service.start()
threading.Event().wait(1)
self.service.stop()
self.service._subscribe(test_channels, sender)
self.service._list_channel(True, sender)
# make sure subscribe success
for ch in test_channels:
ok_(ch.lower() in ' '.join(interfaces.get('test').msg_queue[sender]).lower())
interfaces.get('test').reset()
self.service._unsub_all(True, sender)
self.service._list_channel(True, sender)
for ch in test_channels:
ok_(ch.lower() not in ' '.join(interfaces.get('test').msg_queue[sender]).lower())
def test_unsubscribe_multi_not_found(self):
self.test_subscribe_one()
fake_sender = CommandSubmitter('test', 'fake_sender')
self.service._subscribe(['testch2'], fake_sender)
self.service._unsubscribe(['testch3', 'testch1', 'testch2'], fake_sender)
ok_('testch1' not in self.service._sublist[fake_sender],
'sublist = '+str(self.service._sublist[fake_sender]))
ok_('testch2' not in self.service._sublist[fake_sender],
'sublist = '+str(self.service._sublist[fake_sender]))
ok_('not found' in ' '.join(interfaces.get('test').msg_queue[fake_sender]))
ok_(self.service._channel_sub_count['testch1'] == 1)
def test_cmd_process(self):
# test command server can properly handle 1 or multiple commands
def default_process(args, sender):
ok_(getattr(args, sender.code) is True)
default_process.called += 1
lap = LinotArgParser('testcmd', self.parser, default_process)
lap.add_argument('-a', action='store_true')
lap.add_argument('-b', action='store_true')
lap.add_argument('-c', action='store_true')
# Test 1 cmd return by polling_command
# command = [(sender, cmd string), ...]
sender = CommandSubmitter('test', 'a')
fake_cmd = [(sender, 'testcmd -a')]
interfaces.get('test').add_command_list(fake_cmd)
default_process.called = 0
command_server.start(self.parser, ['test'])
threading.Event().wait(.5)
command_server.stop()
ok_(default_process.called == 1)
# Test 3 cmds return by polling_command
# command = [(sender, cmd string), ...]
sender_a = CommandSubmitter('test', 'a')
sender_b = CommandSubmitter('test', 'b')
sender_c = CommandSubmitter('test', 'c')
fake_cmd = [
(sender_a, 'testcmd -a'),
(sender_b, 'testcmd -b'),
(sender_c, 'testcmd -c'),
]
interfaces.get('test').add_command_list(fake_cmd)
command_server.start(self.parser, ['test'])
default_process.called = 0
threading.Event().wait(.5)
command_server.stop()
ok_(default_process.called == 3)
def test_unsubscribe_one_not_found(self):
self.test_subscribe_one()
fake_sender = CommandSubmitter('test', 'fake_sender')
self.service._subscribe(['testch2'], fake_sender)
self.service._unsubscribe(['testch3'], fake_sender)
ok_('testch1' in self.service._sublist[fake_sender],
'sublist = ' + str(self.service._sublist[fake_sender]))
ok_('testch2' in self.service._sublist[fake_sender],
'sublist = ' + str(self.service._sublist[fake_sender]))
ok_('not found' in ' '.join(
interfaces.get('test').msg_queue[fake_sender]))
ok_(self.service._channel_sub_count['testch1'] == 2)
def test_direct_command(self):
def default_process(args, sender):
ok_(False) # should not reach here
lap = LinotArgParser('testcmd', self.parser, default_process)
def cmd_checker(match_list, cmd, sender):
ok_('somechannel' in match_list)
ok_(len(match_list) == 1)
ok_(cmd == 'www.twitch.tv/somechannel')
cmd_checker.runned = True
lap.add_direct_command(cmd_checker, 'twitch\.tv/(\w+)[\s\t,]*', re.IGNORECASE)
cmd_checker.runned = False
sender = CommandSubmitter('test', 'sender')
fake_cmd = [
(sender, 'www.twitch.tv/somechannel')
]
interfaces.get('test').add_command_list(fake_cmd)
command_server.start(self.parser, ['test'])
threading.Event().wait(.5)
command_server.stop()
ok_(cmd_checker.runned)
def test_skip_already_lived_on_boot(self):
self.twitch.live_ch_list = {
'testch1': {
'status': 'test_status',
'game': 'test_game',
'url': 'test_url'
}
}
fake_sender = CommandSubmitter('test', 'fake_sender')
self.sublist = {fake_sender: ['testch1']}
self.checker.start()
threading.Event().wait(1)
self.checker.stop()
ok_(len(interfaces.get('test').msg_queue[fake_sender]) == 0)
def test_skip_already_lived_on_boot(self):
self.twitch.live_ch_list = {
'testch1': {
'status': 'test_status',
'game': 'test_game',
'url': 'test_url'
}
}
fake_sender = CommandSubmitter('test', 'fake_sender')
self.sublist = {fake_sender: ['testch1']}
self.checker.start()
threading.Event().wait(1)
self.checker.stop()
ok_(len(interfaces.get('test').msg_queue[fake_sender]) == 0)
def test_unsub_all(self):
sender = CommandSubmitter('test', 'sender')
test_channels = ['testch1', 'testch2', 'testch3']
self.service._twitch.set_exists_channel_list(test_channels)
self.service.setup(self.parser)
self.service.start()
threading.Event().wait(1)
self.service.stop()
self.service._subscribe(test_channels, sender)
self.service._list_channel(True, sender)
# make sure subscribe success
for ch in test_channels:
ok_(ch.lower() in ' '.join(
interfaces.get('test').msg_queue[sender]).lower())
interfaces.get('test').reset()
self.service._unsub_all(True, sender)
self.service._list_channel(True, sender)
for ch in test_channels:
ok_(ch.lower() not in ' '.join(
interfaces.get('test').msg_queue[sender]).lower())
def test_list_users_no_user(self):
# issue #10, list user hit exception while there is no user
config['interface']['test'] = {'admin_id': 'test_admin'}
try:
os.remove(self.service.SUB_FILE)
except OSError:
pass
self.service.start()
threading.Event().wait(.1)
self.service.stop()
fake_sender = CommandSubmitter('test', 'test_admin')
self.service._list_users([], fake_sender)
# check there is a response to user
ok_(len(''.join(interfaces.get('test').msg_queue[fake_sender])) > 0)
def test_subscribe_multi(self):
try:
os.remove(self.service.SUB_FILE)
except OSError:
pass
self.service.start()
threading.Event().wait(.1)
self.service.stop()
self.service._twitch.set_exists_channel_list(['testch1', 'testch2'])
fake_sender = CommandSubmitter('test', 'fake_sender')
self.service._subscribe(['testch1', 'testch2'], fake_sender)
ok_('testch1' in self.service._sublist[fake_sender],
'sublist = '+str(self.service._sublist[fake_sender]))
ok_('testch2' in self.service._sublist[fake_sender],
'sublist = '+str(self.service._sublist[fake_sender]))
ok_('not found' not in ' '.join(interfaces.get('test').msg_queue[fake_sender]))
def test_list_users_no_user(self):
# issue #10, list user hit exception while there is no user
config['interface']['test'] = {'admin_id': 'test_admin'}
try:
os.remove(self.service.SUB_FILE)
except OSError:
pass
self.service.start()
threading.Event().wait(.1)
self.service.stop()
fake_sender = CommandSubmitter('test', 'test_admin')
self.service._list_users([], fake_sender)
# check there is a response to user
ok_(len(''.join(interfaces.get('test').msg_queue[fake_sender])) > 0)
def setUp(self):
self.twitch = MockTwitchEngine()
self.checker = Checker(300, self.twitch, self.get_sublist)
self.sublist = {}
interfaces.get('test').reset()
def test_import(self):
sender = CommandSubmitter('test', 'tester')
short_test_channels = ['testch'+str(x) for x in range(3)]
long_test_channels = ['testch'+str(x) for x in range(10)]
self.service.setup(self.parser)
self.service.start()
threading.Event().wait(1)
self.service.stop()
# test small amount of followed channel
self.service._twitch.followed_ch_list = {'some_twitch_user': short_test_channels}
self.service._import(['some_twitch_user'], sender)
self.service._list_channel(True, sender)
for ch in short_test_channels:
ok_(ch.lower() in ' '.join(interfaces.get('test').msg_queue[sender]).lower())
interfaces.get('test').reset()
self.service._unsub_all(True, sender)
# test large amount of followed channel
self.service._twitch.followed_ch_list = {'some_twitch_user': long_test_channels}
self.service._import(['some_twitch_user'], sender)
self.service._list_channel(True, sender)
for ch in long_test_channels:
ok_(ch.lower() in ' '.join(interfaces.get('test').msg_queue[sender]).lower())
ok_('a while' in ' '.join(interfaces.get('test').msg_queue[sender]).lower())
interfaces.get('test').reset()
self.service._unsub_all(True, sender)
# test not found
self.service._twitch.followed_ch_list = {'some_twitch_user': long_test_channels}
self.service._import(['some_twitch_user2'], sender)
self.service._list_channel(True, sender)
for ch in long_test_channels:
ok_(ch.lower() not in ' '.join(interfaces.get('test').msg_queue[sender]).lower())
ok_('not found' in ' '.join(interfaces.get('test').msg_queue[sender]).lower())
interfaces.get('test').reset()
self.service._unsub_all(True, sender)
def setUp(self):
self.twitch = MockTwitchEngine()
self.checker = Checker(300, self.twitch, self.get_sublist)
self.sublist = {}
interfaces.get('test').reset()
def test_list_users(self):
self.test_subscribe_one()
self.service._subscribe(['testch2'], CommandSubmitter('test', 'fake_sender2'))
config['interface']['test'] = {'admin_id': 'test_admin'}
# check admin only
fake_sender = CommandSubmitter('test', 'fake_sender')
self.service._list_users([], fake_sender)
ok_('Channels' not in ''.join(interfaces.get('test').msg_queue[fake_sender]))
fake_sender = CommandSubmitter('test', 'test_admin')
self.service._list_users([], fake_sender)
# check msg response
ok_('fake_sender' in ''.join(interfaces.get('test').msg_queue[fake_sender]))
ok_('fake_sender2' in ''.join(interfaces.get('test').msg_queue[fake_sender]))
ok_('testch1' not in ''.join(interfaces.get('test').msg_queue[fake_sender]))
ok_('testch2' not in ''.join(interfaces.get('test').msg_queue[fake_sender]))
# test list single user, channels
interfaces.get('test').reset()
for index in range(2):
self.service._list_users([index], fake_sender)
msg = ' '.join(interfaces.get('test').msg_queue[fake_sender])
if 'fake_sender' in msg:
ok_('testch1' in msg)
elif 'fake_sender2' in msg:
ok_('testch2' in msg)
ok_('testch1' in msg)
# test list multiple user channels
interfaces.get('test').reset()
self.service._list_users(range(2), fake_sender)
ok_('fake_sender' in ''.join(interfaces.get('test').msg_queue[fake_sender]))
ok_('fake_sender2' in ''.join(interfaces.get('test').msg_queue[fake_sender]))
ok_('testch1' in ''.join(interfaces.get('test').msg_queue[fake_sender]))
ok_('testch2' in ''.join(interfaces.get('test').msg_queue[fake_sender]))
def test_list_users(self):
self.test_subscribe_one()
self.service._subscribe(['testch2'],
CommandSubmitter('test', 'fake_sender2'))
config['interface']['test'] = {'admin_id': 'test_admin'}
# check admin only
fake_sender = CommandSubmitter('test', 'fake_sender')
self.service._list_users([], fake_sender)
ok_('Channels' not in ''.join(
interfaces.get('test').msg_queue[fake_sender]))
fake_sender = CommandSubmitter('test', 'test_admin')
self.service._list_users([], fake_sender)
# check msg response
ok_('fake_sender' in ''.join(
interfaces.get('test').msg_queue[fake_sender]))
ok_('fake_sender2' in ''.join(
interfaces.get('test').msg_queue[fake_sender]))
ok_('testch1' not in ''.join(
interfaces.get('test').msg_queue[fake_sender]))
ok_('testch2' not in ''.join(
interfaces.get('test').msg_queue[fake_sender]))
# test list single user, channels
interfaces.get('test').reset()
for index in range(2):
self.service._list_users([index], fake_sender)
msg = ' '.join(interfaces.get('test').msg_queue[fake_sender])
if 'fake_sender' in msg:
ok_('testch1' in msg)
elif 'fake_sender2' in msg:
ok_('testch2' in msg)
ok_('testch1' in msg)
# test list multiple user channels
interfaces.get('test').reset()
self.service._list_users(range(2), fake_sender)
ok_('fake_sender' in ''.join(
interfaces.get('test').msg_queue[fake_sender]))
ok_('fake_sender2' in ''.join(
interfaces.get('test').msg_queue[fake_sender]))
ok_('testch1' in ''.join(
interfaces.get('test').msg_queue[fake_sender]))
ok_('testch2' in ''.join(
interfaces.get('test').msg_queue[fake_sender]))
def setUp(self):
self.service = Service()
self.service._twitch = MockTwitchEngine()
self.parser = LinotParser(usage=argparse.SUPPRESS, add_help=False)
interfaces.get('test').reset()
def setUp(self):
self.service = Service()
self.service._twitch = MockTwitchEngine()
self.parser = LinotParser(usage=argparse.SUPPRESS, add_help=False)
interfaces.get('test').reset()
def setUp(self):
self.sender = CommandSubmitter('test', 'sender')
interfaces.get('test').reset()
self.interface = interfaces.get('test')
def test_import(self):
sender = CommandSubmitter('test', 'tester')
short_test_channels = ['testch' + str(x) for x in range(3)]
long_test_channels = ['testch' + str(x) for x in range(10)]
self.service.setup(self.parser)
self.service.start()
threading.Event().wait(1)
self.service.stop()
# test small amount of followed channel
self.service._twitch.followed_ch_list = {
'some_twitch_user': short_test_channels
}
self.service._import(['some_twitch_user'], sender)
self.service._list_channel(True, sender)
for ch in short_test_channels:
ok_(ch.lower() in ' '.join(
interfaces.get('test').msg_queue[sender]).lower())
interfaces.get('test').reset()
self.service._unsub_all(True, sender)
# test large amount of followed channel
self.service._twitch.followed_ch_list = {
'some_twitch_user': long_test_channels
}
self.service._import(['some_twitch_user'], sender)
self.service._list_channel(True, sender)
for ch in long_test_channels:
ok_(ch.lower() in ' '.join(
interfaces.get('test').msg_queue[sender]).lower())
ok_('a while' in ' '.join(interfaces.get(
'test').msg_queue[sender]).lower())
interfaces.get('test').reset()
self.service._unsub_all(True, sender)
# test not found
self.service._twitch.followed_ch_list = {
'some_twitch_user': long_test_channels
}
self.service._import(['some_twitch_user2'], sender)
self.service._list_channel(True, sender)
for ch in long_test_channels:
ok_(ch.lower() not in ' '.join(
interfaces.get('test').msg_queue[sender]).lower())
ok_('not found' in ' '.join(interfaces.get(
'test').msg_queue[sender]).lower())
interfaces.get('test').reset()
self.service._unsub_all(True, sender)
