Some code update and rearrangement
Signed-off-by: Eric Zhao <sczyh16@gmail.com>
This commit is contained in:
parent
62decf0464
commit
fe6b4f4865
|
|
@ -47,4 +47,12 @@ public class StringResourceWrapper extends ResourceWrapper {
|
|||
public EntryType getType() {
|
||||
return type;
|
||||
}
|
||||
|
||||
@Override
|
||||
public String toString() {
|
||||
return "StringResourceWrapper{" +
|
||||
"name='" + name + '\'' +
|
||||
", type=" + type +
|
||||
'}';
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -20,9 +20,9 @@ import java.util.concurrent.ExecutorService;
|
|||
import java.util.concurrent.Executors;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
import com.alibaba.csp.sentinel.context.ContextTestUtil;
|
||||
import com.alibaba.csp.sentinel.context.ContextUtil;
|
||||
import com.alibaba.csp.sentinel.node.DefaultNode;
|
||||
import com.alibaba.csp.sentinel.node.EntranceNode;
|
||||
import com.alibaba.csp.sentinel.node.Node;
|
||||
import com.alibaba.csp.sentinel.slots.block.BlockException;
|
||||
|
||||
|
|
@ -41,10 +41,7 @@ public class AsyncEntryIntegrationTest {
|
|||
|
||||
@Before
|
||||
public void clearContext() {
|
||||
if (ContextUtil.getContext() != null) {
|
||||
ContextUtil.getContext().setCurEntry(null);
|
||||
ContextUtil.exit();
|
||||
}
|
||||
ContextTestUtil.cleanUpContext();
|
||||
}
|
||||
|
||||
private final ExecutorService pool = Executors.newFixedThreadPool(10);
|
||||
|
|
|
|||
|
|
@ -13,15 +13,18 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package com.alibaba.csp.sentinel;
|
||||
package com.alibaba.csp.sentinel.context;
|
||||
|
||||
import com.alibaba.csp.sentinel.context.Context;
|
||||
import com.alibaba.csp.sentinel.context.ContextUtil;
|
||||
import com.alibaba.csp.sentinel.Constants;
|
||||
|
||||
/**
|
||||
* Util class for testing context-related functions.
|
||||
* Only for test. DO NOT USE IN PRODUCTION!
|
||||
*
|
||||
* @author Eric Zhao
|
||||
* @since 0.2.0
|
||||
*/
|
||||
public final class TestUtil {
|
||||
public final class ContextTestUtil {
|
||||
|
||||
public static void cleanUpContext() {
|
||||
Context context = ContextUtil.getContext();
|
||||
|
|
@ -31,5 +34,10 @@ public final class TestUtil {
|
|||
}
|
||||
}
|
||||
|
||||
private TestUtil() {}
|
||||
public static void resetContextMap() {
|
||||
ContextUtil.resetContextMap();
|
||||
Constants.ROOT.removeChildList();
|
||||
}
|
||||
|
||||
private ContextTestUtil() {}
|
||||
}
|
||||
|
|
@ -1,6 +1,6 @@
|
|||
package com.alibaba.csp.sentinel.slots.block.authority;
|
||||
|
||||
import com.alibaba.csp.sentinel.TestUtil;
|
||||
import com.alibaba.csp.sentinel.context.ContextTestUtil;
|
||||
import com.alibaba.csp.sentinel.context.ContextUtil;
|
||||
import com.alibaba.csp.sentinel.slots.block.RuleConstant;
|
||||
|
||||
|
|
@ -18,7 +18,7 @@ public class AuthorityRuleCheckerTest {
|
|||
|
||||
@Before
|
||||
public void setUp() {
|
||||
TestUtil.cleanUpContext();
|
||||
ContextTestUtil.cleanUpContext();
|
||||
}
|
||||
|
||||
@Test
|
||||
|
|
|
|||
|
|
@ -3,7 +3,7 @@ package com.alibaba.csp.sentinel.slots.block.authority;
|
|||
import java.util.Collections;
|
||||
|
||||
import com.alibaba.csp.sentinel.EntryType;
|
||||
import com.alibaba.csp.sentinel.TestUtil;
|
||||
import com.alibaba.csp.sentinel.context.ContextTestUtil;
|
||||
import com.alibaba.csp.sentinel.context.ContextUtil;
|
||||
import com.alibaba.csp.sentinel.slotchain.ResourceWrapper;
|
||||
import com.alibaba.csp.sentinel.slotchain.StringResourceWrapper;
|
||||
|
|
@ -120,13 +120,13 @@ public class AuthoritySlotTest {
|
|||
|
||||
@Before
|
||||
public void setUp() {
|
||||
TestUtil.cleanUpContext();
|
||||
ContextTestUtil.cleanUpContext();
|
||||
AuthorityRuleManager.loadRules(null);
|
||||
}
|
||||
|
||||
@After
|
||||
public void tearDown() {
|
||||
TestUtil.cleanUpContext();
|
||||
ContextTestUtil.cleanUpContext();
|
||||
AuthorityRuleManager.loadRules(null);
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue