Examples with Origin com.hotels.styx.api.extension.Origin used on opensource projects

Example 21 with Origin

use of com.hotels.styx.api.extension.Origin in project styx by ExpediaGroup.

the class RoundRobinStrategyTest method remoteHostFor.

private static RemoteHost remoteHostFor(String host, int port) {
    Origin origin = newOriginBuilder(host, port).build();
    LoadBalancingMetricSupplier metric = mock(LoadBalancingMetricSupplier.class);
    when(metric.loadBalancingMetric()).thenReturn(new LoadBalancingMetric(45));
    return remoteHost(origin, mock(HttpHandler.class), metric);
}

Example 22 with Origin

use of com.hotels.styx.api.extension.Origin in project styx by ExpediaGroup.

the class OriginRestrictionLoadBalancingStrategyTest method dontDelegateWhenOriginRestrictionCookieMatches.

@Test
public void dontDelegateWhenOriginRestrictionCookieMatches() {
    Random mockRandom = mock(Random.class);
    when(mockRandom.nextInt(eq(1))).thenReturn(0);
    strategy = new OriginRestrictionLoadBalancingStrategy(new ActiveOrigins() {

        @Override
        public Iterable<RemoteHost> snapshot() {
            return origins;
        }

        @Override
        public String getApplicationId() {
            return null;
        }

        @Override
        public List<Origin> origins() {
            return null;
        }
    }, delegate, mockRandom);
    strategy.choose(lbPreference(Optional.of("origin-1")));
    strategy.choose(lbPreference(Optional.of("origin-1")));
    strategy.choose(lbPreference(Optional.of("origin-1")));
    strategy.choose(lbPreference(Optional.of("origin-1")));
    strategy.choose(lbPreference(Optional.of("origin-1")));
    strategy.choose(lbPreference(Optional.of("origin-1")));
    verify(delegate, never()).choose(any(LoadBalancer.Preferences.class));
}

Example 23 with Origin

use of com.hotels.styx.api.extension.Origin in project styx by ExpediaGroup.

the class OriginRestrictionLoadBalancingStrategyTest method usesAllOriginsWhenCookieIsAbsent.

@Test
public void usesAllOriginsWhenCookieIsAbsent() {
    Random mockRandom = mock(Random.class);
    strategy = new OriginRestrictionLoadBalancingStrategy(new ActiveOrigins() {

        @Override
        public Iterable<RemoteHost> snapshot() {
            return origins;
        }

        @Override
        public String getApplicationId() {
            return null;
        }

        @Override
        public List<Origin> origins() {
            return null;
        }
    }, delegate, mockRandom);
    Optional<RemoteHost> chosenOne = strategy.choose(lbPreference(empty()));
    assertThat(chosenOne.get(), is(origins.get(0)));
    verify(mockRandom, never()).nextInt(any(Integer.class));
    verify(delegate).choose(any(LoadBalancer.Preferences.class));
}

Example 24 with Origin

use of com.hotels.styx.api.extension.Origin in project styx by ExpediaGroup.

the class OriginRestrictionLoadBalancingStrategyTest method randomlyChoosesOneOfTheOriginsWithInvalidPreferences.

@Test
public void randomlyChoosesOneOfTheOriginsWithInvalidPreferences() {
    Random mockRandom = mock(Random.class);
    when(mockRandom.nextInt(any(Integer.class))).thenReturn(3);
    strategy = new OriginRestrictionLoadBalancingStrategy(new ActiveOrigins() {

        @Override
        public Iterable<RemoteHost> snapshot() {
            return origins;
        }

        @Override
        public String getApplicationId() {
            return null;
        }

        @Override
        public List<Origin> origins() {
            return null;
        }
    }, delegate, mockRandom);
    Optional<RemoteHost> chosenOne = strategy.choose(lbPreference(Optional.of("*-01")));
    // Chooses one from *all (= 7)* origins:
    verify(mockRandom).nextInt(eq(7));
    assertThat(chosenOne.get().id(), is(id("origin-3")));
}

Example 25 with Origin

use of com.hotels.styx.api.extension.Origin in project styx by ExpediaGroup.

the class OriginRestrictionLoadBalancingStrategyTest method usesSingleOriginMatchingRegularExpression.

@Test
public void usesSingleOriginMatchingRegularExpression() {
    Random mockRandom = mock(Random.class);
    when(mockRandom.nextInt(eq(1))).thenReturn(0);
    strategy = new OriginRestrictionLoadBalancingStrategy(new ActiveOrigins() {

        @Override
        public Iterable<RemoteHost> snapshot() {
            return origins;
        }

        @Override
        public String getApplicationId() {
            return null;
        }

        @Override
        public List<Origin> origins() {
            return null;
        }
    }, delegate, mockRandom);
    Optional<RemoteHost> chosenOne = strategy.choose(lbPreference(Optional.of("origin-1")));
    assertThat(chosenOne.get(), is(origins.get(1)));
}