def testCmdProcess_Unknown(self): mock = mock_engine() lap = LinotArgParser('testcmd', self.parser, mock.cmdProcess, None) lap.add_argument('-a', action='store_true') self.cmd_server = LineCmdServer(mock, self.parser) def cmdChecker(args, sender, done): ok_(sender) ok_(args.a) cmdChecker.testpass = True def msgChecker(recv, msg, done): ok_(recv == 'sender') ok_('Unknown' in msg) msgChecker.passtime += 1 if msgChecker.passtime == 2: done() cmdChecker.testpass = False msgChecker.passtime = 0 fake_cmd = [ ['sender', None, fakeMessage('some_unknown_cmds')], [True, None, fakeMessage('testcmd -a')], ['sender', None, fakeMessage('some_unknown_cmds')], ] mock.addTest(fake_cmd, cmdChecker, msgChecker) self.cmd_server.start() mock.test_finished.wait(10) ok_(cmdChecker.testpass) self.cmd_server.stop() self.cmd_server.join(10) ok_(self.cmd_server.stopped()) ok_(not self.cmd_server.isAlive())
def test_direct_command(self): mock = mock_engine() lap = LinotArgParser('testcmd', self.parser, mock.cmdProcess, None) 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 cmd_checker.runned = False lap.add_direct_command(cmd_checker, 'twitch\.tv/(\w+)[\s\t,]*', re.IGNORECASE) self.cmd_server = LineCmdServer(mock, self.parser) fake_cmd = [ ['sender', None, fakeMessage('www.twitch.tv/somechannel')], ] mock.addTest(fake_cmd, None, None) self.cmd_server.start() mock.test_finished.wait(10) ok_(cmd_checker.runned) self.cmd_server.stop()
def testCmdProcess_Normal(self): mock = mock_engine() lap = LinotArgParser('testcmd', self.parser, mock.cmdProcess, None) lap.add_argument('-a', action='store_true') lap.add_argument('-b', action='store_true') self.cmd_server = LineCmdServer(mock, self.parser) # Test 1 cmd in op_list fake_cmd = [ [True, None, fakeMessage('testcmd -a')] ] def cmdChecker_1(args, sender, done): ok_(args.a) ok_(sender) done() mock.addTest(fake_cmd, cmdChecker_1, None) # Test 2 cmds in op_list fake_cmd = [ [True, None, fakeMessage('testcmd -a')], [True, None, fakeMessage('testcmd -b')], ] def cmdChecker_2(args, sender, done): ok_(args.a or args.b) if args.a: cmdChecker_2.var += 'a' if args.b: cmdChecker_2.var += 'b' if cmdChecker_2.var == 'ab': done() elif len(cmdChecker_2.var) >= 2: ok_(False) cmdChecker_2.var = '' mock.addTest(fake_cmd, cmdChecker_2, None) self.cmd_server.start() mock.test_finished.wait(10) self.cmd_server.stop() self.cmd_server.join(10) ok_(not self.cmd_server.isAlive())
class TestLineCmdServer: def setUp(self): parser = LinotParser(usage=argparse.SUPPRESS, add_help=False) sub_cmd_parser = parser.get_sub_parser() self.sub_parser = sub_cmd_parser self.parser = parser def tearDown(self): self.cmd_server.stop() self.cmd_server = None @timed(10) def testCmdProcess_Normal(self): mock = mock_engine() lap = LinotArgParser('testcmd', self.parser, mock.cmdProcess, None) lap.add_argument('-a', action='store_true') lap.add_argument('-b', action='store_true') self.cmd_server = LineCmdServer(mock, self.parser) # Test 1 cmd in op_list fake_cmd = [ [True, None, fakeMessage('testcmd -a')] ] def cmdChecker_1(args, sender, done): ok_(args.a) ok_(sender) done() mock.addTest(fake_cmd, cmdChecker_1, None) # Test 2 cmds in op_list fake_cmd = [ [True, None, fakeMessage('testcmd -a')], [True, None, fakeMessage('testcmd -b')], ] def cmdChecker_2(args, sender, done): ok_(args.a or args.b) if args.a: cmdChecker_2.var += 'a' if args.b: cmdChecker_2.var += 'b' if cmdChecker_2.var == 'ab': done() elif len(cmdChecker_2.var) >= 2: ok_(False) cmdChecker_2.var = '' mock.addTest(fake_cmd, cmdChecker_2, None) self.cmd_server.start() mock.test_finished.wait(10) self.cmd_server.stop() self.cmd_server.join(10) ok_(not self.cmd_server.isAlive()) @timed(10) def testCmdProcess_Unknown(self): mock = mock_engine() lap = LinotArgParser('testcmd', self.parser, mock.cmdProcess, None) lap.add_argument('-a', action='store_true') self.cmd_server = LineCmdServer(mock, self.parser) def cmdChecker(args, sender, done): ok_(sender) ok_(args.a) cmdChecker.testpass = True def msgChecker(recv, msg, done): ok_(recv == 'sender') ok_('Unknown' in msg) msgChecker.passtime += 1 if msgChecker.passtime == 2: done() cmdChecker.testpass = False msgChecker.passtime = 0 fake_cmd = [ ['sender', None, fakeMessage('some_unknown_cmds')], [True, None, fakeMessage('testcmd -a')], ['sender', None, fakeMessage('some_unknown_cmds')], ] mock.addTest(fake_cmd, cmdChecker, msgChecker) self.cmd_server.start() mock.test_finished.wait(10) ok_(cmdChecker.testpass) self.cmd_server.stop() self.cmd_server.join(10) ok_(self.cmd_server.stopped()) ok_(not self.cmd_server.isAlive()) def test_direct_command(self): mock = mock_engine() lap = LinotArgParser('testcmd', self.parser, mock.cmdProcess, None) 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 cmd_checker.runned = False lap.add_direct_command(cmd_checker, 'twitch\.tv/(\w+)[\s\t,]*', re.IGNORECASE) self.cmd_server = LineCmdServer(mock, self.parser) fake_cmd = [ ['sender', None, fakeMessage('www.twitch.tv/somechannel')], ] mock.addTest(fake_cmd, None, None) self.cmd_server.start() mock.test_finished.wait(10) ok_(cmd_checker.runned) self.cmd_server.stop()