appc-config/appc-data-services/provider/src/test/java/org/onap/appc/data/services/node/MockDbServiceBuilder.java

   1 package org.onap.appc.data.services.node;
   2
   3 import static org.mockito.Matchers.any;
   4 import static org.mockito.Matchers.anyString;
   5 import static org.mockito.Matchers.eq;
   6 import static org.mockito.Mockito.doReturn;
   7 import static org.mockito.Mockito.mock;
   8 import org.onap.appc.data.services.db.DGGeneralDBService;
   9 import org.onap.ccsdk.sli.core.sli.SvcLogicContext;
  10 import org.onap.ccsdk.sli.core.sli.SvcLogicException;
  11 import org.onap.ccsdk.sli.core.sli.SvcLogicResource;
  12
  13 class MockDbServiceBuilder {
  14
  15     private final DGGeneralDBService dbServiceMock;
  16
  17     MockDbServiceBuilder() throws SvcLogicException {
  18         dbServiceMock = mock(DGGeneralDBService.class);
  19
  20         doReturn(SvcLogicResource.QueryStatus.SUCCESS)
  21             .when(dbServiceMock)
  22             .getConfigFileReferenceByFileTypeNVnfType(any(SvcLogicContext.class), anyString(), anyString());
  23     }
  24
  25     MockDbServiceBuilder configFileReference(String prefix, String fileType, SvcLogicResource.QueryStatus status) throws SvcLogicException {
  26         doReturn(status)
  27             .when(dbServiceMock)
  28             .getConfigFileReferenceByFileTypeNVnfType(any(SvcLogicContext.class), eq(prefix), eq(fileType));
  29
  30         return this;
  31     }
  32
  33     DGGeneralDBService build() {
  34         return dbServiceMock;
  35     }
  36 }