From 31fc64bc31b43c5dcddbe263a38645d146fc4e5e Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Mon, 23 Dec 2024 11:57:36 +0000
Subject: Add TomlTableWithSource.getLong()

---
 .../config/toml/TomlTableWithSourceTests.java      | 81 ++++++++++++++++++++++
 1 file changed, 81 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 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
@@ -11,6 +11,87 @@ import static org.zwobble.precisely.AssertThat.assertThat;
 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("""
-- 
cgit v1.2.3