RESTClient Groovy с расширениями Спока

Вышеупомянутое расширение сообщает Споку, что когда он встречает @RESTClient , он должен создать перехватчик с аргументами, обозначающими поле, в которое должен быть установлен экземпляр RESTClient, URL-адрес веб-службы и логическое значение, чтобы указать, следует ли использовать макет. используется или реальный веб-сервис должен быть использован. (Значение двух значений можно объединить, чтобы получить только URL, и если он установлен, то должен использоваться реальный веб-сервис.)

Перехватчик создаст либо фиктивный RESTClient, либо реальный. Это будет сделано, когда поле клиента должно быть установлено в классе InjectedRestServiceTest. Если требуется настоящий RESTClient, то экземпляр, использующий URL, будет создан и установлен с использованием fieldInfo.writeValue.метод. Если макет требуется, то создается макет Mockito RESTClient, и ожидаемое поведение макетируется с использованием экземпляров класса HttpResponseDecorator (хотя это могут быть и макеты.) Опять же, это устанавливается с  помощью метода fieldInfo.writeValue :

import groovy.mock.interceptor.MockFor
import groovyx.net.http.HttpResponseDecorator
import groovyx.net.http.RESTClient
import org.spockframework.runtime.extension.AbstractMethodInterceptor;
import org.spockframework.runtime.extension.IMethodInvocation;
import org.spockframework.runtime.model.FieldInfo;
import org.spockframework.runtime.model.SpecInfo;

import static org.mockito.Mockito.*
import static org.mockito.Matchers.*

class RESTWebClientMethodInterceptor extends AbstractMethodInterceptor {

 private final FieldInfo fieldInfo
 private final String url
 private final boolean useMock

 RESTWebClientMethodInterceptor(FieldInfo fieldInfo, String url, boolean useMock) {
  this.fieldInfo = fieldInfo
  this.url = url
  this.useMock = useMock
 }

 @Override
 void interceptSetupMethod(IMethodInvocation methodInvocation) {
  setupRESTClient(methodInvocation.target)
  methodInvocation.proceed()
 }

 @Override
 void install(SpecInfo specInfo) {
  specInfo.setupMethod.addInterceptor this
 }

 private void setupRESTClient(target) {

  if (useMock) {

   def xmlServerOne =
     """
      <serverStatus>
        <serverName>ServerOne</serverName>
        <isRunning>true</isRunning>
      </serverStatus>
      """

   def xmlServerTwo =
     """
      <serverStatus>
        <serverName>ServerTwo</serverName>
        <isRunning>false</isRunning>
      </serverStatus>
      """

   def httpResponseDecoratorServerOne = new HttpResponseDecorator(
     null, new XmlParser().parseText(xmlServerOne))
   def httpResponseDecoratorServerTwo = new HttpResponseDecorator(
     null, new XmlParser().parseText(xmlServerTwo))
   def mapServerOne = [path:"server/status/ServerOne"]
   def mapServerTwo = [path:"server/status/ServerTwo"]

   RESTClient mockRESTClient = org.mockito.Mockito.mock(RESTClient)
   when(mockRESTClient.get(mapServerOne)).thenReturn(httpResponseDecoratorServerOne);
   when(mockRESTClient.get(mapServerTwo)).thenReturn(httpResponseDecoratorServerTwo);

   fieldInfo.writeValue(target, mockRESTClient)
  }
  else {
   fieldInfo.writeValue(target, new RESTClient(url))
  }
 }

}