copilot-sdk/java/sdk/src/test/java/com/github/copilot/GitHubTokenProviderRegistryTest.java at main · santhosh2005msd-maker/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
/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *--------------------------------------------------------------------------------------------*/

package com.github.copilot;

import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertNull;
import static org.junit.jupiter.api.Assertions.assertInstanceOf;
import static org.junit.jupiter.api.Assertions.assertThrows;

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CompletionException;

import org.junit.jupiter.api.Test;

import com.github.copilot.rpc.GitHubTokenProviderResult;
import com.github.copilot.rpc.PermissionHandler;
import com.github.copilot.rpc.ResumeSessionConfig;
import com.github.copilot.rpc.SessionConfig;

class GitHubTokenProviderRegistryTest {

    @Test
    void staticTokenAndProviderAreMutuallyExclusive() {
        try (var client = new CopilotClient()) {
            var create = new SessionConfig().setGitHubToken("static")
                    .setGitHubTokenProvider(
                            args -> CompletableFuture.completedFuture(GitHubTokenProviderResult.cancelled()))
                    .setOnPermissionRequest(PermissionHandler.APPROVE_ALL);
            var createError = assertThrows(CompletionException.class, () -> client.createSession(create).join());
            assertInstanceOf(IllegalArgumentException.class, createError.getCause());

            var resume = new ResumeSessionConfig().setGitHubToken("static")
                    .setGitHubTokenProvider(
                            args -> CompletableFuture.completedFuture(GitHubTokenProviderResult.cancelled()))
                    .setOnPermissionRequest(PermissionHandler.APPROVE_ALL);
            var resumeError = assertThrows(CompletionException.class,
                    () -> client.resumeSession("session", resume).join());
            assertInstanceOf(IllegalArgumentException.class, resumeError.getCause());
        }
    }

    @Test
    void registrationRollbackAndResumeReplacementAreIsolated() {
        var registry = new GitHubTokenProviderRegistry();
        var first = registry.register(args -> CompletableFuture.completedFuture(GitHubTokenProviderResult.cancelled()));
        var second = registry
                .register(args -> CompletableFuture.completedFuture(GitHubTokenProviderResult.cancelled()));

        assertNotNull(registry.get(first.id()));
        assertNotNull(registry.get(second.id()));
        first.claim("session-1");
        second.claim("session-1");
        assertNull(registry.get(first.id()));
        assertNotNull(registry.get(second.id()));

        first.close();
        assertNotNull(registry.get(second.id()));
        second.close();
        assertNull(registry.get(second.id()));

        var retired = registry
                .register(args -> CompletableFuture.completedFuture(GitHubTokenProviderResult.cancelled()));
        retired.claim("session-2");
        registry.retire("session-2");
        assertNull(registry.get(retired.id()));

        var failed = registry
                .register(args -> CompletableFuture.completedFuture(GitHubTokenProviderResult.cancelled()));
        failed.close();
        assertNull(registry.get(failed.id()));
    }
}