如何模拟从特定方法返回的Map？

import org.junit.Test;
import org.mockito.Mockito;

import java.util.HashMap;
import java.util.Map;

import static org.junit.Assert.assertEquals;

public class MyClassTest {

    @Test
    public void testMethod() {
        // 创建模拟对象
        MyClass myClassMock = Mockito.mock(MyClass.class);

        // 创建一个Map对象作为返回值
        Map<String, Integer> expectedMap = new HashMap<>();
        expectedMap.put("key1", 1);
        expectedMap.put("key2", 2);

        // 模拟特定方法的返回值
        Mockito.when(myClassMock.method()).thenReturn(expectedMap);

        // 调用被测试的方法
        Map<String, Integer> actualMap = myClassMock.method();

        // 断言返回的Map对象与预期的Map对象相等
        assertEquals(expectedMap, actualMap);
    }
}