（八）同一方法调用返回时序结果--实战篇

浅浅的花香味﹌ 2022-02-27 01:50 217阅读 0赞

有时候，我们需要对同一方法调用时，返回结果为一个时序结果：即同一个方法的调用，返回值不同。

比如，应用程序中有通过tcp/http连结外部网络操作，写单元测试时，不希望应用程序真正连结外部网络（因为可能会连结失败），因此需要对外部网络的tcp/http数据流返回数据进行mock。

下面直接用代码，来描述如何mock.

**1. Mock：一个类的所有实例**

// 一个类所有实例的某个方法，返回时序结果。  
// 适用场景：每次调用，期望返回的数据不一样。比如从tcp数据流中拿数据

//同一方法返回时序结果
    public class ReturnSequenceResultInOneMethodTest {
        // 一个类所有实例的某个方法，返回时序结果。
        // 适用场景：每次调用，期望返回的数据不一样。比如从tcp数据流中拿数据
        @Test
        public void testIfMethodOfClass() {
            AnOrdinaryClass instance = new AnOrdinaryClass();
            new Expectations(AnOrdinaryClass.class) {
                {
                    instance.ordinaryMethod();
                    // 对类AnOrdinaryClass所有实例调用ordinaryMethod方法时，依次返回1，2，3，4，5
                    result = new int[] { 1, 2, 3, 4, 5 };
                }
            };
            AnOrdinaryClass instance1 = new AnOrdinaryClass();
            Assert.assertTrue(instance1.ordinaryMethod() == 1);
            Assert.assertTrue(instance1.ordinaryMethod() == 2);
            Assert.assertTrue(instance1.ordinaryMethod() == 3);
            Assert.assertTrue(instance1.ordinaryMethod() == 4);
            Assert.assertTrue(instance1.ordinaryMethod() == 5);
            // 因为在上面录制脚本中，只录制了5个结果，当大于5时，就以最后一次结果为准
            Assert.assertTrue(instance1.ordinaryMethod() == 5);
            Assert.assertTrue(instance1.ordinaryMethod() == 5);
        }

**2. Mock：仅仅是对某一个实例的返回值进行录制**

// 与上述不一样的地方，仅仅是对某一个实例的返回值进行录制
        @Test
        public void testIfMethodOfIntance() {
            AnOrdinaryClass instance = new AnOrdinaryClass();
            new Expectations(instance) {
                {
                    instance.ordinaryMethod();
                    // 对实例instance调用ordinaryMethod方法时，依次返回1，2，3，4，5
                    result = new int[] { 1, 2, 3, 4, 5 };
                }
            };
            // 只影响了instance这个实例
            Assert.assertTrue(instance.ordinaryMethod() == 1);
            Assert.assertTrue(instance.ordinaryMethod() == 2);
            Assert.assertTrue(instance.ordinaryMethod() == 3);
            Assert.assertTrue(instance.ordinaryMethod() == 4);
            Assert.assertTrue(instance.ordinaryMethod() == 5);
            // 因为在上面录制脚本中，只录制了5个结果，当大于5时，就以最后一次结果为准
            Assert.assertTrue(instance.ordinaryMethod() == 5);
            Assert.assertTrue(instance.ordinaryMethod() == 5);
     
            // 类AnOrdinaryClass的其它实例并不会受到影响
            AnOrdinaryClass instance1 = new AnOrdinaryClass();
            // ordinaryMethod这个方法本来就返回2
            Assert.assertTrue(instance1.ordinaryMethod() == 2);
            Assert.assertTrue(instance1.ordinaryMethod() == 2);
        }

1.  **Mock： tcp数据流返回数据**

// 下面以对tcp数据流返回数据为例子，进行mock
        @Test
        public void testInputStreamSequence() {
            try {
                URL url = new URL("http://jmockit.cn");              // 依据地址创建URL
                URLConnection connection = url.openConnection();    // 获得urlConnecion
                connection.connect();                              // 打开连接
                InputStream in = connection.getInputStream();
                //现在对jmockit.cn服务器返回的数据进行mock
                new Expectations(in) {
                    {
                        in.read();
                        // -1表示流数据结束了
                        result = new int[] { 1, 2, 3, 4, 5, -1 };
                    }
                };
                // 读取jmockit.cn服务器返回的内容，如果没有上面的mock,返回将是实际的内容
                Assert.assertTrue(in.read() == 1);
                Assert.assertTrue(in.read() == 2);
                Assert.assertTrue(in.read() == 3);
                Assert.assertTrue(in.read() == 4);
                Assert.assertTrue(in.read() == 5);
                Assert.assertTrue(in.read() == -1);
                in.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        }