| // 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.mockito.ArgumentMatchers.any; |
| import static org.mockito.Mockito.doThrow; |
| import static org.mockito.Mockito.verify; |
| import static org.mockito.Mockito.when; |
| |
| 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.junit.Before; |
| import org.junit.BeforeClass; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.mockito.Mock; |
| import org.mockito.junit.MockitoJUnitRunner; |
| |
| import java.io.BufferedReader; |
| import java.io.IOException; |
| import java.io.StringReader; |
| |
| import javax.servlet.http.HttpServletRequest; |
| import javax.servlet.http.HttpServletResponse; |
| |
| @RunWith(MockitoJUnitRunner.class) |
| public class SyncEventsRestApiServletTest { |
| private static final String ERR_MSG = "some Error"; |
| |
| @Mock |
| private EventDispatcher dispatcher; |
| @Mock |
| private HttpServletRequest req; |
| @Mock |
| private HttpServletResponse rsp; |
| private SyncEventsRestApiServlet syncEventsRestApiServlet; |
| |
| @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); |
| when(req.getContentType()).thenReturn(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}"; |
| when(req.getReader()) |
| .thenReturn(new BufferedReader(new StringReader(event))); |
| dispatcher.postEvent(any(RefReplicationDoneEvent.class)); |
| syncEventsRestApiServlet.doPost(req, rsp); |
| verify(rsp).setStatus(SC_NO_CONTENT); |
| } |
| |
| @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}"; |
| when(req.getReader()) |
| .thenReturn(new BufferedReader(new StringReader(event))); |
| doThrow(new OrmException(ERR_MSG)).when(dispatcher) |
| .postEvent(any(RefReplicationDoneEvent.class)); |
| syncEventsRestApiServlet.doPost(req, rsp); |
| verify(rsp).sendError(SC_NOT_FOUND, "Change not found\n"); |
| } |
| |
| @Test |
| public void testDoPostBadRequest() throws Exception { |
| doThrow(new IOException(ERR_MSG)).when(req).getReader(); |
| syncEventsRestApiServlet.doPost(req, rsp); |
| verify(rsp).sendError(SC_BAD_REQUEST, ERR_MSG); |
| } |
| |
| @Test |
| public void testDoPostWrongMediaType() throws Exception { |
| when(req.getContentType()) |
| .thenReturn(MediaType.APPLICATION_XML_UTF_8.toString()); |
| syncEventsRestApiServlet.doPost(req, rsp); |
| verify(rsp).sendError(SC_UNSUPPORTED_MEDIA_TYPE, |
| "Expecting " + JSON_UTF_8.toString() + " content type"); |
| } |
| |
| @Test |
| public void testDoPostErrorWhileSendingErrorMessage() throws Exception { |
| doThrow(new IOException(ERR_MSG)).when(req).getReader(); |
| doThrow(new IOException("someOtherError")).when(rsp) |
| .sendError(SC_BAD_REQUEST, ERR_MSG); |
| syncEventsRestApiServlet.doPost(req, rsp); |
| } |
| |
| 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; |
| } |
| } |
| } |