Exemplo n.º 1
0
    def setUp(self):
        patcher1 = patch('charlesbot.robot.Robot.initialize_robot')
        self.addCleanup(patcher1.stop)
        self.mock_initialize_robot = patcher1.start()

        from charlesbot.robot import Robot
        self.robot = Robot()
Exemplo n.º 2
0
def main(args=None):
    logging.basicConfig(
        stream=sys.stdout,
        level=logging.INFO,
        format='%(asctime)s: %(levelname)s [%(name)s:%(lineno)d] %(message)s')
    log = logging.getLogger(__name__)
    log.info("Starting CharlesBOT now!")

    slackbot = Robot()
    slackbot.start()
Exemplo n.º 3
0
def main(args=None):
    logging.basicConfig(
        stream=sys.stdout,
        level=logging.INFO,
        format='%(asctime)s: %(levelname)s [%(name)s:%(lineno)d] %(message)s'
    )
    log = logging.getLogger(__name__)
    log.info("Starting CharlesBOT now!")

    slackbot = Robot()
    slackbot.start()
Exemplo n.º 4
0
class TestRobot(asynctest.TestCase):

    def setUp(self):
        patcher1 = patch('charlesbot.robot.Robot.initialize_robot')
        self.addCleanup(patcher1.stop)
        self.mock_initialize_robot = patcher1.start()

        from charlesbot.robot import Robot
        self.robot = Robot()

    @asynctest.ignore_loop
    def test_exit_cleanly_no_plugins(self):
        self.robot.plugin_list = []
        self.robot.exit_cleanly()
        self.assertEqual(self.robot.is_running(), False)

    @asynctest.ignore_loop
    def test_exit_cleanly_multiple_plugins(self):
        plugin1 = MagicMock()
        plugin2 = MagicMock()
        self.robot.plugin_list = [plugin1, plugin2]
        self.robot.exit_cleanly()
        self.assertEqual(self.robot.is_running(), False)
        self.assertFalse(plugin1.is_running)
        self.assertFalse(plugin2.is_running)

    def test_queue_message_empty_msg_1(self):
        plugin1 = MagicMock()
        msg = ""
        yield from self.robot.queue_message(msg, plugin1)
        self.assertEqual(plugin1.method_calls, [])

    def test_queue_message_empty_msg_2(self):
        plugin1 = MagicMock()
        msg = []
        yield from self.robot.queue_message(msg, plugin1)
        self.assertEqual(plugin1.method_calls, [])

    def test_queue_message_actual_msg_1(self):
        plugin1 = MagicMock()
        msg = "HelloPlugin"
        yield from self.robot.queue_message(msg, plugin1)
        expected = call.queue_message('HelloPlugin')
        self.assertEqual(plugin1.method_calls, [expected])

    def test_queue_message_actual_msg_2(self):
        plugin1 = MagicMock()
        msg = ["HelloPlugin"]
        yield from self.robot.queue_message(msg, plugin1)
        expected = call.queue_message(['HelloPlugin'])
        self.assertEqual(plugin1.method_calls, [expected])
    def setUp(self):
        patcher1 = patch('charlesbot.robot.Robot.initialize_robot')
        self.addCleanup(patcher1.stop)
        self.mock_initialize_robot = patcher1.start()

        from charlesbot.robot import Robot
        self.robot = Robot()
class TestRobotInitializePlugin(unittest.TestCase):

    def setUp(self):
        patcher1 = patch('charlesbot.robot.Robot.initialize_robot')
        self.addCleanup(patcher1.stop)
        self.mock_initialize_robot = patcher1.start()

        from charlesbot.robot import Robot
        self.robot = Robot()

    def test_empty_enabled_plugins_list(self):
        self.robot.enabled_plugins = []
        enabled_list = self.robot.initialize_plugins()
        self.assertEqual(enabled_list, [])

    def test_dummy_class_loader(self):
        self.robot.enabled_plugins = ['charlesbot.robot.Robot']
        enabled_list = self.robot.initialize_plugins()
        self.assertEqual(len(enabled_list), 1)
Exemplo n.º 7
0
    def setUp(self):
        patcher1 = patch('charlesbot.robot.Robot.initialize_robot')
        self.addCleanup(patcher1.stop)
        self.mock_initialize_robot = patcher1.start()

        patcher2 = patch('charlesbot.robot.Robot.is_running')
        self.addCleanup(patcher2.stop)
        self.mock_is_running = patcher2.start()

        self.mock_slack = MagicMock()
        self.mock_slack.get_stream_messages = CoroutineMock()

        patcher3 = patch('charlesbot.robot.Robot.queue_message')
        self.addCleanup(patcher3.stop)
        self.mock_queue_message = patcher3.start()

        from charlesbot.robot import Robot
        self.robot = Robot()
        self.robot.slack = self.mock_slack
        self.mock_is_running.side_effect = [True, False]  # single iteration
class TestRobotHelpMessages(asynctest.TestCase):

    def setUp(self):
        patcher1 = patch('charlesbot.robot.Robot.initialize_robot')
        self.addCleanup(patcher1.stop)
        self.mock_initialize_robot = patcher1.start()

        from charlesbot.robot import Robot
        self.robot = Robot()

    @asynctest.ignore_loop
    def test_empty_plugin_list(self):
        self.robot.plugin_list = []
        self.robot.initialize_static_plugins()
        self.assertEqual(len(self.robot.plugin_list), 2)

    @asynctest.ignore_loop
    def test_plugin_list_one_entry(self):
        plugin1 = MagicMock()
        self.robot.plugin_list = [plugin1]
        self.robot.initialize_static_plugins()
        self.assertEqual(len(self.robot.plugin_list), 3)

    @asynctest.ignore_loop
    def test_plugin_list_two_entries(self):
        plugin1 = MagicMock()
        plugin2 = MagicMock()
        self.robot.plugin_list = [plugin1, plugin2]
        self.robot.initialize_static_plugins()
        self.assertEqual(len(self.robot.plugin_list), 4)
