def test_ToolBar_set_logged_out():
"""
Ensure the UI reverts to the logged out state.
"""
tb = ToolBar(None)
tb.user_state = mock.MagicMock()
tb.login = mock.MagicMock()
tb.logout = mock.MagicMock()
tb.refresh = mock.MagicMock()
tb.set_logged_out()
tb.user_state.setText.assert_called_once_with('Signed out.')
tb.login.setVisible.assert_called_once_with(True)
tb.logout.setVisible.assert_called_once_with(False)
tb.refresh.setVisible.assert_called_once_with(False)
def test_ToolBar_set_logged_in_as():
"""Given a username, the user_state is updated and login/logout
buttons, and refresh buttons, are in the correct state.
"""
tb = ToolBar(None)
tb.user_state = mock.MagicMock()
tb.login = mock.MagicMock()
tb.logout = mock.MagicMock()
tb.refresh = mock.MagicMock()
tb.set_logged_in_as('test')
tb.user_state.setText.assert_called_once_with('Signed in as: test')
tb.login.setVisible.assert_called_once_with(False)
tb.logout.setVisible.assert_called_once_with(True)
tb.refresh.setVisible.assert_called_once_with(True)