javatests/com/google/gerrit/acceptance/server/event/InstanceIdInEventIT.java

// Copyright (C) 2020 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.google.gerrit.acceptance.server.event;

import static com.google.common.truth.Truth.assertThat;
import static org.junit.Assert.fail;

import com.google.common.collect.ImmutableList;
import com.google.gerrit.acceptance.AbstractDaemonTest;
import com.google.gerrit.acceptance.NoHttpd;
import com.google.gerrit.acceptance.config.GerritConfig;
import com.google.gerrit.extensions.registration.DynamicItem;
import com.google.gerrit.server.events.Event;
import com.google.gerrit.server.events.EventDispatcher;
import com.google.gerrit.server.events.EventTypes;
import com.google.inject.Inject;
import org.junit.Before;
import org.junit.Test;

@NoHttpd
public class InstanceIdInEventIT extends AbstractDaemonTest {

  public static class TestDispatcher {
    private final DynamicItem<EventDispatcher> eventDispatcher;

    @Inject
    TestDispatcher(DynamicItem<EventDispatcher> eventDispatcher) {
      this.eventDispatcher = eventDispatcher;
    }

    public void postEvent(TestEvent event) {
      try {
        eventDispatcher.get().postEvent(event);
      } catch (Exception e) {
        fail("Exception raised when posting Event " + e.getCause());
      }
    }
  }

  public static class TestEvent extends Event {
    private static final String TYPE = "test-event-instance-id";

    public TestEvent() {
      super(TYPE);
    }
  }

  @Inject private DynamicItem<EventDispatcher> eventDispatcher;
  TestDispatcher testDispatcher;

  @Before
  public void setUp() throws Exception {
    testDispatcher = new TestDispatcher(eventDispatcher);
    EventTypes.register(TestEvent.TYPE, TestEvent.class);
  }

  @Test
  @GerritConfig(name = "gerrit.instanceId", value = "testInstanceId")
  public void shouldSetInstanceIdWhenDefined() {
    testDispatcher.postEvent(new TestEvent());

    ImmutableList<Event> events = eventRecorder.getGenericEvents(TestEvent.TYPE, 1);
    assertThat(events.get(0).instanceId).isEqualTo("testInstanceId");
  }

  @Test
  public void shouldNotSetInstanceIdWhenNotDefined() {
    testDispatcher.postEvent(new TestEvent());

    ImmutableList<Event> events = eventRecorder.getGenericEvents(TestEvent.TYPE, 1);
    assertThat(events.get(0).instanceId).isNull();
  }
}