// Copyright (C) 2017 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.webhooks.processors;

import static com.google.common.truth.Truth.assertThat;
import static org.mockito.Mockito.when;

import com.google.gerrit.entities.Project;
import com.google.gerrit.server.events.ProjectCreatedEvent;
import com.google.gerrit.server.events.ProjectEvent;
import com.google.gerrit.server.events.RefUpdatedEvent;
import com.googlesource.gerrit.plugins.webhooks.EventProcessor;
import com.googlesource.gerrit.plugins.webhooks.RemoteConfig;
import java.util.Optional;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnitRunner;

@RunWith(MockitoJUnitRunner.class)
public class AbstractEventProcessorTest {
  private static final String PROJECT = "p";
  private static final Project.NameKey PROJECT_NAME = Project.nameKey(PROJECT);

  private static final ProjectCreatedEvent PROJECT_CREATED =
      new ProjectCreatedEvent() {
        @Override
        public Project.NameKey getProjectNameKey() {
          return PROJECT_NAME;
        }
      };

  private static final RefUpdatedEvent REF_UPDATED =
      new RefUpdatedEvent() {
        @Override
        public Project.NameKey getProjectNameKey() {
          return PROJECT_NAME;
        }
      };

  @Mock private RemoteConfig remote;

  private TestEventProcessor processor;

  @Before
  public void setup() throws Exception {
    processor = new TestEventProcessor();
  }

  @Test
  public void eventsNotSpecifiedAllEventsShouldProcess() throws Exception {
    when(remote.getEvents()).thenReturn(new String[] {});
    boolean actual = processor.shouldProcess(PROJECT_CREATED, remote);
    assertThat(actual).isTrue();

    actual = processor.shouldProcess(REF_UPDATED, remote);
    assertThat(actual).isTrue();
  }

  @Test
  public void specifiedEventTypesShouldProcess() throws Exception {
    when(remote.getEvents()).thenReturn(new String[] {"project-created"});
    boolean actual = processor.shouldProcess(PROJECT_CREATED, remote);
    assertThat(actual).isTrue();
  }

  @Test
  public void nonSpecifiedProjectEventTypesNotProcess() throws Exception {
    when(remote.getEvents()).thenReturn(new String[] {"project-created"});
    boolean actual = processor.shouldProcess(REF_UPDATED, remote);
    assertThat(actual).isFalse();
  }

  private static class TestEventProcessor extends AbstractEventProcessor {
    @Override
    public Optional<EventProcessor.Request> doProcess(ProjectEvent event, RemoteConfig config) {
      // do nothing
      return Optional.empty();
    }
  }
}
