copilot-sdk/java/src/test/java/com/github/copilot/CapiSessionOptionsTest.java at main · svandor/copilot-sdk · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *--------------------------------------------------------------------------------------------*/

package com.github.copilot;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertNull;
import static org.junit.jupiter.api.Assertions.assertSame;
import static org.junit.jupiter.api.Assertions.assertTrue;

import org.junit.jupiter.api.Test;

import com.fasterxml.jackson.databind.JsonNode;

import com.github.copilot.rpc.CapiSessionOptions;
import com.github.copilot.rpc.ResumeSessionConfig;
import com.github.copilot.rpc.SessionConfig;

/**
 * Tests for CAPI provider-scoped session options.
 */
class CapiSessionOptionsTest {

    @Test
    void defaultsAreNull() {
        var capi = new CapiSessionOptions();

        assertNull(capi.getEnableWebSocketResponses());
    }

    @Test
    void fluentSetterReturnsSameInstance() {
        var capi = new CapiSessionOptions();

        assertSame(capi, capi.setEnableWebSocketResponses(true));
        assertEquals(Boolean.TRUE, capi.getEnableWebSocketResponses());
    }

    @Test
    void serializesEnableWebSocketResponses() {
        var capi = new CapiSessionOptions().setEnableWebSocketResponses(true);

        JsonNode json = JsonRpcClient.getObjectMapper().valueToTree(capi);

        assertTrue(json.get("enableWebSocketResponses").asBoolean());
    }

    @Test
    void omitsUnsetEnableWebSocketResponses() {
        var capi = new CapiSessionOptions();

        JsonNode json = JsonRpcClient.getObjectMapper().valueToTree(capi);

        assertTrue(json.path("enableWebSocketResponses").isMissingNode());
        assertEquals(0, json.size());
    }

    @Test
    void createRequestIncludesCapiWhenSet() {
        var config = new SessionConfig().setCapi(new CapiSessionOptions().setEnableWebSocketResponses(true));

        var request = SessionRequestBuilder.buildCreateRequest(config, "session-1");
        JsonNode json = JsonRpcClient.getObjectMapper().valueToTree(request);

        assertNotNull(request.getCapi());
        assertTrue(json.get("capi").get("enableWebSocketResponses").asBoolean());
    }

    @Test
    void createRequestOmitsCapiWhenUnset() {
        var config = new SessionConfig();

        var request = SessionRequestBuilder.buildCreateRequest(config, "session-1");
        JsonNode json = JsonRpcClient.getObjectMapper().valueToTree(request);

        assertNull(request.getCapi());
        assertTrue(json.path("capi").isMissingNode());
    }

    @Test
    void resumeRequestIncludesCapiWhenSet() {
        var config = new ResumeSessionConfig().setCapi(new CapiSessionOptions().setEnableWebSocketResponses(true));

        var request = SessionRequestBuilder.buildResumeRequest("session-1", config);
        JsonNode json = JsonRpcClient.getObjectMapper().valueToTree(request);

        assertNotNull(request.getCapi());
        assertTrue(json.get("capi").get("enableWebSocketResponses").asBoolean());
    }

    @Test
    void resumeRequestOmitsCapiWhenUnset() {
        var config = new ResumeSessionConfig();

        var request = SessionRequestBuilder.buildResumeRequest("session-1", config);
        JsonNode json = JsonRpcClient.getObjectMapper().valueToTree(request);

        assertNull(request.getCapi());
        assertTrue(json.path("capi").isMissingNode());
    }

    @Test
    void sessionConfigCloneCopiesCapiReference() {
        var capi = new CapiSessionOptions().setEnableWebSocketResponses(true);

        var clone = new SessionConfig().setCapi(capi).clone();

        assertSame(capi, clone.getCapi());
    }

    @Test
    void resumeSessionConfigCloneCopiesCapiReference() {
        var capi = new CapiSessionOptions().setEnableWebSocketResponses(true);

        var clone = new ResumeSessionConfig().setCapi(capi).clone();

        assertSame(capi, clone.getCapi());
    }

    @Test
    void falseValueIsSerializedWhenExplicitlySet() {
        var capi = new CapiSessionOptions().setEnableWebSocketResponses(false);

        JsonNode json = JsonRpcClient.getObjectMapper().valueToTree(capi);

        assertFalse(json.get("enableWebSocketResponses").asBoolean());
    }
}