fix: use correct type for Byron address network ID

blinklabs-io · Jan 9, 2025 · 192f949 · 192f949
1 parent 474274b
commit 192f949
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/ledger/common/address.go b/ledger/common/address.go
@@ -425,7 +425,7 @@ type byronAddressPayload struct {
 
 type ByronAddressAttributes struct {
 	Payload []byte
-	Network *uint8
+	Network *uint64
 }
 
 func (a *ByronAddressAttributes) UnmarshalCBOR(data []byte) error {
@@ -438,7 +438,7 @@ func (a *ByronAddressAttributes) UnmarshalCBOR(data []byte) error {
 	}
 	a.Payload = tmpData.Payload
 	if len(tmpData.NetworkRaw) > 0 {
-		var tmpNetwork uint8
+		var tmpNetwork uint64
 		if _, err := cbor.Decode(tmpData.NetworkRaw, &tmpNetwork); err != nil {
 			return err
 		}

diff --git a/ledger/common/address_test.go b/ledger/common/address_test.go
@@ -181,8 +181,8 @@ func TestByronAddressFromParts(t *testing.T) {
 				"5d5e698eba3dd9452add99a1af9461beb0ba61b8bece26e7399878dd",
 			),
 			addressAttr: ByronAddressAttributes{
-				// We have to jump through this hoop to get an inline pointer to a uint8
-				Network: func() *uint8 { ret := uint8(2); return &ret }(),
+				// We have to jump through this hoop to get an inline pointer to a uint64
+				Network: func() *uint64 { ret := uint64(2); return &ret }(),
 			},
 			expectedAddress: "FHnt4NL7yPXvDWHa8bVs73UEUdJd64VxWXSFNqetECtYfTd9TtJguJ14Lu3feth",
 		},