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