rulemgt/src/test/java/org/onap/holmes/rulemgt/InitializerTest.java

   1 /**
   2  * Copyright 2020 ZTE Corporation.
   3  * <p>
   4  * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
   5  * in compliance with the License. You may obtain a copy of the License at
   6  * <p>
   7  * http://www.apache.org/licenses/LICENSE-2.0
   8  * <p>
   9  * Unless required by applicable law or agreed to in writing, software distributed under the License
  10  * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
  11  * or implied. See the License for the specific language governing permissions and limitations under
  12  * the License.
  13  */
  14
  15 package org.onap.holmes.rulemgt;
  16
  17 import org.easymock.EasyMock;
  18 import org.junit.Test;
  19 import org.junit.runner.RunWith;
  20 import org.onap.holmes.common.config.MicroServiceConfig;
  21 import org.onap.holmes.common.utils.MsbRegister;
  22 import org.onap.msb.sdk.discovery.entity.MicroServiceInfo;
  23 import org.powermock.api.easymock.PowerMock;
  24 import org.powermock.core.classloader.annotations.PrepareForTest;
  25 import org.powermock.core.classloader.annotations.SuppressStaticInitializationFor;
  26 import org.powermock.modules.junit4.PowerMockRunner;
  27 import org.powermock.reflect.internal.WhiteboxImpl;
  28
  29 import java.util.concurrent.TimeUnit;
  30
  31 @RunWith(PowerMockRunner.class)
  32 @PrepareForTest(MicroServiceConfig.class)
  33 @SuppressStaticInitializationFor("org.onap.holmes.common.utils.CommonUtils")
  34 public class InitializerTest {
  35
  36     @Test
  37     public void process() throws Exception {
  38         MsbRegister mockedMsbRegister = PowerMock.createMock(MsbRegister.class);
  39         Initializer initializer = new Initializer(mockedMsbRegister);
  40
  41         PowerMock.mockStaticPartial(MicroServiceConfig.class, "getMicroServiceIpAndPort", "getEnv");
  42         EasyMock.expect(MicroServiceConfig.getMicroServiceIpAndPort()).andReturn(new String[]{"127.0.0.1", "443"});
  43         System.setProperty("ENABLE_ENCRYPT", "true");
  44
  45         mockedMsbRegister.register2Msb(EasyMock.anyObject(MicroServiceInfo.class));
  46         EasyMock.expectLastCall();
  47
  48         PowerMock.replayAll();
  49
  50         setReadyFlagAfter(3);
  51
  52         WhiteboxImpl.invokeMethod(initializer, "init");
  53
  54         TimeUnit.SECONDS.sleep(6);
  55
  56         PowerMock.verifyAll();
  57     }
  58
  59     private void setReadyFlagAfter(final int second) {
  60         new Thread(() -> {
  61             try {
  62                 TimeUnit.SECONDS.sleep(second);
  63             } catch (InterruptedException e) {
  64                 e.printStackTrace();
  65             }
  66             Initializer.setReadyForMsbReg(true);
  67         }).start();
  68     }
  69 }