PreExecutionInterceptorContextTest.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 PreExecutionInterceptorContextTest {

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

	@Test
	public void mustCollectCallTree() throws Exception {
		interceptorContext.shouldNotCollectCallTree("reasons");
		assertFalse(interceptorContext.isCollectCallTree());

		interceptorContext.mustCollectCallTree("reasons");
		assertTrue(interceptorContext.isCollectCallTree());

		interceptorContext.shouldNotCollectCallTree("reasons");
		assertTrue(interceptorContext.isCollectCallTree());
	}

	@Test
	public void shouldNotCollectCallTree() throws Exception {
		assertTrue(interceptorContext.isCollectCallTree());

		interceptorContext.shouldNotCollectCallTree("reasons");

		assertFalse(interceptorContext.isCollectCallTree());
	}

}