def test_list_like_behaviour():
menu = teek.Menu()
menu.append(teek.MenuItem("Click me", print))
assert len(menu) == 1
assert menu[0].config['label'] == "Click me"
menu[-1] = teek.MenuItem("No, click me instead", print)
assert len(menu) == 1
assert menu[0].config['label'] == "No, click me instead"
menu.insert(123, teek.MenuItem("Last item", print))
assert menu[-1].config['label'] == 'Last item'
def test_indexes_dont_mess_up_ever_like_srsly_not_ever_never():
menu = teek.Menu()
def check():
assert [item._index for item in menu] == list(range(len(menu)))
menu.append(teek.MenuItem())
check()
menu.append(teek.MenuItem())
check()
menu.extend(
[teek.MenuItem(),
teek.MenuItem(),
teek.MenuItem(),
teek.MenuItem()])
check()
menu.pop()
check()
menu.pop(-2)
check()
del menu[0]
check()
menu.insert(1, teek.MenuItem())
check()
menu[1] = teek.MenuItem()
check()
menu.clear()
check()
def test_readding_to_menus():
menu1 = teek.Menu()
menu2 = teek.Menu()
def lengths():
return (len(menu1), len(menu2))
item = teek.MenuItem()
assert lengths() == (0, 0)
menu1.append(item)
assert lengths() == (1, 0)
menu1.remove(item)
assert lengths() == (0, 0)
menu1.append(item)
assert lengths() == (1, 0)
menu1.remove(item)
assert lengths() == (0, 0)
menu2.append(item)
assert lengths() == (0, 1)
menu2.remove(item)
assert lengths() == (0, 0)
menu1.append(item)
assert lengths() == (1, 0)
menu1.remove(item)
assert lengths() == (0, 0)
def test_config(check_config_types):
boolean_var = teek.BooleanVar()
string_var = teek.StringVar()
menu = teek.Menu([
teek.MenuItem("Click me", print),
teek.MenuItem("Check me", boolean_var),
teek.MenuItem("Choice A", string_var, "a"),
teek.MenuItem("Choice B", string_var, "b"),
teek.MenuItem("Choice C", string_var, "c"),
teek.MenuItem(),
teek.MenuItem("Submenu", []),
])
for item in menu:
check_config_types(item.config, 'menu %s item' % item.type)
assert isinstance(menu[0].config['command'], teek.Callback)
menu[0].config['command'].disconnect(print) # fails if not connected
assert isinstance(menu[1].config['variable'], teek.BooleanVar)
assert menu[1].config['variable'] == boolean_var
for index in (2, 3, 4):
assert isinstance(menu[index].config['variable'], teek.StringVar)
assert menu[index].config['variable'] == string_var
def test_popup():
menu = teek.Menu()
assert not menu.winfo_ismapped()
menu.popup(123, 456)
assert menu.winfo_ismapped()
menu2 = teek.Menu([teek.MenuItem("Click me", print)])
menu2.popup(123, 456, menu2[0])
assert menu2.winfo_ismapped()
def test_added_to_2_menus_at_the_same_time():
item = teek.MenuItem("Click me", print)
menu = teek.Menu()
menu.append(item)
check = pytest.raises(
RuntimeError,
match=(r'^cannot add a MenuItem to two different menus '
r'or twice to the same menu$'))
with check:
menu.append(item)
with check:
teek.Menu().append(item)
def test_not_added_to_menu_yet():
menu = teek.Menu()
item = teek.MenuItem("Click me", print)
check = pytest.raises(RuntimeError,
match=r".*hasn't been added to a Menu.*")
with check:
item.config['label']
menu.append(item)
assert item.config['label'] == "Click me"
assert menu.pop() is item
with check:
item.config['label']
def test_menuitem_objects():
item = teek.MenuItem("Click me",
print,
activeforeground=teek.Color(1, 2, 3))
assert repr(item) == ("<MenuItem('Click me', <built-in function print>): "
"type='command', not added to a menu yet>")
menu = teek.Menu([item])
assert repr(item) == ("<MenuItem('Click me', <built-in function print>): "
"type='command', added to a menu>")
# the menu should cache the item
assert menu[0] is item
assert menu[0] is menu[0]
def test_destroying_deletes_commands(handy_callback):
@handy_callback
def callback_func():
pass
menu = teek.Menu([
teek.MenuItem("Click me", callback_func),
])
command_string = teek.tcl_call(str, menu, 'entrycget', 0, '-command')
teek.tcl_call(None, command_string)
assert callback_func.ran_once()
menu.destroy()
with pytest.raises(teek.TclError):
teek.tcl_call(None, command_string)
def test_item_types():
menu = teek.Menu([
teek.MenuItem("Click me", print),
teek.MenuItem(), # separator
teek.MenuItem("Check me", teek.BooleanVar()),
teek.MenuItem("Choose me or a friend of mine", teek.StringVar(), "me"),
teek.MenuItem("Submenu", [teek.MenuItem("Click here", print)]),
teek.MenuItem("Different kind of submenu",
teek.Menu([
teek.MenuItem("Click me", print),
])),
])
assert [item.type for item in menu] == [
'command', 'separator', 'checkbutton', 'radiobutton', 'cascade',
'cascade'
]
for submenu_item in (menu[-2], menu[-1]):
assert submenu_item.config['menu'][0].type == 'command'
with pytest.raises(TypeError, match=r'^expected .* arguments.*$'):
teek.MenuItem("adding only 1 argument is wrong")
def test_repr():
menu = teek.Menu([
teek.MenuItem("Click me", print),
teek.MenuItem("Check me", teek.BooleanVar()),
])
assert repr(menu) == '<teek.Menu widget: contains 2 items>'