VcInventorySyncServiceTest.java

/***************************************************************************
 * Copyright (c) 2015 VMware, Inc. All Rights Reserved.
 * 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.vmware.bdd.service.resmgmt.impl;

import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.databind.ObjectMapper;

import com.vmware.bdd.service.resmgmt.VC_RESOURCE_TYPE;
import com.vmware.bdd.service.resmgmt.sync.filter.VcResourceFilters;
import org.apache.log4j.Logger;
import org.testng.annotations.BeforeTest;
import org.testng.annotations.Test;

import java.io.IOException;
import java.util.concurrent.*;

/**
 * Created by xiaoliangl on 7/27/15.
 */
public class VcInventorySyncServiceTest extends BaseResourceTest {
   private final static Logger LOGGER = Logger.getLogger(VcInventorySyncServiceTest.class);

   private VcInventorySyncService syncService = new VcInventorySyncService();
   private VcInventorySyncCounters counters = new VcInventorySyncCounters();

   private boolean isVcInited = false;
   private boolean logStarted = false;

   @BeforeTest
   public void setupBeforeClass() throws InterruptedException {
      if (!isVcInited) {
         initVC();
      }

      Thread.sleep(60000);
   }

   private void logPerformance(final VcInventorySyncCounters counters) {
      if(logStarted) return;

      ScheduledExecutorService ses = Executors.newScheduledThreadPool(1);

      ses.scheduleAtFixedRate(new Runnable() {
         @Override
         public void run() {
            try {
               LOGGER.info("performance summary: " + convertObjectToJsonBytes(counters));
            } catch (IOException ex) {
               ex.printStackTrace();
               LOGGER.error(ex);
            }
         }
      }, 0, 5, TimeUnit.SECONDS);

      logStarted = true;
   }

   @Test
   public void testSync() throws InterruptedException, IOException {
      syncService.setCounters(counters);

      ExecutorService es = Executors.newFixedThreadPool(2);

      Runnable refresher = new Runnable() {
         @Override
         public void run() {
            try {
               syncService.refreshInventory();
            } catch (InterruptedException e) {
               e.printStackTrace();
            }
         }
      };
      Future<?> future1 = es.submit(refresher);

      logPerformance(counters);

      while (!future1.isDone()) {
         try {
            Thread.sleep(500);
         } catch (InterruptedException e) {
            e.printStackTrace();
         }
      }
   }

   @Test
   public void testSyncConcurrent() {
      syncService.setCounters(counters);

      ExecutorService es = Executors.newFixedThreadPool(2);

      Runnable refresher = new Runnable() {
         @Override
         public void run() {
            try {
               syncService.refreshInventory();
            } catch (InterruptedException e) {
               e.printStackTrace();
            }
         }
      };
      Future<?> future1 = es.submit(refresher);
      Future<?> future2 = es.submit(refresher);

      logPerformance(counters);

      while (!future1.isDone() || !future2.isDone()) {
         try {
            Thread.sleep(500);
         } catch (InterruptedException e) {
            e.printStackTrace();
         }
      }
   }

   @Test
   public void testSyncWithFilter() throws InterruptedException, IOException {
      syncService.setCounters(counters);

      ExecutorService es = Executors.newFixedThreadPool(2);

      Runnable refresher = new Runnable() {
         @Override
         public void run() {
            try {
               VcResourceFilters vcResourceFilters = new VcResourceFilters();
               vcResourceFilters = vcResourceFilters.addNameFilter(VC_RESOURCE_TYPE.DATA_STORE, new String[]{"DS_DC0_C0_H0_0",
                     "DS_DC0_C0_H0_1", "DS_DC0_C0_H0_2", "DS_DC0_C0_H0_3", "DS_DC0_C0_H0_4", "DS_DC0_C0_H0_5",
                     "DS_DC0_C0_H0_6", "DS_DC0_C0_H0_7", "DS_DC0_C0_H0_8", "DS_DC0_C0_H0_9"}, false);
               syncService.refreshInventory(vcResourceFilters);
            } catch (InterruptedException e) {
               e.printStackTrace();
            }
         }
      };
      Future<?> future1 = es.submit(refresher);

      logPerformance(counters);

      while (!future1.isDone()) {
         try {
            Thread.sleep(500);
         } catch (InterruptedException e) {
            e.printStackTrace();
         }
      }
   }

   public static String convertObjectToJsonBytes(Object object) throws IOException {
      ObjectMapper mapper = new ObjectMapper();
      mapper.setSerializationInclusion(JsonInclude.Include.NON_NULL);
      return mapper.writeValueAsString(object);
   }
}