src/test/java/com/googlesource/gerrit/plugins/multisite/consumer/IndexEventSubscriberTest.java

// 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.nowTs());
  }
}