test: TestClaimType

0xPolygon · Dec 9, 2024 · 1e9ed4b · 1e9ed4b
1 parent 8e9c279
commit 1e9ed4b
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/agglayer/types_test.go b/agglayer/types_test.go
@@ -912,3 +912,29 @@ func TestL1InfoTreeLeafString(t *testing.T) {
 		})
 	}
 }
+
+func TestClaimType(t *testing.T) {
+	cases := []struct {
+		name         string
+		claim        Claim
+		expectedType string
+	}{
+		{
+			name:         "Mainnet claim",
+			claim:        &ClaimFromMainnnet{},
+			expectedType: "Mainnet",
+		},
+		{
+			name:         "Rollup claim",
+			claim:        &ClaimFromRollup{},
+			expectedType: "Rollup",
+		},
+	}
+
+	for _, c := range cases {
+		t.Run(c.name, func(t *testing.T) {
+			actualType := c.claim.Type()
+			require.Equal(t, c.expectedType, actualType)
+		})
+	}
+}