From 5e54269bed1266534435f54c950fb7b9645dff24 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Mon, 23 Dec 2024 12:00:42 +0000
Subject: Add TomlTableWithSource.getLongOrNone()

---
 .../config/toml/TomlTableWithSourceTests.java      | 73 ++++++++++++++++++++++
 1 file changed, 73 insertions(+)

(limited to 'src/test/java')

diff --git a/src/test/java/org/shedlang/compiler/config/toml/TomlTableWithSourceTests.java b/src/test/java/org/shedlang/compiler/config/toml/TomlTableWithSourceTests.java
index 0165f8df..41bb3c0f 100644
--- a/src/test/java/org/shedlang/compiler/config/toml/TomlTableWithSourceTests.java
+++ b/src/test/java/org/shedlang/compiler/config/toml/TomlTableWithSourceTests.java
@@ -5,6 +5,7 @@ import org.shedlang.compiler.config.errors.ConfigError;
 import org.shedlang.compiler.sources.Sources;
 
 import java.util.Optional;
+import java.util.OptionalLong;
 
 import static org.junit.jupiter.api.Assertions.assertThrows;
 import static org.zwobble.precisely.AssertThat.assertThat;
@@ -91,6 +92,78 @@ public class TomlTableWithSourceTests {
 
         assertThat(result, equalTo(2L));
     }
+    @Test
+    public void whenKeyIsMissingThenGetLongOrNoneReturnsEmptyOptional() {
+        var source = Sources.string("""
+            [[a.b]]
+            x = 1
+            
+            [[a.b]]
+            """);
+        var table = TomlTableWithSource.parse(source);
+
+        var result = table
+            .getTable("a")
+            .orElseThrow()
+            .getArrayOfTables("b")
+            .orElseThrow()
+            .get(1)
+            .getLongOrNone("x");
+
+        assertThat(result, equalTo(OptionalLong.empty()));
+    }
+
+    @Test
+    public void whenKeyIsNotLongThenGetLongOrNoneThrowsError() {
+        var source = Sources.string("""
+            [[a.b]]
+            x = 1
+            
+            [[a.b]]
+            x = true
+            """);
+        var table = TomlTableWithSource.parse(source);
+
+        var error = assertThrows(
+            ConfigError.class,
+            () -> table
+                .getTable("a")
+                .orElseThrow()
+                .getArrayOfTables("b")
+                .orElseThrow()
+                .get(1)
+                .getLongOrNone("x")
+        );
+
+        assertThat(error.getMessage(), equalTo("Expected a.b[1].x to be a long, but was Boolean"));
+        assertThat(error.source().describe(), equalTo("""
+            <string>:5:1
+            x = true
+            ^"""
+        ));
+    }
+
+    @Test
+    public void whenKeyIsLongOrNoneThenGetLongReturnsValue() {
+        var source = Sources.string("""
+            [[a.b]]
+            x = 1
+            
+            [[a.b]]
+            x = 2
+            """);
+        var table = TomlTableWithSource.parse(source);
+
+        var result = table
+            .getTable("a")
+            .orElseThrow()
+            .getArrayOfTables("b")
+            .orElseThrow()
+            .get(1)
+            .getLongOrNone("x");
+
+        assertThat(result, equalTo(OptionalLong.of(2L)));
+    }
 
     @Test
     public void whenKeyIsMissingThenGetStringThrowsError() {
-- 
cgit v1.2.3