import unittest
from ttk import setup_master
from Dialog import Dialog

class TestDialog(unittest.TestCase):

    dlg_args = {'title': '', 'text': '', 'bitmap': '', 'strings': range(5)}

    def test_return(self):
        master = setup_master()
        def test(evt):
            for child in master.winfo_children():
                if isinstance(child, Dialog):
                    break
            else:
                self.fail("Didn't find Dialog")
            def end_dlg():
                child.event_generate('<Return>')
            child.after_idle(end_dlg)
        master.bind('<FocusOut>', test)

        for _ in range(2):
            for count in range(0, len(self.dlg_args['strings'])):
                self.dlg_args.update({'default': count})
                dlg = Dialog(master, self.dlg_args)
                self.assertFalse(isinstance(count, bool))
                self.assertEqual(dlg.num, count)

if __name__ == "__main__":
    from test.test_support import run_unittest
    run_unittest(TestDialog)