Mocking DataSource #groovy #spock

SofmToggleLockStateTest.groovy

package com.containerstore.order.service

import org.joda.time.LocalDate
import org.springframework.jdbc.core.JdbcTemplate
import spock.lang.Specification
import spock.lang.Unroll

import javax.sql.DataSource
import java.sql.Connection
import java.sql.DatabaseMetaData

class SofmToggleLockStateTest extends Specification {

  @Unroll
  def 'Honor order_services_faux_sofm feature'() {
    given:
      def jdbcTemplate = jdbc()
      def feature = featureService(featureEnabled)
      def lock = new SofmToggleLockState(feature, jdbcTemplate)

    when:
      lock.toggleLock('027372662', LocalDate.now())

    then: 'Only call DB procedure when faux feature not enabled'
      expectedCallCount * jdbcTemplate.execute(*_)

    where:
      featureEnabled | expectedCallCount
      true           | 0
      false          | 1

  }

  def featureService(enabled) {
    Mock(FeatureService) {
      it.isFeatureActiveToday(*_) >> enabled
    }
  }

  def jdbc() {
    Mock(JdbcTemplate) {
      it.call(*_) >> [:]

      it.getDataSource() >> Mock(DataSource) {
        it.getConnection() >> Mock(Connection) {
          it.getMetaData() >> Mock(DatabaseMetaData)
        }
      }
    }
  }
}