Skip to content

Commit 8638748

Browse files
ChrisEdwardsclaude
andcommitted
Refactor test to extract sessionMetadata variable for readability
🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
1 parent de1cf64 commit 8638748

File tree

1 file changed

+13
-16
lines changed

1 file changed

+13
-16
lines changed

sdk/src/test/java/com/contrastsecurity/GsonTest.java

Lines changed: 13 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@
3636
import com.contrastsecurity.models.Rules;
3737
import com.contrastsecurity.models.SecurityCheck;
3838
import com.contrastsecurity.models.Servers;
39+
import com.contrastsecurity.models.SessionMetadata;
3940
import com.contrastsecurity.models.Story;
4041
import com.contrastsecurity.models.StoryResponse;
4142
import com.contrastsecurity.models.Tags;
@@ -351,21 +352,17 @@ public void testGetTracesWithSessionMetadata() {
351352
assertThat(trace.getSessionMetadata()).isNotNull();
352353
assertThat(trace.getSessionMetadata()).hasSize(1);
353354

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");
370367
}
371368
}

0 commit comments

Comments
 (0)