| // Copyright (C) 2021 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.googlesource.gerrit.plugins.multisite.consumer; |
| |
| import static org.mockito.ArgumentMatchers.any; |
| import static org.mockito.Mockito.mock; |
| import static org.mockito.Mockito.never; |
| import static org.mockito.Mockito.times; |
| import static org.mockito.Mockito.verify; |
| import static org.mockito.Mockito.when; |
| |
| import com.google.common.collect.ImmutableList; |
| import com.google.gerrit.entities.Account; |
| import com.google.gerrit.entities.BranchNameKey; |
| import com.google.gerrit.entities.Change; |
| import com.google.gerrit.entities.Project; |
| import com.google.gerrit.server.change.ChangeFinder; |
| import com.google.gerrit.server.events.Event; |
| import com.google.gerrit.server.notedb.ChangeNotes; |
| import com.google.gerrit.server.permissions.PermissionBackendException; |
| import com.google.gerrit.server.util.time.TimeUtil; |
| import com.googlesource.gerrit.plugins.multisite.forwarder.CacheNotFoundException; |
| import com.googlesource.gerrit.plugins.multisite.forwarder.events.AccountIndexEvent; |
| import com.googlesource.gerrit.plugins.multisite.forwarder.events.ChangeIndexEvent; |
| import com.googlesource.gerrit.plugins.multisite.forwarder.events.IndexEvent; |
| import com.googlesource.gerrit.plugins.multisite.forwarder.events.ProjectIndexEvent; |
| import com.googlesource.gerrit.plugins.multisite.forwarder.router.ForwardedEventRouter; |
| import com.googlesource.gerrit.plugins.multisite.forwarder.router.IndexEventRouter; |
| import java.io.IOException; |
| import java.util.List; |
| import java.util.Optional; |
| import org.junit.Test; |
| import org.mockito.Mock; |
| |
| public class IndexEventSubscriberTest extends AbstractSubscriberTestBase { |
| private static final boolean DELETED = false; |
| private static final int CHANGE_ID = 1; |
| private static final String EMPTY_PROJECT_NAME = ""; |
| |
| @Mock protected ChangeFinder changeFinderMock; |
| |
| @SuppressWarnings("unchecked") |
| @Test |
| public void shouldConsumeNonProjectAndNonChangeIndexingEventsTypes() |
| throws IOException, PermissionBackendException, CacheNotFoundException { |
| IndexEvent event = new AccountIndexEvent(1, INSTANCE_ID); |
| |
| objectUnderTest.getConsumer().accept(event); |
| |
| verify(projectsFilter, never()).matches(PROJECT_NAME); |
| verify(eventRouter, times(1)).route(event); |
| verify(droppedEventListeners, never()).onEventDropped(event); |
| } |
| |
| @SuppressWarnings("unchecked") |
| @Test |
| public void shouldConsumeDeleteChangeIndexEventWithEmptyProjectNameWhenFound() |
| throws IOException, PermissionBackendException, CacheNotFoundException { |
| ChangeIndexEvent event = new ChangeIndexEvent(EMPTY_PROJECT_NAME, CHANGE_ID, true, INSTANCE_ID); |
| |
| ChangeNotes changeNotesMock = mock(ChangeNotes.class); |
| when(changeNotesMock.getChange()).thenReturn(newChange()); |
| when(changeFinderMock.findOne(any(Change.Id.class))).thenReturn(Optional.of(changeNotesMock)); |
| when(projectsFilter.matches(PROJECT_NAME)).thenReturn(true); |
| |
| objectUnderTest.getConsumer().accept(event); |
| |
| verify(projectsFilter, times(1)).matches(PROJECT_NAME); |
| verify(eventRouter, times(1)).route(event); |
| } |
| |
| @SuppressWarnings("unchecked") |
| @Test |
| public void shouldNOTConsumeDeleteChangeIndexEventWithEmptyProjectNameWhenNotFound() |
| throws IOException, PermissionBackendException, CacheNotFoundException { |
| ChangeIndexEvent event = new ChangeIndexEvent("", CHANGE_ID, true, INSTANCE_ID); |
| |
| when(changeFinderMock.findOne(any(Change.Id.class))).thenReturn(Optional.empty()); |
| when(projectsFilter.matches(EMPTY_PROJECT_NAME)).thenReturn(false); |
| |
| objectUnderTest.getConsumer().accept(event); |
| |
| verify(projectsFilter, times(1)).matches(EMPTY_PROJECT_NAME); |
| verify(eventRouter, never()).route(event); |
| } |
| |
| @SuppressWarnings("rawtypes") |
| @Override |
| protected ForwardedEventRouter eventRouter() { |
| return mock(IndexEventRouter.class); |
| } |
| |
| @Override |
| protected List<Event> events() { |
| return ImmutableList.of( |
| new ProjectIndexEvent(PROJECT_NAME, INSTANCE_ID), |
| new ChangeIndexEvent(PROJECT_NAME, CHANGE_ID, DELETED, INSTANCE_ID)); |
| } |
| |
| @Override |
| protected AbstractSubcriber objectUnderTest() { |
| return new IndexEventSubscriber( |
| (IndexEventRouter) eventRouter, |
| asDynamicSet(droppedEventListeners), |
| NODE_INSTANCE_ID, |
| msgLog, |
| subscriberMetrics, |
| cfg, |
| projectsFilter, |
| changeFinderMock); |
| } |
| |
| private Change newChange() { |
| return new Change( |
| Change.key(Integer.toString(CHANGE_ID)), |
| Change.id(CHANGE_ID), |
| Account.id(9999), |
| BranchNameKey.create(Project.nameKey(PROJECT_NAME), "refs/heads/master"), |
| TimeUtil.now()); |
| } |
| } |