testable-mock/docs/en-us/doc/use-mock.md
2021-01-03 22:54:38 +08:00

9.6 KiB

Fast Mocking

Compared with the class-granularity mocking practices of existing mock tools, TestableMock allows developers to directly define a single method and use it for mocking. With the principle of convention over configuration, mock method replacement will automatically happen when the specified method in the test class match an invocation in the class under test.

In summary, there are two simple rules:

  • Mock non-constructive method, copy the original method definition to the test class, add a parameter of the same type as the caller, and add a @MockMethod annotation
  • Mock construction method, copy the original method definition to the test class, replace the return value with the constructed type, the method name is arbitrary, and add a @MockContructor annotation

Mock convention:

  • The name of the test class should be <NameOfClassUnderTest> + Test (and in the same package path), which is usually the by-default naming convention of Java project managed by Maven or Gradle. This constraint may be relaxed or removed in future versions of TestableMock.
  • Do NOT access any non-static members in mock methods. Currently, methods that is decorated by @MockMethod or @MockContructor annotations will be automatically modified to static methods during runtime. In future versions, this constraint will be removed.

The detail mock method definition convention is as follows:

1. Mock method calls of any class

Define an ordinary method annotated with @MockMethod in the test class with exactly the same signature (name, parameter, and return value type) as the method to be mocked, and then add an extra parameter as the first parameter of method, with the same type as the object that the method originally belongs to.

At this time, all invocations to that original method in the class under test will be automatically replaced with invocations to the above-mentioned mock method when the unit test is running.

Note: When several methods to be mocked have the same name, you can put the name of the method to be mocked in the targetMethod parameter of @MockMethod annotation, so that the mock method itself can be named at will.

For example, there is a call to "anything".substring(1, 2) in the class under test, and we want to change it to a fixed string when running the test, we only need to define the following method in the test class:

// The original method signature is `String substring(int, int)`
// The object `"anything"` that invokes this method is of type `String`
// Adds a `String` type parameter to the first position the mock method parameter list (parameter name is arbitrary)
// This parameter can be used to get the value and context of the actual invoker at runtime
@MockMethod
private String substring(String self, int i, int j) {
    return "sub_string";
}

The following example shows the usage of the targetMethod parameter, and its effect is the same as the above example:

// Use `targetMethod` to specify the name of the method that needs to be mocked
// The method itself can now be named arbitrarily, but the method parameters still need to follow the same matching rules
@MockMethod(targetMethod = "substring")
private String use_any_mock_method_name(String self, int i, int j) {
    return "sub_string";
}

For complete code examples, see the should_able_to_mock_common_method() test cases in the java-demo and kotlin-demo sample projects. (Because Kotlin has made magical changes to the String type, the method under test in the Kotlin example adds a layer of encapsulation to the BlackBox class)

2. Mock the member method of the class under test itself

Sometimes, when testing certain methods, it is desirable to mock out some other member methods of the class under test itself.

The solution is the same as the previous case. The first parameter type of the mock method needs to be the same as that of the class under test.

For example, there is a private method with the signature String innerFunc(String) in the class under test. If we want to replace it during testing, we only need to define the following method in the test class:

// The type to test is `DemoMock`
// So when defining the mock method, add a parameter of type `DemoMock` to the first position of parameter list (the name is arbitrary)
@MockMethod
private String innerFunc(DemoMock self, String text) {
    return "mock_" + text;
}

For complete code examples, see the should_able_to_mock_member_method() test case in the java-demo and kotlin-demo sample projects.

3. Mock static methods of any class

Mock for static methods is the same as for any ordinary methods. But it should be noted that when the mock method of a static method is called, the actual value of the first parameter passed in is always null.

For example, if the static method secretBox() of the BlackBox type is invoked in the class under test, and the method signature is changed to BlackBox secretBox(), the mock method is as follows:

// The target static method is defined in the `BlackBox` type
// When defining the mock method, add a parameter of type `BlackBox` to the first position parameter list (the name is arbitrary)
// This parameter is only used to identify the target type, the actual incoming value will always be `null`
@MockMethod
private BlackBox secretBox(BlackBox ignore) {
    return new BlackBox("not_secret_box");
}

For complete code examples, see the should_able_to_mock_static_method() test case in the java-demo and kotlin-demo sample projects.

4. Mock new operation of any type

Define an ordinary method annotated with @MockContructor in the test class, make the return value type of the method the type of the object to be created, and the method parameters are exactly the same as the constructor parameters to be mocked, the method name is arbitrary.

At this time, all operations in the class under test that use new to create the specified class (and use the constructor that is consistent with the mock method parameters) will be replaced with calls to the custom method.

For example, if there is a call to new BlackBox("something") in the class under test, and you want to change it during unit testing (usually by replacing it with a mock object, or by replacing it with a temporary object created with test parameters), just define the following mock method:

// The signature of the constructor to be mocked is `BlackBox(String)`
// No need to add additional parameters to the mock method parameter list, and the name of the mock method is arbitrary 
@MockContructor
private BlackBox createBlackBox(String text) {
    return new BlackBox("mock_" + text);
}

You can still use the @MockMethod annotation, and configure the targetMethod parameter value to "<init>", and the rest is the same as above. The effect is the same as using the @MockContructor annotation

For complete code examples, see the should_able_to_mock_new_object() test case in the java-demo and kotlin-demo sample projects.

5. Identify the current test case and invoke source

In the mock method, you can use the TestableTool.SOURCE_METHOD variable to identify the method name of the class under test before entering the mock method; in addition, the TestableTool.MOCK_CONTEXT variable can inject additional context parameters into the mock method, to distinguish and process different calling scenarios.

For example, to verify the impact on the target method under test when the mock method returns different results in the test case:

@Test
public void testDemo() {
    MOCK_CONTEXT.set("case", "data-ready");
    assertEquals(true, demo());
    MOCK_CONTEXT.set("case", "has-error");
    assertEquals(false, demo());
    MOCK_CONTEXT.clear();
}

Take out the injected parameters in the mock method and return different results according to the situation:

@MockMethod
private Data mockDemo() {
    switch((String)MOCK_CONTEXT.get("case")) {
        case "data-ready":
            return new Data();
        case "has-error":
            throw new NetworkException();
        default:
            return null;
    }
}

Note that because TestableMock does not (and won't to) rely on any specific test framework, it cannot automatically identify the end position of a single test case, which makes the parameters set to the TestableTool.MOCK_CONTEXT variable may exist cross test cases in the same test class. It is recommended to always use MOCK_CONTEXT.clear() to clear the context immediately after use. You can also add this statement to the unified position where the test case ends of the specific unit test framework, such as the @AfterEach method of JUnit 5.

In the current version, the effect of this variable at runtime is similar to a normal Map type member object in the test class, but please try to use this variable instead of a custom object to pass additional mock parameters in order to get better compatibility in the upcomingv0.5 version.

The TestableTool.MOCK_CONTEXT variable is currently shared within the test class. When the unit test runs in parallel, it is recommended to select the parallel type as classes

For complete code examples, see the should_able_to_get_source_method_name() and should_able_to_get_test_case_name() test cases in the java-demo and kotlin-demo sample projects.

6. Verify the sequence and parameters of the mock method being invoked

In test cases, you can use the TestableTool.verify() method, and cooperate with with(), withInOrder(), without(), withTimes() and other methods to verify the mock call situation.

For details, please refer to the Check Mock Call document.