PostExecutionInterceptorContextTest.java

package org.stagemonitor.tracing.sampling;

import org.junit.Test;
import org.stagemonitor.tracing.SpanContextInformation;

import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
import static org.mockito.Mockito.mock;

public class PostExecutionInterceptorContextTest {

	private PostExecutionInterceptorContext interceptorContext = new PostExecutionInterceptorContext(mock(SpanContextInformation.class));

	@Test
	public void excludeCallTree() throws Exception {
		assertFalse(interceptorContext.isExcludeCallTree());

		interceptorContext.excludeCallTree("reasons");

		assertTrue(interceptorContext.isExcludeCallTree());
	}

	@Test
	public void mustPreserveCallTree() throws Exception {
		interceptorContext.excludeCallTree("reasons");
		interceptorContext.mustPreserveCallTree("reasons");
		assertFalse(interceptorContext.isExcludeCallTree());

		interceptorContext.excludeCallTree("reasons");

		assertFalse(interceptorContext.isExcludeCallTree());
	}

}