Skip to content

Commit

Permalink
Add test cases on ProjectionsToken
Browse files Browse the repository at this point in the history
  • Loading branch information
terrymanu committed Nov 16, 2024
1 parent fa4e76d commit e17795b
Show file tree
Hide file tree
Showing 2 changed files with 50 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,11 @@ public ProjectionsToken(final int startIndex, final Map<RouteUnit, Collection<St
this.projections = projections;
}

@Override
public int getStopIndex() {
return getStartIndex();
}

@Override
public String toString(final RouteUnit routeUnit) {
StringBuilder result = new StringBuilder();
Expand All @@ -46,9 +51,4 @@ public String toString(final RouteUnit routeUnit) {
}
return result.toString();
}

@Override
public int getStopIndex() {
return getStartIndex();
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
/*
* 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.shardingsphere.sharding.rewrite.token.pojo;

import org.apache.shardingsphere.infra.route.context.RouteMapper;
import org.apache.shardingsphere.infra.route.context.RouteUnit;
import org.junit.jupiter.api.Test;

import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.MatcherAssert.assertThat;

class ProjectionsTokenTest {

@Test
void assertGetStopIndex() {
assertThat(new ProjectionsToken(10, Collections.emptyMap()).getStopIndex(), is(10));
}

@Test
void assertToStringWithoutRouteUnit() {
Map<RouteUnit, Collection<String>> projections = new HashMap<>(1, 1);
RouteUnit routeUnit = new RouteUnit(new RouteMapper("foo_ds", "foo_ds_0"), Collections.singleton(new RouteMapper("foo_tbl", "foo_tbl_1")));
assertThat(new ProjectionsToken(0, Collections.singletonMap(routeUnit, Collections.singleton("foo_col"))).toString(routeUnit), is(", foo_col"));
}
}

0 comments on commit e17795b

Please sign in to comment.