Add TomlTableWithSource.getLong()

1 files changed, 81 insertions, 0 deletions

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 3dcc17f0..0165f8df 100644
--- a/src/test/java/org/shedlang/compiler/config/toml/TomlTableWithSourceTests.java
+++ b/src/test/java/org/shedlang/compiler/config/toml/TomlTableWithSourceTests.java
@@ -12,6 +12,87 @@ import static org.zwobble.precisely.Matchers.equalTo;
 
 public class TomlTableWithSourceTests {
     @Test
+    public void whenKeyIsMissingThenGetLongThrowsError() {
+        var source = Sources.string("""
+            [[a.b]]
+            x = 1
+            
+            [[a.b]]
+            """);
+        var table = TomlTableWithSource.parse(source);
+
+        var error = assertThrows(
+            ConfigError.class,
+            () -> table
+                .getTable("a")
+                .orElseThrow()
+                .getArrayOfTables("b")
+                .orElseThrow()
+                .get(1)
+                .getLong("x")
+        );
+
+        assertThat(error.getMessage(), equalTo("a.b[1] is missing x"));
+        assertThat(error.source().describe(), equalTo("""
+            <string>:4:1
+            [[a.b]]
+            ^"""
+        ));
+    }
+
+    @Test
+    public void whenKeyIsNotLongThenGetLongThrowsError() {
+        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)
+                .getLong("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 whenKeyIsLongThenGetLongReturnsValue() {
+        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)
+            .getLong("x");
+
+        assertThat(result, equalTo(2L));
+    }
+
+    @Test
     public void whenKeyIsMissingThenGetStringThrowsError() {
         var source = Sources.string("""
             [[shed.bins]]