class TestCommandSubmitter:
def setUp(self):
self.sender = CommandSubmitter('test', 'sender')
interfaces.get('test').reset()
self.interface = interfaces.get('test')
def test_code(self):
ok_(self.sender.code == 'sender')
def test_send_message(self):
test_msg = 'test message'
self.sender.send_message(test_msg)
ok_(len(self.interface.msg_queue[self.sender.code]) == 1)
ok_(self.interface.msg_queue[self.sender.code][0] == test_msg,
self.interface.msg_queue)
def test_get_display_name(self):
disp_name = self.sender.get_display_name()
ok_(disp_name == self.interface.get_display_name(self.sender))
def test_unicode(self):
u_name = unicode(self.sender)
ok_(u_name == unicode(self.sender.get_display_name()))
def test_dict_hashable(self):
sender_2 = CommandSubmitter('test', 'sender2')
sender_3 = CommandSubmitter('some', 'sender')
sender_same = CommandSubmitter('test', 'sender')
some_dict = defaultdict(lambda: False)
some_dict[self.sender] = True
ok_(some_dict[self.sender] is True)
ok_(some_dict[sender_2] is False)
ok_(some_dict[sender_3] is False)
ok_(some_dict[sender_same] is True)
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(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_dict_hashable(self):
sender_2 = CommandSubmitter('test', 'sender2')
sender_3 = CommandSubmitter('some', 'sender')
sender_same = CommandSubmitter('test', 'sender')
some_dict = defaultdict(lambda: False)
some_dict[self.sender] = True
ok_(some_dict[self.sender] is True)
ok_(some_dict[sender_2] is False)
ok_(some_dict[sender_3] is False)
ok_(some_dict[sender_same] is True)
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_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_sub_by_url(self):
sender = CommandSubmitter('test', 'sender')
self.service._twitch.exists_ch_list = ['KayDaDa', 'LinotServant']
self.service.setup(self.parser)
self.service.start()
threading.Event().wait(1)
self.service.stop()
self.service._sub_by_url(['KayDaDa', 'LinotServant'], 'dummy', sender)
ok_('KayDaDa'.lower() in self.service._sublist[sender],
'sublist = ' + str(self.service._sublist[sender]))
ok_('LinotServant'.lower() in self.service._sublist[sender],
'sublist = ' + str(self.service._sublist[sender]))
self.service._unsubscribe(['KayDaDa', 'LinotServant'], sender)
ok_('KayDaDa'.lower() not in self.service._sublist[sender],
'sublist = ' + str(self.service._sublist[sender]))
ok_('LinotServant'.lower() not in self.service._sublist[sender],
'sublist = ' + str(self.service._sublist[sender]))
# Integration test
self.parser.process_direct_commands(
'www.twitch.tv/KayDaDa twitch.tv/LinotServant', sender)
ok_('KayDaDa'.lower() in self.service._sublist[sender],
'sublist = ' + str(self.service._sublist[sender]))
ok_('LinotServant'.lower() in self.service._sublist[sender],
'sublist = ' + str(self.service._sublist[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_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_unsubscribe_one(self):
self.test_subscribe_one()
fake_sender = CommandSubmitter('test', 'fake_sender')
fake_sender2 = CommandSubmitter('test', 'fake_sender2')
self.service._subscribe(['testch2'], fake_sender)
self.service._unsubscribe(['testch1'], fake_sender)
ok_('testch1' not 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)
self.service._unsubscribe(['testch1'], fake_sender2)
ok_('testch1' not in self.service._sublist[fake_sender2],
'sublist = ' + str(self.service._sublist[fake_sender2]))
ok_(self.service._channel_sub_count['testch1'] == 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_refresh(self):
# check admin only
self.service.start()
threading.Event().wait(.1)
self.service.stop()
def self_ret():
self_ret.val = True
self_ret.val = False
self.service._check_thread.refresh = self_ret
config['interface']['test'] = {'admin_id': 'test_admin'}
fake_sender = CommandSubmitter('test', 'fake_sender')
self.service._refresh(True, fake_sender)
ok_(self_ret.val is not True)
self.service._refresh(True, CommandSubmitter('test', 'test_admin'))
ok_(self_ret.val is True)
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_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_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 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_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_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 polling_command(self):
# hide longPoll debug msg
org_stdout = sys.stdout
sys.stdout = io.BytesIO()
self._client.lock.acquire(True)
ge = self._client.longPoll()
op_list = []
for op in ge:
op_list.append(op)
self._client.lock.release()
sys.stdout = org_stdout
# construct formal command list
command_list = []
for op in op_list:
submitter = CommandSubmitter(self.NAME, op[0].id)
command_list.append((submitter, op[2].text))
return command_list
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_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_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 setUp(self):
self.sender = CommandSubmitter('test', 'sender')
interfaces.get('test').reset()
self.interface = interfaces.get('test')