Fix: update defi lama column

celenium-io · Dec 13, 2024 · 475d4c5 · 475d4c5
1 parent fe01c99
commit 475d4c5
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/internal/storage/postgres/transaction.go b/internal/storage/postgres/transaction.go
@@ -659,6 +659,9 @@ func (tx Transaction) UpdateRollup(ctx context.Context, rollup *models.Rollup) e
 	if rollup.VM != "" {
 		query = query.Set("vm = ?", rollup.VM)
 	}
+	if rollup.DeFiLama != "" {
+		query = query.Set("defi_lama = ?", rollup.DeFiLama)
+	}
 
 	_, err := query.Exec(ctx)
 	return err

diff --git a/internal/storage/postgres/transaction_test.go b/internal/storage/postgres/transaction_test.go
@@ -757,6 +757,7 @@ func (s *TransactionTestSuite) TestSaveUpdateAndDeleteRollup() {
 	s.Require().Greater(rollup.Id, uint64(0))
 
 	rollup.GitHub = "https://github.com/rollup2"
+	rollup.DeFiLama = "test"
 	err = tx.UpdateRollup(ctx, rollup)
 	s.Require().NoError(err)
 
@@ -774,6 +775,7 @@ func (s *TransactionTestSuite) TestSaveUpdateAndDeleteRollup() {
 	s.Require().EqualValues(testLink, newRollup.L2Beat)
 	s.Require().EqualValues(testLink, newRollup.Explorer)
 	s.Require().EqualValues(testLink, newRollup.BridgeContract)
+	s.Require().EqualValues("test", newRollup.DeFiLama)
 	s.Require().EqualValues("stack", newRollup.Stack)
 	s.Require().Len(newRollup.Links, 1)
 

diff --git a/test/data/rollup.yml b/test/data/rollup.yml
@@ -9,6 +9,7 @@
   category: finance
   stack: OP Stack
   type: settled
+  defi_lama: defi
 - id: 2
   name: Rollup 2
   description: The second
@@ -20,6 +21,7 @@
   category: gaming
   stack: OP Stack
   type: settled
+  defi_lama: lama
 - id: 3
   name: Rollup 3
   description: The third
@@ -31,3 +33,4 @@
   category: nft
   stack: Custom Stack
   type: sovereign
+  defi_lama: name