| // Copyright (C) 2015 The Android Open Source 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 com.ericsson.gerrit.plugins.highavailability.cache; |
| |
| import static com.github.tomakehurst.wiremock.client.WireMock.aResponse; |
| import static com.github.tomakehurst.wiremock.client.WireMock.any; |
| import static com.github.tomakehurst.wiremock.client.WireMock.anyUrl; |
| import static com.github.tomakehurst.wiremock.client.WireMock.givenThat; |
| import static com.github.tomakehurst.wiremock.client.WireMock.postRequestedFor; |
| import static com.github.tomakehurst.wiremock.client.WireMock.urlEqualTo; |
| import static com.github.tomakehurst.wiremock.client.WireMock.verify; |
| import static com.github.tomakehurst.wiremock.core.WireMockConfiguration.options; |
| import static com.google.common.truth.Truth.assertThat; |
| |
| import com.github.tomakehurst.wiremock.junit.WireMockRule; |
| import com.google.common.cache.LoadingCache; |
| import com.google.gerrit.acceptance.LightweightPluginDaemonTest; |
| import com.google.gerrit.acceptance.NoHttpd; |
| import com.google.gerrit.acceptance.TestPlugin; |
| import com.google.gerrit.acceptance.UseLocalDisk; |
| import com.google.gerrit.acceptance.UseSsh; |
| import com.google.gerrit.acceptance.config.GerritConfig; |
| import com.google.gerrit.acceptance.config.GlobalPluginConfig; |
| import com.google.gerrit.entities.AccountGroup; |
| import com.google.inject.Inject; |
| import com.google.inject.name.Named; |
| import java.util.Collections; |
| import java.util.Set; |
| import java.util.concurrent.CountDownLatch; |
| import java.util.concurrent.TimeUnit; |
| import org.apache.http.HttpStatus; |
| import org.junit.Rule; |
| import org.junit.Test; |
| |
| @NoHttpd |
| @UseSsh |
| @TestPlugin( |
| name = "high-availability", |
| sysModule = "com.ericsson.gerrit.plugins.highavailability.Module", |
| httpModule = "com.ericsson.gerrit.plugins.highavailability.HttpModule") |
| public class CacheEvictionIT extends LightweightPluginDaemonTest { |
| private static final int PORT = 18888; |
| private static final String URL = "http://localhost:" + PORT; |
| private static final String GROUP_CACHE = "ldap_groups"; |
| |
| @Rule public WireMockRule wireMockRule = new WireMockRule(options().port(PORT)); |
| |
| @Inject |
| @Named(GROUP_CACHE) |
| private LoadingCache<String, Set<AccountGroup.UUID>> membershipCache; |
| |
| private final CountDownLatch expectedRequestLatch = new CountDownLatch(1); |
| |
| @Override |
| public void setUpTestPlugin() throws Exception { |
| givenThat(any(anyUrl()).willReturn(aResponse().withStatus(HttpStatus.SC_NO_CONTENT))); |
| super.setUpTestPlugin(); |
| } |
| |
| @Test |
| @UseLocalDisk |
| @GlobalPluginConfig(pluginName = "high-availability", name = "peerInfo.static.url", value = URL) |
| @GlobalPluginConfig(pluginName = "high-availability", name = "http.retryInterval", value = "100") |
| @GerritConfig(name = "auth.type", value = "ldap") |
| public void flushProjectsCacheShouldSendPostForEvictingRemoteCache() throws Exception { |
| final String flushRequest = "/plugins/high-availability/cache/" + Constants.PROJECTS; |
| |
| expectRestApiCall(flushRequest); |
| |
| adminSshSession.exec("gerrit flush-caches --cache " + Constants.PROJECTS); |
| assertThat(waitForEvictionEvents()).isTrue(); |
| verify(postRequestedFor(urlEqualTo(flushRequest))); |
| } |
| |
| @Test |
| @UseLocalDisk |
| @GlobalPluginConfig(pluginName = "high-availability", name = "peerInfo.static.url", value = URL) |
| @GlobalPluginConfig(pluginName = "high-availability", name = "http.retryInterval", value = "100") |
| @GerritConfig(name = "auth.type", value = "ldap") |
| public void ldapCacheLoadShouldNotSendAnyPostEvictionForLdapGroups() throws Exception { |
| final String flushRequest = "/plugins/high-availability/cache/" + GROUP_CACHE; |
| String username = "username"; |
| Set<AccountGroup.UUID> groups = Collections.emptySet(); |
| |
| expectRestApiCall(flushRequest); |
| |
| loadLdapGroupMembers(username, groups); |
| loadLdapGroupMembers(username, groups); // For triggering an eviction |
| assertThat(waitForEvictionEvents()).isFalse(); |
| verify(0, postRequestedFor(urlEqualTo(flushRequest))); |
| } |
| |
| private void expectRestApiCall(final String flushRequest) { |
| wireMockRule.addMockServiceRequestListener( |
| (request, response) -> { |
| if (request.getAbsoluteUrl().contains(flushRequest)) { |
| expectedRequestLatch.countDown(); |
| } |
| }); |
| } |
| |
| private boolean waitForEvictionEvents() throws InterruptedException { |
| return expectedRequestLatch.await(5, TimeUnit.SECONDS); |
| } |
| |
| private void loadLdapGroupMembers(String username, Set<AccountGroup.UUID> groups) { |
| membershipCache.put(username, groups); |
| } |
| } |