HttpRequestMatcherTest.java

package com.github.dreamhead.moco.handler.failover;

import com.github.dreamhead.moco.HttpMethod;
import com.github.dreamhead.moco.HttpProtocolVersion;
import com.github.dreamhead.moco.HttpRequest;
import com.github.dreamhead.moco.model.DefaultHttpRequest;
import com.github.dreamhead.moco.model.HttpRequestFailoverMatcher;
import org.junit.Test;

import static com.google.common.collect.ImmutableMap.of;
import static org.hamcrest.CoreMatchers.is;
import static org.junit.Assert.assertThat;

public class HttpRequestMatcherTest {
    @Test
    public void should_be_match_if_request_is_same() {
        HttpRequest request = DefaultHttpRequest.builder()
                .withUri("/uri")
                .withVersion(HttpProtocolVersion.VERSION_1_1)
                .withMethod(HttpMethod.POST)
                .withTextContent("proxy")
                .withHeaders(of("Cookie", "loggedIn=true", "Host", "localhost:12306"))
                .build();

        assertThat(new HttpRequestFailoverMatcher(request).match(request), is(true));
    }

    @Test
    public void should_not_be_match_if_request_is_different() {
        HttpRequest request = DefaultHttpRequest.builder()
                .withUri("/uri")
                .withVersion(HttpProtocolVersion.VERSION_1_1)
                .withMethod(HttpMethod.POST)
                .withTextContent("proxy")
                .withHeaders(of("Cookie", "loggedIn=true", "Host", "localhost:12306"))
                .build();

        HttpRequest another = DefaultHttpRequest.builder()
                .withUri("/uri")
                .withVersion(HttpProtocolVersion.VERSION_1_1)
                .withMethod(HttpMethod.POST)
                .withTextContent("different")
                .withHeaders(of("Cookie", "loggedIn=true", "Host", "localhost:12306"))
                .build();

        assertThat(new HttpRequestFailoverMatcher(request).match(another), is(false));
    }

    @Test
    public void should_not_be_match_if_uri_is_different() {
        HttpRequest request = DefaultHttpRequest.builder()
                .withUri("/uri")
                .withVersion(HttpProtocolVersion.VERSION_1_1)
                .withMethod(HttpMethod.POST)
                .withTextContent("proxy")
                .withUri("/foo")
                .withHeaders(of("Cookie", "loggedIn=true", "Host", "localhost:12306"))
                .build();

        HttpRequest another = DefaultHttpRequest.builder()
                .withUri("/uri")
                .withVersion(HttpProtocolVersion.VERSION_1_1)
                .withMethod(HttpMethod.POST)
                .withTextContent("proxy")
                .withUri("/bar")
                .withHeaders(of("Cookie", "loggedIn=true", "Host", "localhost:12306"))
                .build();

        assertThat(new HttpRequestFailoverMatcher(request).match(another), is(false));
    }

    @Test
    public void should_be_match_if_failover_field_is_null() {
        HttpRequest request = DefaultHttpRequest.builder()
                .withUri("/uri")
                .withVersion(HttpProtocolVersion.VERSION_1_1)
                .withMethod(HttpMethod.POST)
                .withTextContent("proxy")
                .withHeaders(of("Cookie", "loggedIn=true", "Host", "localhost:12306"))
                .build();

        HttpRequest failover = DefaultHttpRequest.builder()
                .withUri("/uri")
                .withMethod(HttpMethod.POST)
                .withTextContent("proxy")
                .withHeaders(of("Cookie", "loggedIn=true", "Host", "localhost:12306"))
                .build();

        assertThat(new HttpRequestFailoverMatcher(failover).match(request), is(true));
    }

    @Test
    public void should_be_match_even_if_target_request_has_more_headers() {
        HttpRequest request = DefaultHttpRequest.builder()
                .withUri("/uri")
                .withVersion(HttpProtocolVersion.VERSION_1_1)
                .withMethod(HttpMethod.POST)
                .withTextContent("proxy")
                .withHeaders(of("Cookie", "loggedIn=true", "Host", "localhost:12306"))
                .build();

        HttpRequest failover = DefaultHttpRequest.builder()
                .withUri("/uri")
                .withVersion(HttpProtocolVersion.VERSION_1_1)
                .withMethod(HttpMethod.POST)
                .withTextContent("proxy")
                .withHeaders(of("Host", "localhost:12306"))
                .build();

        assertThat(new HttpRequestFailoverMatcher(failover).match(request), is(true));
    }
}