@Test public void can_get_buttons() { Prompt prompt = new Prompt(evaluator, id); List<Button> buttons = new ArrayList<Button>(); buttons.add(null); buttons.add(null); when(evaluator.buttons(prompt)).thenReturn(buttons); assertThat(prompt.buttons().size(), is(2)); }