src/test/java/com/ericsson/gerrit/plugins/syncevents/SyncEventsRestApiServletTest.java

// Copyright (C) 2016 Ericsson
//
// 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.syncevents;

import static com.google.common.net.MediaType.JSON_UTF_8;
import static javax.servlet.http.HttpServletResponse.SC_BAD_REQUEST;
import static javax.servlet.http.HttpServletResponse.SC_NOT_FOUND;
import static javax.servlet.http.HttpServletResponse.SC_NO_CONTENT;
import static javax.servlet.http.HttpServletResponse.SC_UNSUPPORTED_MEDIA_TYPE;
import static org.easymock.EasyMock.expect;
import static org.easymock.EasyMock.expectLastCall;
import static org.easymock.EasyMock.isA;

import com.google.common.net.MediaType;
import com.google.gerrit.common.EventDispatcher;
import com.google.gerrit.reviewdb.client.Project;
import com.google.gerrit.server.events.EventTypes;
import com.google.gerrit.server.events.RefEvent;
import com.google.gwtorm.client.KeyUtil;
import com.google.gwtorm.server.OrmException;
import com.google.gwtorm.server.StandardKeyEncoder;

import org.easymock.EasyMockSupport;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.StringReader;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

public class SyncEventsRestApiServletTest extends EasyMockSupport {

  private EventDispatcher dispatcher = createStrictMock(EventDispatcher.class);
  private SyncEventsRestApiServlet syncEventsRestApiServlet;
  private HttpServletRequest req;
  private HttpServletResponse rsp;

  @BeforeClass
  public static void setup() {
    EventTypes.register(RefReplicationDoneEvent.TYPE, RefReplicationDoneEvent.class);
    KeyUtil.setEncoderImpl(new StandardKeyEncoder());
  }

  @Before
  public void createSyncEventsRestApiServlet() throws Exception {
    syncEventsRestApiServlet = new SyncEventsRestApiServlet(dispatcher);
    req = createNiceMock(HttpServletRequest.class);
    rsp = createNiceMock(HttpServletResponse.class);
    expect(req.getContentType()).andReturn(MediaType.JSON_UTF_8.toString());
  }

  @Test
  public void testDoPostRefReplicationDoneEvent() throws Exception {
    String event = "{\"project\":\"gerrit/test-sync-index\",\"ref\":"
        + "\"refs/changes/76/669676/2\",\"nodesCount\":1,\"type\":"
        + "\"ref-replication-done\",\"eventCreatedOn\":1451415011}";
    expect(req.getReader())
        .andReturn(new BufferedReader(new StringReader(event)));
    dispatcher.postEvent(isA(RefReplicationDoneEvent.class));
    rsp.setStatus(SC_NO_CONTENT);
    expectLastCall().once();
    replayAll();

    syncEventsRestApiServlet.doPost(req, rsp);
    verifyAll();
  }

  @Test
  public void testDoPostDispatcherFailure() throws Exception {
    String event = "{\"project\":\"gerrit/test-sync-index\",\"ref\":"
        + "\"refs/changes/76/669676/2\",\"nodesCount\":1,\"type\":"
        + "\"ref-replication-done\",\"eventCreatedOn\":1451415011}";
    expect(req.getReader())
        .andReturn(new BufferedReader(new StringReader(event)));
    dispatcher.postEvent(isA(RefReplicationDoneEvent.class));
    expectLastCall().andThrow(new OrmException("some Error"));
    rsp.sendError(SC_NOT_FOUND, "Change not found\n");
    expectLastCall().once();
    replayAll();

    syncEventsRestApiServlet.doPost(req, rsp);
    verifyAll();
  }

  @Test
  public void testDoPostBadRequest() throws Exception {
    expect(req.getReader()).andThrow(new IOException());
    replayAll();
    syncEventsRestApiServlet.doPost(req, rsp);
    verifyAll();
  }

  @Test
  public void testDoPostWrongMediaType() throws Exception {
    resetAll();
    expect(req.getContentType())
        .andReturn(MediaType.APPLICATION_XML_UTF_8.toString()).anyTimes();
    rsp.sendError(SC_UNSUPPORTED_MEDIA_TYPE,
        "Expecting " + JSON_UTF_8.toString() + " content type");
    expectLastCall().once();
    replayAll();
    syncEventsRestApiServlet.doPost(req, rsp);
    verifyAll();
  }

  @Test
  public void testDoPostErrorWhileSendingErrorMessage() throws Exception {
    expect(req.getReader()).andThrow(new IOException("someError"));
    rsp.sendError(SC_BAD_REQUEST, "someError");
    expectLastCall().andThrow(new IOException("someOtherError"));
    replayAll();
    syncEventsRestApiServlet.doPost(req, rsp);
    verifyAll();
  }

  static class RefReplicationDoneEvent extends RefEvent {
    public static final String TYPE = "ref-replication-done";
    public final String project;
    public final String ref;
    public final int nodesCount;

    public RefReplicationDoneEvent(String project, String ref, int nodesCount) {
      super(TYPE);
      this.project = project;
      this.ref = ref;
      this.nodesCount = nodesCount;
    }

    @Override
    public Project.NameKey getProjectNameKey() {
      return new Project.NameKey(project);
    }

    @Override
    public String getRefName() {
      return ref;
    }
  }
}