|
36 | 36 | import com.contrastsecurity.models.Rules; |
37 | 37 | import com.contrastsecurity.models.SecurityCheck; |
38 | 38 | import com.contrastsecurity.models.Servers; |
| 39 | +import com.contrastsecurity.models.SessionMetadata; |
39 | 40 | import com.contrastsecurity.models.Story; |
40 | 41 | import com.contrastsecurity.models.StoryResponse; |
41 | 42 | import com.contrastsecurity.models.Tags; |
@@ -351,21 +352,17 @@ public void testGetTracesWithSessionMetadata() { |
351 | 352 | assertThat(trace.getSessionMetadata()).isNotNull(); |
352 | 353 | assertThat(trace.getSessionMetadata()).hasSize(1); |
353 | 354 |
|
354 | | - assertThat(trace.getSessionMetadata().get(0).getSessionId()).isEqualTo("session-123"); |
355 | | - assertThat(trace.getSessionMetadata().get(0).getMetadata()).isNotNull(); |
356 | | - assertThat(trace.getSessionMetadata().get(0).getMetadata()).hasSize(2); |
357 | | - |
358 | | - assertThat(trace.getSessionMetadata().get(0).getMetadata().get(0).getValue()).isEqualTo("prod"); |
359 | | - assertThat(trace.getSessionMetadata().get(0).getMetadata().get(0).getDisplayLabel()) |
360 | | - .isEqualTo("Environment"); |
361 | | - assertThat(trace.getSessionMetadata().get(0).getMetadata().get(0).getAgentLabel()) |
362 | | - .isEqualTo("env"); |
363 | | - |
364 | | - assertThat(trace.getSessionMetadata().get(0).getMetadata().get(1).getValue()) |
365 | | - .isEqualTo("user123"); |
366 | | - assertThat(trace.getSessionMetadata().get(0).getMetadata().get(1).getDisplayLabel()) |
367 | | - .isEqualTo("User ID"); |
368 | | - assertThat(trace.getSessionMetadata().get(0).getMetadata().get(1).getAgentLabel()) |
369 | | - .isEqualTo("user_id"); |
| 355 | + SessionMetadata sessionMetadata = trace.getSessionMetadata().get(0); |
| 356 | + assertThat(sessionMetadata.getSessionId()).isEqualTo("session-123"); |
| 357 | + assertThat(sessionMetadata.getMetadata()).isNotNull(); |
| 358 | + assertThat(sessionMetadata.getMetadata()).hasSize(2); |
| 359 | + |
| 360 | + assertThat(sessionMetadata.getMetadata().get(0).getValue()).isEqualTo("prod"); |
| 361 | + assertThat(sessionMetadata.getMetadata().get(0).getDisplayLabel()).isEqualTo("Environment"); |
| 362 | + assertThat(sessionMetadata.getMetadata().get(0).getAgentLabel()).isEqualTo("env"); |
| 363 | + |
| 364 | + assertThat(sessionMetadata.getMetadata().get(1).getValue()).isEqualTo("user123"); |
| 365 | + assertThat(sessionMetadata.getMetadata().get(1).getDisplayLabel()).isEqualTo("User ID"); |
| 366 | + assertThat(sessionMetadata.getMetadata().get(1).getAgentLabel()).isEqualTo("user_id"); |
370 | 367 | } |
371 | 368 | } |
0 commit comments