engine-d/src/test/java/org/openo/holmes/engine/db/CorrelationRuleDaoTest.java

   1 /**\r
   2  * Copyright 2017 ZTE Corporation.\r
   3  *\r
   4  * Licensed under the Apache License, Version 2.0 (the "License");\r
   5  * you may not use this file except in compliance with the License.\r
   6  * You may obtain a copy of the License at\r
   7  *\r
   8  * http://www.apache.org/licenses/LICENSE-2.0\r
   9  *\r
  10  * Unless required by applicable law or agreed to in writing, software\r
  11  * distributed under the License is distributed on an "AS IS" BASIS,\r
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  13  * See the License for the specific language governing permissions and\r
  14  * limitations under the License.\r
  15  */\r
  16 package org.openo.holmes.engine.db;\r
  17 \r
  18 import org.easymock.EasyMock;\r
  19 import org.hamcrest.core.IsNull;\r
  20 import org.junit.Assert;\r
  21 import org.junit.Before;\r
  22 import org.junit.Test;\r
  23 import org.openo.holmes.common.api.entity.CorrelationRule;\r
  24 import org.powermock.api.easymock.PowerMock;\r
  25 \r
  26 import java.util.ArrayList;\r
  27 import java.util.List;\r
  28 \r
  29 /**\r
  30  * Created by Administrator on 2017/2/20.\r
  31  */\r
  32 public class CorrelationRuleDaoTest {\r
  33     private CorrelationRuleDao correlationRuleDao;\r
  34 \r
  35     @Before\r
  36     public void setUp() {\r
  37         correlationRuleDao = PowerMock.createMock(CorrelationRuleDao.class);\r
  38     }\r
  39 \r
  40     @Test\r
  41     public void queryRuleByEnable() {\r
  42         int enable = 0;\r
  43         EasyMock.expect(correlationRuleDao.queryRuleByEnable(EasyMock.anyInt())).andReturn(new ArrayList<CorrelationRule>());\r
  44         PowerMock.replayAll();\r
  45         List<CorrelationRule> correlationRules = correlationRuleDao.queryRuleByEnable(enable);\r
  46         PowerMock.verifyAll();\r
  47         Assert.assertThat(correlationRules, IsNull.<List<CorrelationRule>>notNullValue());\r
  48     }\r
  49 \r
  50 }\r