1 /* 2 * Copyright (c) 2007 Mockito contributors 3 * This program is made available under the terms of the MIT License. 4 */ 5 package org.mockitousage.basicapi; 6 7 import static org.junit.Assert.assertEquals; 8 import static org.mockito.Mockito.mock; 9 import static org.mockito.Mockito.when; 10 11 import java.util.Set; 12 13 import org.junit.Test; 14 15 public class MockAccessTest { 16 17 @Test shouldAllowStubbedMockReferenceAccess()18 public void shouldAllowStubbedMockReferenceAccess() throws Exception { 19 Set<?> expectedMock = mock(Set.class); 20 21 Set<?> returnedMock = when(expectedMock.isEmpty()).thenReturn(false).getMock(); 22 23 assertEquals(expectedMock, returnedMock); 24 } 25 26 @Test stubbedMockShouldWorkAsUsual()27 public void stubbedMockShouldWorkAsUsual() throws Exception { 28 Set<?> returnedMock = when(mock(Set.class).isEmpty()).thenReturn(false, true).getMock(); 29 30 assertEquals(false, returnedMock.isEmpty()); 31 assertEquals(true, returnedMock.isEmpty()); 32 } 33 } 34