Skip to content

Commit

Permalink
fix test
Browse files Browse the repository at this point in the history
  • Loading branch information
@EddeCCC
EddeCCC committed Sep 15, 2023
1 parent 142ecb4 commit 072ff20
Showing 1 changed file with 17 additions and 11 deletions.
28 changes: 17 additions & 11 deletions ...ocks/inspectit/ocelot/core/instrumentation/browser/BrowserPropagationDataStorageTest.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,9 @@
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import org.springframework.test.annotation.DirtiesContext;
import org.springframework.test.context.TestPropertySource;
import rocks.inspectit.ocelot.core.SpringTestBase;
import rocks.inspectit.ocelot.core.instrumentation.config.model.propagation.PropagationMetaData;
import rocks.inspectit.ocelot.core.instrumentation.context.ContextUtil;
import rocks.inspectit.ocelot.core.instrumentation.context.InspectitContextImpl;
Expand All @@ -18,8 +21,10 @@
import static org.mockito.ArgumentMatchers.*;
import static org.mockito.Mockito.when;

@TestPropertySource(properties = {"inspectit.exporters.tags.http.enabled=true", "inspectit.exporters.tags.http.session-id-header=SessionKey"})
@DirtiesContext
@ExtendWith(MockitoExtension.class)
public class BrowserPropagationDataStorageTest {
public class BrowserPropagationDataStorageTest extends SpringTestBase {

@Mock
PropagationMetaData propagation;
Expand All @@ -28,15 +33,16 @@ public class BrowserPropagationDataStorageTest {

Map<String, String> headers;

private static final String sessionID = "test=83311527d6a6de76a60a72a041808a63;b0b2b4cf=ad9fef38-4942-453a-9243-7d8422803604";
private static final String sessionIdHeader = "SessionKey";
private static final String sessionId = "test=83311527d6a6de76a60a72a041808a63;b0b2b4cf=ad9fef38-4942-453a-9243-7d8422803604";

@BeforeEach
void prepareTest() {
// Create session storage to store BrowserPropagationDataStorages
sessionStorage = BrowserPropagationSessionStorage.getInstance();
// Create HTTP header to pass it to the initial InspectIT-Context
headers = new HashMap<>();
headers.put("cookie", sessionID);
headers.put(sessionIdHeader, sessionId);
}

@AfterEach
Expand All @@ -50,7 +56,7 @@ public class WriteBrowserPropagationData {
@Test
void verifyNoDataHasBeenWritten() {
when(propagation.isPropagatedWithBrowser(any())).thenReturn(false);
BrowserPropagationDataStorage dataStorage = sessionStorage.getOrCreateDataStorage(sessionID);
BrowserPropagationDataStorage dataStorage = sessionStorage.getOrCreateDataStorage(sessionId);
InspectitContextImpl ctx = InspectitContextImpl.createFromCurrent(Collections.emptyMap(), propagation, false);
ctx.readDownPropagationHeaders(headers);
ctx.makeActive();
Expand All @@ -67,7 +73,7 @@ void verifyNoDataHasBeenWritten() {
void verifyDataHasBeenWritten() {
when(propagation.isPropagatedWithBrowser(anyString())).thenReturn(false);
when(propagation.isPropagatedWithBrowser(eq("keyA"))).thenReturn(true);
BrowserPropagationDataStorage dataStorage = sessionStorage.getOrCreateDataStorage(sessionID);
BrowserPropagationDataStorage dataStorage = sessionStorage.getOrCreateDataStorage(sessionId);
InspectitContextImpl ctx = InspectitContextImpl.createFromCurrent(Collections.emptyMap(), propagation, false);
ctx.readDownPropagationHeaders(headers);
ctx.makeActive();
Expand All @@ -87,7 +93,7 @@ void verifyDataHasBeenWritten() {
void verifyDataHasBeenOverwritten() {
when(propagation.isPropagatedWithBrowser(any())).thenReturn(true);
when(propagation.isPropagatedDownWithinJVM(any())).thenReturn(true);
BrowserPropagationDataStorage dataStorage = sessionStorage.getOrCreateDataStorage(sessionID);
BrowserPropagationDataStorage dataStorage = sessionStorage.getOrCreateDataStorage(sessionId);
Map<String, Object> oldData = new HashMap<>();
oldData.put("keyA", "value0");
dataStorage.writeData(oldData);
Expand All @@ -114,7 +120,7 @@ void verifyDataHasBeenOverwritten() {
@Test
void verifyAttributeCountLimit() {
when(propagation.isPropagatedWithBrowser(any())).thenReturn(true);
BrowserPropagationDataStorage dataStorage = sessionStorage.getOrCreateDataStorage(sessionID);
BrowserPropagationDataStorage dataStorage = sessionStorage.getOrCreateDataStorage(sessionId);
Map<String, Object> dummyMap = IntStream.rangeClosed(1, 128).boxed()
.collect(Collectors.toMap(i -> "key"+i, i -> "value"+i));
dataStorage.writeData(dummyMap);
Expand All @@ -136,7 +142,7 @@ void verifyAttributeCountLimit() {
@Test
void verifyValidEntries() {
when(propagation.isPropagatedWithBrowser(any())).thenReturn(true);
BrowserPropagationDataStorage dataStorage = sessionStorage.getOrCreateDataStorage(sessionID);
BrowserPropagationDataStorage dataStorage = sessionStorage.getOrCreateDataStorage(sessionId);
InspectitContextImpl ctx = InspectitContextImpl.createFromCurrent(Collections.emptyMap(), propagation, false);
ctx.readDownPropagationHeaders(headers);
ctx.makeActive();
Expand All @@ -161,7 +167,7 @@ void verifyNoDownPropagation() {
when(propagation.isPropagatedWithBrowser(any())).thenReturn(true);
when(propagation.isPropagatedDownWithinJVM(any())).thenReturn(false);
when(propagation.isPropagatedDownWithinJVM(eq("remote_session_id"))).thenReturn(true);
BrowserPropagationDataStorage dataStorage = sessionStorage.getOrCreateDataStorage(sessionID);
BrowserPropagationDataStorage dataStorage = sessionStorage.getOrCreateDataStorage(sessionId);
Map<String, Object> data = new HashMap<>();
data.put("keyA", "valueA");
dataStorage.writeData(data);
Expand All @@ -186,7 +192,7 @@ void verifyNoDownPropagation() {
void verifyDownPropagation() {
when(propagation.isPropagatedWithBrowser(any())).thenReturn(true);
when(propagation.isPropagatedDownWithinJVM(any())).thenReturn(true);
BrowserPropagationDataStorage dataStorage = sessionStorage.getOrCreateDataStorage(sessionID);
BrowserPropagationDataStorage dataStorage = sessionStorage.getOrCreateDataStorage(sessionId);
InspectitContextImpl ctxA = InspectitContextImpl.createFromCurrent(Collections.emptyMap(), propagation, false);
Map<String, Object> data = new HashMap<>();
data.put("keyA", "valueA");
Expand Down Expand Up @@ -219,7 +225,7 @@ void verifyUpPropagation() {
when(propagation.isPropagatedWithBrowser(any())).thenReturn(true);
when(propagation.isPropagatedDownWithinJVM(any())).thenReturn(true);
when(propagation.isPropagatedUpWithinJVM(eq("keyB"))).thenReturn(true);
BrowserPropagationDataStorage dataStorage = sessionStorage.getOrCreateDataStorage(sessionID);
BrowserPropagationDataStorage dataStorage = sessionStorage.getOrCreateDataStorage(sessionId);
InspectitContextImpl ctxA = InspectitContextImpl.createFromCurrent(Collections.emptyMap(), propagation, false);
Map<String, Object> data = new HashMap<>();
data.put("keyA", "valueA");
Expand Down

0 comments on commit 072ff20

Please sign in to comment.