Skip to content

Commit

Permalink
Add MatchStrategyFactory Test (apache#1513)
Browse files Browse the repository at this point in the history
  • Loading branch information
ttttangzhen authored May 22, 2021
1 parent 22481f1 commit 1c48831
Showing 1 changed file with 78 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,78 @@
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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 org.apache.shenyu.plugin.base.condition.strategy;

import com.google.common.collect.Lists;
import org.apache.shenyu.common.dto.ConditionData;
import org.apache.shenyu.common.enums.MatchModeEnum;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.springframework.mock.http.server.reactive.MockServerHttpRequest;
import org.springframework.mock.web.server.MockServerWebExchange;
import org.springframework.web.server.ServerWebExchange;

import java.util.List;

/**
* Test cases for MatchStrategyFactory.
*/
public final class MatchStrategyFactoryTest {

private ServerWebExchange exchange;

private List<ConditionData> conditionDataList;

@Before
public void setUp() {
this.conditionDataList = Lists.newArrayListWithCapacity(2);
ConditionData matchConditionData = new ConditionData();
matchConditionData.setOperator("match");
matchConditionData.setParamName("shenyu");
matchConditionData.setParamType("uri");
matchConditionData.setParamValue("/http/**");
ConditionData eqConditionData = new ConditionData();
eqConditionData.setOperator("=");
eqConditionData.setParamName("shenyu");
eqConditionData.setParamType("uri");
eqConditionData.setParamValue("/http/test");
conditionDataList.add(matchConditionData);
conditionDataList.add(eqConditionData);
this.exchange = MockServerWebExchange.from(MockServerHttpRequest.get("/http/shenyu")
.build());
}

@Test
public void testNewInstance() {
MatchStrategy andMatchStrategy = MatchStrategyFactory.newInstance(MatchModeEnum.AND.getCode());
Assert.assertEquals(andMatchStrategy.getClass(), AndMatchStrategy.class);

MatchStrategy orMatchStrategy = MatchStrategyFactory.newInstance(MatchModeEnum.OR.getCode());
Assert.assertEquals(orMatchStrategy.getClass(), OrMatchStrategy.class);

int nonExistCode = -1;
MatchStrategy defaultMatchStrategy = MatchStrategyFactory.newInstance(nonExistCode);
Assert.assertEquals(defaultMatchStrategy.getClass(), AndMatchStrategy.class);
}

@Test
public void testMatch() {
Assert.assertFalse(MatchStrategyFactory.match(MatchModeEnum.AND.getCode(), conditionDataList, exchange));
Assert.assertTrue(MatchStrategyFactory.match(MatchModeEnum.OR.getCode(), conditionDataList, exchange));
}
}

0 comments on commit 1c48831

Please sign in to comment.