content: /**
   * Copyright (C) 2010-2016 eBusiness Information, Excilys Group
   * Copyright (C) 2016-2020 the AndroidAnnotations project
   *
   * Licensed under the Apache License, Version 2.0 (the "License"); you may not
   * use this file except in compliance with the License. You may obtain a copy of
   * the License at
   *
   * http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed To in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
   * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
   * License for the specific language governing permissions and limitations under
   * the License.
   */
  package org.androidannotations.rest.spring.test;
  
  import static org.mockito.ArgumentMatchers.argThat;
  import static org.mockito.Mockito.verify;
  
  import java.util.Arrays;
  import java.util.HashMap;
  import java.util.Map;
  import java.util.Set;
  
  import org.apache.http.Header;
  import org.apache.http.message.BasicHeader;
  import org.mockito.ArgumentMatcher;
  import org.mockito.ArgumentMatchers;
  import org.mockito.Mockito;
  import org.robolectric.shadows.httpclient.FakeHttp;
  import org.springframework.http.HttpEntity;
  import org.springframework.http.HttpHeaders;
  import org.springframework.http.HttpMethod;
  import org.springframework.http.HttpStatus;
  import org.springframework.web.client.RestTemplate;
  
  import junit.framework.AssertionFailedError;
  
  public class RequestTestBuilder {
  	private MyService_ myService = new MyService_(null);
  	private HttpEntityArgumentMatcher entityArgumentMatcher = new HttpEntityArgumentMatcher();
  	private RestTemplate restTemplate;
  
  	private Map<String, String> requestHeaders = new HashMap<String, String>();
  	private Map<String, String> requestCookies = new HashMap<String, String>();
  	private Map<String, String> responseCookies = new HashMap<String, String>();
  	private String responseContent;
  	private boolean hasUrlVariables;
  
  	public static RequestTestBuilder build() {
  		return new RequestTestBuilder();
  	}
  
  	public RequestTestBuilder() {
  		restTemplate = Mockito.spy(myService.getRestTemplate());
  		myService.setRestTemplate(restTemplate);
  	}
  
  	public RequestTestBuilder requestHeader(String name, String value) {
  		requestHeaders.put(name, value);
  		myService.setHeader(name, value);
  		return this;
  	}
  
  	public RequestTestBuilder expectedHeader(String name, String value) {
  		requestHeaders.put(name, value);
  		return this;
  	}
  
  	public RequestTestBuilder requestCookie(String name, String value) {
  		requestCookies.put(name, value);
  		myService.setCookie(name, value);
  		return this;
  	}
  
  	public RequestTestBuilder responseCookie(String name, String value) {
  		responseCookies.put(name, value);
  		return this;
  	}
  
  	public RequestTestBuilder responseContent(String responseContent) {
  		this.responseContent = responseContent;
  		return this;
  	}
  
  	public RequestTestBuilder hasUrlVariables(boolean hasUrlVariables) {
  		this.hasUrlVariables = hasUrlVariables;
  		return this;
  	}
  
  	public void asserts(RequestTestBuilderExecutor executor) {
  		// Prepare fake response
  		prepareFakeResponse();
  
  		// Make the call
  		executor.execute(myService);
  
  		// Checks
  		checkRequest();
  		checkResponseCookies();
  	}
  
  	private void prepareFakeResponse() {
  		Set<String> cookiesNames = responseCookies.keySet();
  		Header[] headers = new Header[1 + responseCookies.size()];
  		headers[0] = new BasicHeader("content-type", "application/json");
  
  		int i = 1;
  		for (String cookieName : cookiesNames) {
  			headers[i++] = new BasicHeader("Set-Cookie", cookieName + "=" + responseCookies.get(cookieName));
  		}
  
  		String responseBody = responseContent != null ? responseContent.replaceAll("'", "\"") : "";
  
  		FakeHttp.addPendingHttpResponse(HttpStatus.OK.value(), responseBody, headers);
  	}
  
  	private void checkRequest() {
  		if (hasUrlVariables) {
  			verify(restTemplate).exchange(ArgumentMatchers.anyString(), //
  					ArgumentMatchers.any(HttpMethod.class), //
  					argThat(entityArgumentMatcher), //
  					ArgumentMatchers.<Class<Object>> any(), //
  					ArgumentMatchers.<Map<String, Object>> any());
  		} else {
  			verify(restTemplate).exchange(ArgumentMatchers.anyString(), //
  					ArgumentMatchers.any(HttpMethod.class), //
  					argThat(entityArgumentMatcher), //
  					ArgumentMatchers.<Class<Object>> any());
  		}
  	}
  
  	private void checkResponseCookies() throws AssertionFailedError {
  		for (String responseCookieName : responseCookies.keySet()) {
  			String cookieValue = myService.getCookie(responseCookieName);
  			if (cookieValue == null || !cookieValue.equals(responseCookies.get(responseCookieName))) {
  				throw new AssertionFailedError("Response cookie " + responseCookieName + " wasn't set");
  			}
  		}
  	}
  
  	public class HttpEntityArgumentMatcher implements ArgumentMatcher<HttpEntity<Void>> {
  		@Override
  		public boolean matches(HttpEntity<Void> argument) {
  			HttpHeaders httpHeaders = argument.getHeaders();
  
  			// Check that cookies set earlier are sent in the request
  			if (requestCookies.size() > 0) {
  				String[] httpCookies = httpHeaders.getFirst("Cookie").split(";");
  				for (String cookieName : requestCookies.keySet()) {
  					if (Arrays.binarySearch(httpCookies, cookieName + "=" + requestCookies.get(cookieName)) == -1) {
  						throw new AssertionFailedError("The cookie " + cookieName + " is missing!");
  					}
  				}
  			}
  
  			// Check that headers set earlier are sent in the request
  			for (String headerName : requestHeaders.keySet()) {
  				if (!(httpHeaders.containsKey(headerName) && httpHeaders.getFirst(headerName).equals(requestHeaders.get(headerName)))) {
  					throw new AssertionFailedError("The header " + headerName + " is missing!");
  				}
  			}
  
  			return true;
  		}
  	}
  
  	public interface RequestTestBuilderExecutor {
  		void execute(MyService myService);
  	}
  
  }
  