Exemplo n.º 9
0
class TestRobotHelpMessages(asynctest.TestCase):
    def setUp(self):
        patcher1 = patch('charlesbot.robot.Robot.initialize_robot')
        self.addCleanup(patcher1.stop)
        self.mock_initialize_robot = patcher1.start()

        from charlesbot.robot import Robot
        self.robot = Robot()

    @asynctest.ignore_loop
    def test_empty_plugin_list(self):
        self.robot.plugin_list = []
        self.robot.initialize_static_plugins()
        self.assertEqual(len(self.robot.plugin_list), 2)

    @asynctest.ignore_loop
    def test_plugin_list_one_entry(self):
        plugin1 = MagicMock()
        self.robot.plugin_list = [plugin1]
        self.robot.initialize_static_plugins()
        self.assertEqual(len(self.robot.plugin_list), 3)

    @asynctest.ignore_loop
    def test_plugin_list_two_entries(self):
        plugin1 = MagicMock()
        plugin2 = MagicMock()
        self.robot.plugin_list = [plugin1, plugin2]
        self.robot.initialize_static_plugins()
        self.assertEqual(len(self.robot.plugin_list), 4)
Exemplo n.º 10
0
class TestRobotRouteMessageToPlugin(asynctest.TestCase):

    def setUp(self):
        patcher1 = patch('charlesbot.robot.Robot.initialize_robot')
        self.addCleanup(patcher1.stop)
        self.mock_initialize_robot = patcher1.start()

        patcher2 = patch('charlesbot.robot.Robot.is_running')
        self.addCleanup(patcher2.stop)
        self.mock_is_running = patcher2.start()

        self.mock_slack = MagicMock()
        self.mock_slack.get_stream_messages = CoroutineMock()

        patcher3 = patch('charlesbot.robot.Robot.queue_message')
        self.addCleanup(patcher3.stop)
        self.mock_queue_message = patcher3.start()

        from charlesbot.robot import Robot
        self.robot = Robot()
        self.robot.slack = self.mock_slack
        self.mock_is_running.side_effect = [True, False]  # single iteration

    def test_no_msgs_returned_from_stream(self):
        self.mock_slack.get_stream_messages.return_value = []
        yield from self.robot.produce()
        self.assertEqual(self.mock_queue_message.mock_calls, [])

    def test_multiple_produce_iterations(self):
        self.mock_is_running.side_effect = [True, True, False]
        self.mock_slack.get_stream_messages.return_value = []
        yield from self.robot.produce()
        self.assertEqual(self.mock_queue_message.mock_calls, [])

    def test_single_msg_empty_plugin_list(self):
        self.mock_slack.get_stream_messages.return_value = ['msg1']
        self.robot.plugin_list = []
        yield from self.robot.produce()
        self.assertEqual(self.mock_queue_message.mock_calls, [])

    def test_single_msg_one_plugin(self):
        self.mock_slack.get_stream_messages.return_value = ['msg1']
        self.robot.plugin_list = ['plug1']
        yield from self.robot.produce()
        expected_call_1 = call('msg1', 'plug1')
        self.assertTrue(len(self.mock_queue_message.mock_calls), 1)
        self.assertTrue(expected_call_1 in self.mock_queue_message.mock_calls)

    def test_single_msg_multiple_plugins(self):
        self.mock_slack.get_stream_messages.return_value = ['msg1']
        self.robot.plugin_list = ['plug1', 'plug2']
        yield from self.robot.produce()
        expected_call_1 = call('msg1', 'plug1')
        expected_call_2 = call('msg1', 'plug2')
        self.assertTrue(len(self.mock_queue_message.mock_calls), 2)
        self.assertTrue(expected_call_1 in self.mock_queue_message.mock_calls)
        self.assertTrue(expected_call_2 in self.mock_queue_message.mock_calls)

    def test_multiple_msgs_empty_plugin_list(self):
        self.mock_slack.get_stream_messages.return_value = ['msg1', 'msg2']
        self.robot.plugin_list = []
        yield from self.robot.produce()
        self.assertEqual(self.mock_queue_message.mock_calls, [])

    def test_multiple_msgs_one_plugin(self):
        self.mock_slack.get_stream_messages.return_value = ['msg1', 'msg2']
        self.robot.plugin_list = ['plug1']
        yield from self.robot.produce()
        expected_call_1 = call('msg1', 'plug1')
        expected_call_2 = call('msg2', 'plug1')
        self.assertTrue(len(self.mock_queue_message.mock_calls), 2)
        self.assertTrue(expected_call_1 in self.mock_queue_message.mock_calls)
        self.assertTrue(expected_call_2 in self.mock_queue_message.mock_calls)

    def test_multiple_msgs_multiple_plugins(self):
        self.mock_slack.get_stream_messages.return_value = ['msg1', 'msg2']
        self.robot.plugin_list = ['plug1', 'plug2']
        yield from self.robot.produce()
        expected_call_1 = call('msg1', 'plug1')
        expected_call_2 = call('msg2', 'plug1')
        expected_call_3 = call('msg1', 'plug2')
        expected_call_4 = call('msg2', 'plug2')
        self.assertTrue(len(self.mock_queue_message.mock_calls), 4)
        self.assertTrue(expected_call_1 in self.mock_queue_message.mock_calls)
        self.assertTrue(expected_call_2 in self.mock_queue_message.mock_calls)
        self.assertTrue(expected_call_3 in self.mock_queue_message.mock_calls)
        self.assertTrue(expected_call_4 in self.mock_queue_message.mock_calls)