code: /*******************************************************************************
   * Copyright 2018 572682
   * 
   * 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 us.dot.its.jpo.ode.dds;
  
  import static org.junit.Assert.assertEquals;
  
  import org.junit.Test;
  
  import mockit.Expectations;
  import mockit.Mocked;
  import us.dot.its.jpo.ode.dds.DdsRequest.SystemName;
  import us.dot.its.jpo.ode.model.OdeRequest;
  import us.dot.its.jpo.ode.model.OdeRequest.DataSource;
  import us.dot.its.jpo.ode.model.OdeRequestType;
  
  /**
   * Test set for DdsRequestManager static methods
   *
   */
  public class DdsRequestManagerTest {
  
      /**
       * Verify correct system name from known deposit data sources
       */
      @Test
      public void testExpectedSystemNames(@Mocked OdeRequest mockOdeRequest) {
  
          // Test 1: SDC --> SDC
          new Expectations() {
              {
                  mockOdeRequest.getDataSource();
                  result = DataSource.SDC;
              }
          };
  
          assertEquals("Expected SDC --> SDC: ", SystemName.SDC, DdsRequestManager.systemName(mockOdeRequest));
  
          // Test 2: DEPOSIT_SDC --> SDC
          new Expectations() {
              {
                  mockOdeRequest.getDataSource();
                  result = DataSource.DEPOSIT_SDC;
              }
          };
          assertEquals("Expected DEPOSIT_SDC --> SDC: ", SystemName.SDC, DdsRequestManager.systemName(mockOdeRequest));
  
          // Test 3: SDW --> SDW
          new Expectations() {
              {
                  mockOdeRequest.getDataSource();
                  result = DataSource.SDW;
              }
          };
          assertEquals("Expected SDW --> SDW: ", SystemName.SDW, DdsRequestManager.systemName(mockOdeRequest));
  
          // Test 4: DEPOSIT_SDW --> SDW
          new Expectations() {
              {
                  mockOdeRequest.getDataSource();
                  result = DataSource.DEPOSIT_SDW;
              }
          };
          assertEquals("Expected DEPOSIT_SDW --> SDW: ", SystemName.SDW, DdsRequestManager.systemName(mockOdeRequest));
  
          // Test 5: SDPC --> SDPC
          new Expectations() {
              {
                  mockOdeRequest.getDataSource();
                  result = DataSource.SDPC;
              }
          };
          assertEquals("Expected SDPC --> SDPC: ", SystemName.SDPC, DdsRequestManager.systemName(mockOdeRequest));
      }
  
      /**
       * Verify correct defaulting behavior from unknown deposit data sources
       */
      @Test
      public void testUnexpectedSystemNames(@Mocked OdeRequest mockOdeRequest) {
  
          // Test 1: Subscription --> SDC
          new Expectations() {
              {
                  mockOdeRequest.getDataSource();
                  result = DataSource.TEST_UPLOAD;
                  mockOdeRequest.getRequestType();
                  result = OdeRequestType.Subscription;
              }
          };
  
          assertEquals("Expected OdeRequestType.Subscription --> SDC: ", SystemName.SDC,
                  DdsRequestManager.systemName(mockOdeRequest));
  
          // Test 2: Query --> SDW
          new Expectations() {
              {
                  mockOdeRequest.getDataSource();
                  result = DataSource.TEST_UPLOAD;
                  mockOdeRequest.getRequestType();
                  result = OdeRequestType.Query;
              }
          };
          assertEquals("Expected OdeRequestType.Query --> SDW: ", SystemName.SDW,
                  DdsRequestManager.systemName(mockOdeRequest));
      }
      
      /**
       * Repeat above test but test default data source method
       */
      @Test
      public void testSystemNameWithNullDataSource(@Mocked OdeRequest mockOdeRequest) {
          
          // Test 1: Subscription --> SDC
          new Expectations() {
              {
                  mockOdeRequest.getDataSource();
                  result = null;
                  mockOdeRequest.getRequestType();
                  result = OdeRequestType.Subscription;
              }
          };
  
          assertEquals("Expected OdeRequestType.Subscription --> SDC: ", SystemName.SDC,
                  DdsRequestManager.systemName(mockOdeRequest));
  
          // Test 2: Query --> SDW
          new Expectations() {
              {
                  mockOdeRequest.getDataSource();
                  result = null;
                  mockOdeRequest.getRequestType();
                  result = OdeRequestType.Query;
              }
          };
          assertEquals("Expected OdeRequestType.Query --> SDW: ", SystemName.SDW,
                  DdsRequestManager.systemName(mockOdeRequest));
          
      }
  
  }
  