#!/usr/bin/python3

from newfol.exception import ImplementationError
import newfol.main

import unittest


class TestCallbacks(unittest.TestCase):
    def setUp(self):
        self.categories = ["select", "simple-select", "change", "cancel"]

    def test_callback_type(self):
        items = [
            ["select", ["select", "simple-select"]],
            ["next", ["select", "simple-select"]],
            ["next-secondary", ["select"]],
            ["next-tertiary", ["select"]],
            ["change", ["change"]],
            ["cancel", ["cancel"]],
        ]
        for item in items:
            for cat in self.categories:
                result = newfol.main.View.is_callback_type(item[0], cat)
                self.assertEqual(result, cat in item[1])

    def test_unknown_callback_type(self):
        with self.assertRaises(ImplementationError):
            newfol.main.View.is_callback_type("select", "unknown")

if __name__ == '__main__':
    unittest.main()