class CardOptionsTests(unittest.TestCase): def setUp(self): card = TrelloElementMock("card_name") self.card_options = CardOptions(card) self.card_options.command = CommandMock() def test_items_returns_the_card_operations(self): options = ["..", "Open in Browser", "Show", "Comments", "Comment", "Add label", "Remove label", "Move to another List", "Archive", "Exit"] self.assertEqual(options, self.card_options.items()) def test_callback_calls_the_method_at_the_given_index(self): action = self.mock_action_at(1) self.card_options.callback(1) action.assert_called_with() def mock_action_at(self, index): option = self.card_options.options[index] option["action"] = MagicMock() return option["action"]
def setUp(self): card = TrelloElementMock("card_name") self.card_options = CardOptions(card) self.card_options.command = CommandMock()