用 spock 模拟链式方法 return

Question

我一直坚持这一点。是否可以模拟 new URL(url).openStream() 到 return a file.gz？我正在使用 spock 来尝试执行此操作。

public class DownloadFile {
public  BufferedReader downloadGzipCsvFile(String url) throws MalformedURLException {
    BufferedReader br = null;
    if (UrlValidator.getInstance().isValid(url)){
        try {
            br = new BufferedReader(new InputStreamReader(new GZIPInputStream(new URL(url).openStream())));
        } catch (IOException e) {
            e.printStackTrace();
            System.exit(0);
        }
    } else{
        throw new MalformedURLException("Supplied URL: " + url + " is an invalid URL");
    }
    return br;
}
}

测试class

class DownloadFileSpec extends Specification{

def "Should return buffered reader for url for gzip csv file"(){
    given:
    String url = "http://www.test.com"
    DownloadFile downloadFile = new DownloadFile()

    when:
    BufferedReader br = downloadFile.downloadGzipCsvFile(url)
    _.openStream() << new FileInputStream("../../../../resources/test_data.csv.gz")

    then:
    br.ready()
}
}

或者我会更好地编写一个私有方法来 return 流？

Answer 1

可测试性的问题几乎总是一样的：没有依赖注入。这种情况也不例外。这里手头的具体问题是你不能简单地提供一个 file://... URL 因为你使用 Apache Commons Validate 为 isValid(url).

产生 false

解决方案是让您的 class 通过 setter 或额外的构造函数注入 UrlValidator。我在示例代码中选择 setter 方法。如果您不喜欢 public setter，我什至将 setter 包范围化，因此您可以将测试放入与原始 class 相同的包中（无论如何都是常见的做法）才能访问它。

现在只要在测试时注入一个mock就可以愉快的使用一个文件URL:

package de.scrum_master.Whosebug;

import org.apache.commons.validator.routines.UrlValidator;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.zip.GZIPInputStream;

public class DownloadFile {
  private UrlValidator urlValidator = UrlValidator.getInstance();

  void setUrlValidator(UrlValidator urlValidator) {
    this.urlValidator = urlValidator;
  }

  public BufferedReader downloadGzipCsvFile(String url) throws MalformedURLException {
    BufferedReader br = null;
    if (urlValidator.isValid(url)) {
      try {
        br = new BufferedReader(
          new InputStreamReader(
            new GZIPInputStream(
              new URL(url).openStream()
            )
          )
        );
      } catch (IOException e) {
        e.printStackTrace();
        System.exit(0);
      }
    }
    else {
      throw new MalformedURLException("Supplied URL: " + url + " is an invalid URL");
    }
    return br;
  }
}

package de.scrum_master.Whosebug

import org.apache.commons.validator.routines.UrlValidator
import spock.lang.Specification

class DownloadFileSpec extends Specification {
  def "Should return buffered reader for url for gzip csv file"() {
    given:
    String url = new File("src/test/resources/test_data.csv.gz").toURI().toURL()
    def downloadFile = new DownloadFile()
    downloadFile.urlValidator = Mock(UrlValidator) {
      isValid(_) >> true
    }

    when:
    def bufferedReader = downloadFile.downloadGzipCsvFile(url)

    then:
    bufferedReader.ready()
  }
}

顺便说一句，该测试还展示了如何将从相对路径创建的 File 实例转换为文件 URL.

P.S.: 您也可以使用 PowerMock 来模拟静态方法或构造函数，这也可以解决问题。但我相信，每当你需要 PowerMock 时，它就是一种代码味道，你应该重构，这就是我为你所做的。

用 spock 模拟链式方法 return

Mock return of chained method with spock

java

groovy

unit-testing

spock