DeployProjectTest.java

/*
 * Copyright 2016 LinkedIn Corp.
 *
 * 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 azkaban.restli;

import static org.junit.Assert.assertEquals;

import azkaban.project.validator.ValidationReport;
import com.linkedin.restli.common.HttpStatus;
import com.linkedin.restli.server.RestLiServiceException;
import java.util.HashSet;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Set;
import org.junit.Assert;
import org.junit.Test;

/**
 * Test response to deploy with either warn or error reports. If a report has any errors, a
 * RestLiServiceException should be thrown. Warnings should not elicit an exception. If an exception
 * does get raised, it should carry an HTTP 400 status
 */

public class DeployProjectTest {

  @Test
  public void testWarnedDeploy() {
    final ProjectManagerResource resource = new ProjectManagerResource();
    final Map<String, ValidationReport> reports = new LinkedHashMap<>();
    for (int i = 0; i < 3; i++) {
      addMockWarning(reports, "Test warn level info message.");
    }
    // We expect that no exceptions are thrown given a
    // report with warnings. Anything thrown will result in failure
    resource.checkReports(reports);
  }

  @Test
  public void testErrorDeploy() {
    final ProjectManagerResource resource = new ProjectManagerResource();
    final Map<String, ValidationReport> reports = new LinkedHashMap<>();
    for (int i = 0; i < 3; i++) {
      addMockError(reports, "Test error level info message.");
    }
    // We expect that a RestLiServiceException is thrown given a
    // report with errors. Uncaught exceptions will result in failure
    try {
      resource.checkReports(reports);
      Assert.fail();
    } catch (final RestLiServiceException e) {
      //Ensure we have the right status code and exit
      assertEquals(e.getStatus(), HttpStatus.S_400_BAD_REQUEST);
    }
  }

  @Test
  public void testWarnErrorDeploy() {
    final ProjectManagerResource resource = new ProjectManagerResource();
    final Map<String, ValidationReport> reports = new LinkedHashMap<>();
    for (int i = 0; i < 7; i++) {
      // If i is even, make an error report, otherwise make a warning report
      if (i % 2 == 0) {
        addMockError(reports, "Test error level info message.");
      } else {
        addMockWarning(reports, "Test warn level info message.");
      }
    }

    // We expect that a RestLiServiceException is thrown given a
    // report with errors. Uncaught exceptions will result in failure
    try {
      resource.checkReports(reports);
      Assert.fail();
    } catch (final RestLiServiceException e) {
      //Ensure we have the right status code and exit
      assertEquals(e.getStatus(), HttpStatus.S_400_BAD_REQUEST);
    }
  }

  /**
   * Test that an error message is attached to the exception on an error
   */
  @Test
  public void testErrorMessageDeploy() {
    final ProjectManagerResource resource = new ProjectManagerResource();
    final Map<String, ValidationReport> reports = new LinkedHashMap<>();

    addMockError(reports, "This should show up.");

    // We expect that a RestLiServiceException is thrown given a
    // report with errors. Uncaught exceptions will result in failure
    try {
      resource.checkReports(reports);
      Assert.fail();
    } catch (final RestLiServiceException e) {
      //Ensure we have the right status code and exit
      assertEquals(e.getStatus(), HttpStatus.S_400_BAD_REQUEST);
      assertEquals(e.getMessage(), "Validator Error reports errors: This should show up."
          + System.getProperty("line.separator"));
    }
  }

  private void addMockWarning(final Map<String, ValidationReport> reports, final String msg) {
    final Set<String> warnMsgs = new HashSet<>();
    final ValidationReport warnRpt = new ValidationReport();
    warnMsgs.add(msg);
    warnRpt.addWarningMsgs(warnMsgs);
    reports.put("Warn", warnRpt);
  }

  private void addMockError(final Map<String, ValidationReport> reports, final String msg) {
    final Set<String> errorMsgs = new HashSet<>();
    final ValidationReport errorRpt = new ValidationReport();
    errorMsgs.add(msg);
    errorRpt.addErrorMsgs(errorMsgs);
    reports.put("Error", errorRpt);
  }
}