source: java2python/mockito-t/test/testcode/MockitoBasicTest.java

package testcode;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import org.junit.Test;
import org.junit.internal.runners.JUnit4ClassRunner;
import org.junit.runner.RunWith;
import org.mockito.AdditionalMatchers;

class TestClass {
        public int f(int x) {
                return x + 1;
        }
}

class NumberClass {
        public int f(Number x) {
                return x.intValue();
        }
}

/**
 * Test for mockito translator. This program is going to be translated and run
 * in python.
 */
@RunWith(JUnit4ClassRunner.class)
public class MockitoBasicTest {

        /**
         * NOTE NO CONSTRUCTOR. This is needed to check that the proper constructor,
         * matching in this case the implicit superclass TestCase from Python, is
         * created. This is a special case, in normal cases java already enforces
         * all required code including the call to super().
         */

        @Test
        public void testStub1() {
                TestClass test = mock(TestClass.class);
                when(test.f(3)).thenReturn(1);
                // the mock did not do any when.thenReturn and should fail
                assertEquals(1, test.f(3));

        }

        @Test
        public void testEqMatcher() {
                TestClass test = mock(TestClass.class);
                when(test.f(eq(3))).thenReturn(1);
                when(test.f(eq(4))).thenReturn(2);
                assertEquals(1, test.f(3));
                assertEquals(2, test.f(4));
        }

        @Test
        public void testAnyOfClass() {
                NumberClass test = mock(NumberClass.class);
                when(test.f(any(Double.class))).thenReturn(1);
                when(test.f(any(Integer.class))).thenReturn(2);
                assertEquals(1, test.f(3d));
                assertEquals(2, test.f(4));

        }

        @Test
        public void testNot() {
                TestClass test = mock(TestClass.class);
                when(test.f(AdditionalMatchers.not(eq(3)))).thenReturn(1);
                assertEquals(1, test.f(1));
                assertEquals(1, test.f(2));
        }

        @Test
        public void testNotEqual() {
                TestClass a = mock(TestClass.class);
                TestClass b = mock(TestClass.class);
                assertFalse(a.equals(b));
        }

}