| /* |
| * ==================================================================== |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you 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. |
| * ==================================================================== |
| * |
| * This software consists of voluntary contributions made by many |
| * individuals on behalf of the Apache Software Foundation. For more |
| * information on the Apache Software Foundation, please see |
| * <http://www.apache.org/>. |
| * |
| */ |
| package org.apache.hc.client5.http.impl.cache; |
| |
| import java.time.Instant; |
| import java.time.temporal.ChronoUnit; |
| import java.util.Arrays; |
| import java.util.Iterator; |
| import java.util.Objects; |
| import java.util.Set; |
| |
| import org.apache.hc.client5.http.cache.HttpCacheEntry; |
| import org.apache.hc.client5.http.cache.Resource; |
| import org.apache.hc.client5.http.cache.ResourceIOException; |
| import org.apache.hc.core5.http.Header; |
| import org.hamcrest.BaseMatcher; |
| import org.hamcrest.Description; |
| import org.hamcrest.Matcher; |
| |
| public class HttpCacheEntryMatcher extends BaseMatcher<HttpCacheEntry> { |
| |
| private final HttpCacheEntry expectedValue; |
| |
| public HttpCacheEntryMatcher(final HttpCacheEntry expectedValue) { |
| this.expectedValue = expectedValue; |
| } |
| |
| @Override |
| public boolean matches(final Object item) { |
| if (item instanceof HttpCacheEntry) { |
| try { |
| final HttpCacheEntry otherValue = (HttpCacheEntry) item; |
| |
| if (!setEqual(expectedValue.getVariants(), otherValue.getVariants())) { |
| return false; |
| } |
| if (!Objects.equals(expectedValue.getRequestMethod(), otherValue.getRequestMethod())) { |
| return false; |
| } |
| if (!Objects.equals(expectedValue.getRequestURI(), otherValue.getRequestURI())) { |
| return false; |
| } |
| if (!headersEqual(expectedValue.requestHeaderIterator(), otherValue.requestHeaderIterator())) { |
| return false; |
| } |
| if (!instantEqual(expectedValue.getRequestInstant(), otherValue.getRequestInstant())) { |
| return false; |
| } |
| if (expectedValue.getStatus() != otherValue.getStatus()) { |
| return false; |
| } |
| if (!headersEqual(expectedValue.headerIterator(), otherValue.headerIterator())) { |
| return false; |
| } |
| if (!instantEqual(expectedValue.getResponseInstant(), otherValue.getResponseInstant())) { |
| return false; |
| } |
| final Resource expectedResource = expectedValue.getResource(); |
| final byte[] expectedContent = expectedResource != null ? expectedResource.get() : null; |
| final Resource otherResource = otherValue.getResource(); |
| final byte[] otherContent = otherResource != null ? otherResource.get() : null; |
| return Arrays.equals(expectedContent, otherContent); |
| } catch (final ResourceIOException ex) { |
| throw new RuntimeException(ex); |
| } |
| } |
| return false; |
| } |
| |
| private static boolean instantEqual(final Instant expected, final Instant other) { |
| final Instant expectedMs = expected != null ? expected.truncatedTo(ChronoUnit.MILLIS) : null; |
| final Instant otherMs = other != null ? other.truncatedTo(ChronoUnit.MILLIS) : null; |
| return Objects.equals(expectedMs, otherMs); |
| } |
| |
| private static boolean headersEqual(final Iterator<Header> expected, final Iterator<Header> other) { |
| while (expected.hasNext()) { |
| final Header h1 = expected.next(); |
| if (!other.hasNext()) { |
| return false; |
| } |
| final Header h2 = other.next(); |
| if (!h1.getName().equals(h2.getName()) || !Objects.equals(h1.getValue(), h2.getValue())) { |
| return false; |
| } |
| } |
| if (other.hasNext()) { |
| return false; |
| } |
| return true; |
| } |
| |
| private static boolean setEqual(final Set<?> expected, final Set<?> actual) { |
| if (expected.size() != actual.size()) { |
| return false; |
| } |
| return actual.containsAll(expected); |
| } |
| |
| @Override |
| public void describeTo(final Description description) { |
| description.appendValue(expectedValue); |
| } |
| |
| public static Matcher<HttpCacheEntry> equivalent(final HttpCacheEntry target) { |
| return new HttpCacheEntryMatcher(target); |
| } |
| |
| } |