Spock - 如何检查Spy对象的方法调用计数?

时间:2015-10-15 20:11:26

标签: unit-testing spock

我很难让这个spock测试工作。

我有一个Spring repo / DAO类,它多次调用存储过程。我正在尝试编写一个单元测试来验证SP是否被调用了'x'次(对createSP()方法的调用是3次)。

public class PlanConditionRepositoryImpl{

    ....

    public void write() {
        for (int i=0; i<3; i++) {
            createSP(new ConditionGroup(), new Condition()).call();
        }
    }

    protected StoredProcedure<Void> createSP(ConditionGroup planConditionGroup, Condition planCondition) {
        return new StoredProcedure<Void>()
                .jdbcTemplate(getJdbcTemplate())
                .schemaName(SCHEMA_NAME);
    }
}       

然而,下面的实现并没有这样做。如何实现调用计数检查?或者我如何避免调用createSP()方法的实际实现。

def write(){
    def repo = Spy(PlanConditionRepositoryImpl){
        createSP(_, _) >> Spy(StoredProcedure){
            call() >> {
                //do nothing
            }
        }
    }
    when:
    repo.write()

    then:
    3 * repo.createSP(_, _)
}

这就是我用hack解决它的方法。但是,是否存在使用Spock基于交互的测试而不引入额外变量的解决方案?

def "spec"() {
    given:
    def count = 0
    def spy = Spy(PlanConditionRepositoryImpl){
        createSP(_, _) >> {count++}
    }

    when:
    spy.write()

    then:
    count == 3
}

1 个答案:

答案 0 :(得分:3)

你需要的是部分模拟,看看docs。然而,正如我所说,部分嘲笑基本上是不好的做法,可能表明设计不好:

  

(在使用此功能之前请三思。更改可能更好   根据规范设计代码。)

关于部分嘲笑:

// this is now the object under specification, not a collaborator
def persister = Spy(MessagePersister) {
  // stub a call on the same object
  isPersistable(_) >> true
}

when:
persister.receive("msg")

then:
// demand a call on the same object
1 * persister.persist("msg")

以下是测试的编写方式:

@Grab('org.spockframework:spock-core:1.0-groovy-2.4')
@Grab('cglib:cglib-nodep:3.1')

import spock.lang.*

class Test extends Specification {
    def "spec"() {
        given:    
        def mock = Mock(StoredProcedure)
        def spy = Spy(PlanConditionRepositoryImpl) 

        when:
        spy.write()

        then:
        3 * spy.createSP() >> mock
        3 * mock.run()
    }
}

class PlanConditionRepositoryImpl {

    void write() {
        for (int i = 0; i < 3; i++) {
            createSP().run()
        }
    }

    StoredProcedure createSP() {
        new StoredProcedure()    
    }
}

class StoredProcedure {
    def run() {}
}