feat: Convert indexer table to GRT type

Signed-off-by: Anirudh <[email protected]>
graphprotocol · Mar 9, 2023 · 19bd0ac · 19bd0ac
1 parent e0ef0e6
commit 19bd0ac
Show file tree

Hide file tree

Showing 2 changed files with 67 additions and 0 deletions.
diff --git a/src/data.jl b/src/data.jl
@@ -225,3 +225,40 @@ function write(i::FlexTable, a::FlexTable, s::FlexTable, n::FlexTable, config::A
     end
     return ps
 end
+
+"""
+    correcttypes!(::Val{:indexer}, i::FlexTable)
+
+Converts the string currency fields in the indexer table to be in GRT.
+
+```julia
+julia> using AllocationOpt
+julia> i = flextable([
+    Dict(
+        "stakedTokens" => "1",
+        "delegatedTokens" => "0",
+        "id" => "0xa",
+        "lockedTokens" => "0",
+    ),
+    Dict(
+        "stakedTokens" => "1",
+        "delegatedTokens" => "0",
+        "id" => "0xb",
+        "lockedTokens" => "0",
+    ),
+    Dict(
+        "stakedTokens" => "1",
+        "delegatedTokens" => "0",
+        "id" => "0xc",
+        "lockedTokens" => "0",
+    ),
+])
+julia> AllocationOpt.correcttypes!(Val(:indexer), i)
+```
+"""
+function correcttypes!(::Val{:indexer}, i::FlexTable)
+    i.stakedTokens = i.stakedTokens .|> togrt
+    i.delegatedTokens = i.delegatedTokens .|> togrt
+    i.lockedTokens = i.lockedTokens .|> togrt
+    return i
+end
diff --git a/test/data.jl b/test/data.jl
@@ -97,4 +97,34 @@
             ]
         end
     end
+
+    @testset "correcttypes!" begin
+        @testset "indexer" begin
+            i = flextable([
+                Dict(
+                    "stakedTokens" => "1",
+                    "delegatedTokens" => "0",
+                    "id" => "0xa",
+                    "lockedTokens" => "0",
+                ),
+                Dict(
+                    "stakedTokens" => "1",
+                    "delegatedTokens" => "0",
+                    "id" => "0xb",
+                    "lockedTokens" => "0",
+                ),
+                Dict(
+                    "stakedTokens" => "1",
+                    "delegatedTokens" => "0",
+                    "id" => "0xc",
+                    "lockedTokens" => "0",
+                ),
+            ])
+            AllocationOpt.correcttypes!(Val(:indexer), i)
+            @test i.stakedTokens == [1e-18, 1e-18, 1e-18]
+            @test i.delegatedTokens == [0, 0, 0]
+            @test i.id == ["0xa", "0xb", "0xc"]
+            @test i.lockedTokens == [0, 0, 0]
+        end
+    end
 end