|
13 | 13 |
|
14 | 14 | /** Two registered versions; bare-name resolves to default; unknown version errors. */ |
15 | 15 | public final class Main { |
16 | | - public static void main(String[] args) throws Exception { |
17 | | - MemoryTransport.Pair pair = MemoryTransport.pair(); |
18 | | - ArcpRuntime runtime = ArcpRuntime.builder() |
19 | | - .agent("code-refactor", "1.0.0", |
20 | | - (input, ctx) -> JobOutcome.Success.inline( |
21 | | - JsonNodeFactory.instance.objectNode().put("v", "1.0.0"))) |
22 | | - .agent("code-refactor", "2.0.0", |
23 | | - (input, ctx) -> JobOutcome.Success.inline( |
24 | | - JsonNodeFactory.instance.objectNode().put("v", "2.0.0"))) |
25 | | - .build(); |
26 | | - runtime.agents().setDefault("code-refactor", "2.0.0"); |
27 | | - runtime.accept(pair.runtime()); |
| 16 | + public static void main(String[] args) throws Exception { |
| 17 | + MemoryTransport.Pair pair = MemoryTransport.pair(); |
| 18 | + ArcpRuntime runtime = |
| 19 | + ArcpRuntime.builder() |
| 20 | + .agent( |
| 21 | + "code-refactor", |
| 22 | + "1.0.0", |
| 23 | + (input, ctx) -> |
| 24 | + JobOutcome.Success.inline( |
| 25 | + JsonNodeFactory.instance.objectNode().put("v", "1.0.0"))) |
| 26 | + .agent( |
| 27 | + "code-refactor", |
| 28 | + "2.0.0", |
| 29 | + (input, ctx) -> |
| 30 | + JobOutcome.Success.inline( |
| 31 | + JsonNodeFactory.instance.objectNode().put("v", "2.0.0"))) |
| 32 | + .build(); |
| 33 | + runtime.agents().setDefault("code-refactor", "2.0.0"); |
| 34 | + runtime.accept(pair.runtime()); |
28 | 35 |
|
29 | | - try (ArcpClient client = ArcpClient.builder(pair.client()).build()) { |
30 | | - client.connect(Duration.ofSeconds(5)); |
| 36 | + try (ArcpClient client = ArcpClient.builder(pair.client()).build()) { |
| 37 | + client.connect(Duration.ofSeconds(5)); |
31 | 38 |
|
32 | | - JobHandle pinned = client.submit(ArcpClient.jobSubmit( |
33 | | - "code-refactor@1.0.0", JsonNodeFactory.instance.objectNode())); |
34 | | - JobResult pinnedResult = pinned.result().get(5, TimeUnit.SECONDS); |
35 | | - assert "1.0.0".equals(pinnedResult.result().get("v").asText()); |
36 | | - assert "code-refactor@1.0.0".equals(pinned.resolvedAgent()); |
| 39 | + JobHandle pinned = |
| 40 | + client.submit( |
| 41 | + ArcpClient.jobSubmit("code-refactor@1.0.0", JsonNodeFactory.instance.objectNode())); |
| 42 | + JobResult pinnedResult = pinned.result().get(5, TimeUnit.SECONDS); |
| 43 | + assert "1.0.0".equals(pinnedResult.result().get("v").asText()); |
| 44 | + assert "code-refactor@1.0.0".equals(pinned.resolvedAgent()); |
37 | 45 |
|
38 | | - JobHandle bare = client.submit(ArcpClient.jobSubmit( |
39 | | - "code-refactor", JsonNodeFactory.instance.objectNode())); |
40 | | - JobResult bareResult = bare.result().get(5, TimeUnit.SECONDS); |
41 | | - assert "2.0.0".equals(bareResult.result().get("v").asText()); |
42 | | - assert "code-refactor@2.0.0".equals(bare.resolvedAgent()); |
| 46 | + JobHandle bare = |
| 47 | + client.submit( |
| 48 | + ArcpClient.jobSubmit("code-refactor", JsonNodeFactory.instance.objectNode())); |
| 49 | + JobResult bareResult = bare.result().get(5, TimeUnit.SECONDS); |
| 50 | + assert "2.0.0".equals(bareResult.result().get("v").asText()); |
| 51 | + assert "code-refactor@2.0.0".equals(bare.resolvedAgent()); |
43 | 52 |
|
44 | | - try { |
45 | | - client.submit(ArcpClient.jobSubmit( |
46 | | - "code-refactor@9.9.9", JsonNodeFactory.instance.objectNode())); |
47 | | - throw new AssertionError("expected AgentVersionNotAvailableException"); |
48 | | - } catch (RuntimeException e) { |
49 | | - Throwable root = e; |
50 | | - while (root.getCause() != null) { |
51 | | - root = root.getCause(); |
52 | | - } |
53 | | - assert root instanceof AgentVersionNotAvailableException |
54 | | - : "got " + root; |
55 | | - } |
56 | | - |
57 | | - System.out.println("OK agent-versions"); |
| 53 | + try { |
| 54 | + client.submit( |
| 55 | + ArcpClient.jobSubmit("code-refactor@9.9.9", JsonNodeFactory.instance.objectNode())); |
| 56 | + throw new AssertionError("expected AgentVersionNotAvailableException"); |
| 57 | + } catch (RuntimeException e) { |
| 58 | + Throwable root = e; |
| 59 | + while (root.getCause() != null) { |
| 60 | + root = root.getCause(); |
58 | 61 | } |
59 | | - runtime.close(); |
| 62 | + assert root instanceof AgentVersionNotAvailableException : "got " + root; |
| 63 | + } |
| 64 | + |
| 65 | + System.out.println("OK agent-versions"); |
60 | 66 | } |
| 67 | + runtime.close(); |
| 68 | + } |
61 | 69 | } |
0 commit comments