def test_logout():
"""
Ensure the toolbar updates to the logged out state.
"""
w = Window()
w.tool_bar = mock.MagicMock()
w.logout()
w.tool_bar.set_logged_out.assert_called_once_with()
def test_set_logged_in_as():
"""
Given a username, the toolbar is appropriately called to update.
"""
w = Window()
w.tool_bar = mock.MagicMock()
w.set_logged_in_as('test')
w.tool_bar.set_logged_in_as.assert_called_once_with('test')