|
1 | 1 | package ledger |
2 | 2 |
|
3 | 3 | import ( |
| 4 | + "fmt" |
4 | 5 | "testing" |
| 6 | + |
| 7 | + "github.com/blinklabs-io/gouroboros/ledger/allegra" |
| 8 | + "github.com/blinklabs-io/gouroboros/ledger/babbage" |
| 9 | + "github.com/blinklabs-io/gouroboros/ledger/common" |
| 10 | + "github.com/blinklabs-io/gouroboros/ledger/mary" |
| 11 | + "github.com/blinklabs-io/gouroboros/ledger/shelley" |
| 12 | + "github.com/fxamacker/cbor/v2" |
5 | 13 | ) |
6 | 14 |
|
7 | 15 | func TestVerifyBlockBody(t *testing.T) { |
@@ -57,3 +65,132 @@ func TestVerifyBlockBody(t *testing.T) { |
57 | 65 | }) |
58 | 66 | } |
59 | 67 | } |
| 68 | + |
| 69 | +func mockShelleyCBOR() []byte { |
| 70 | + shelleyHeader := shelley.ShelleyBlockHeader{ |
| 71 | + Body: shelley.ShelleyBlockHeaderBody{ |
| 72 | + BlockNumber: 12345, |
| 73 | + Slot: 67890, |
| 74 | + PrevHash: common.Blake2b256{}, |
| 75 | + IssuerVkey: common.IssuerVkey{}, |
| 76 | + VrfKey: []byte{0x01, 0x02}, |
| 77 | + NonceVrf: common.VrfResult{}, |
| 78 | + LeaderVrf: common.VrfResult{}, |
| 79 | + BlockBodySize: 512, |
| 80 | + BlockBodyHash: common.Blake2b256{}, |
| 81 | + OpCertHotVkey: []byte{0x03, 0x04}, |
| 82 | + OpCertSequenceNumber: 10, |
| 83 | + OpCertKesPeriod: 20, |
| 84 | + OpCertSignature: []byte{0x05, 0x06}, |
| 85 | + ProtoMajorVersion: 1, |
| 86 | + ProtoMinorVersion: 0, |
| 87 | + }, |
| 88 | + Signature: []byte{0x07, 0x08}, |
| 89 | + } |
| 90 | + |
| 91 | + // Convert to CBOR |
| 92 | + data, err := cbor.Marshal(shelleyHeader) |
| 93 | + if err != nil { |
| 94 | + fmt.Printf("CBOR Encoding Error: %v\n", err) |
| 95 | + } |
| 96 | + return data |
| 97 | +} |
| 98 | + |
| 99 | +func mockAllegraCBOR() []byte { |
| 100 | + allegraHeader := allegra.AllegraBlockHeader{ShelleyBlockHeader: ShelleyBlockHeader{}} |
| 101 | + data, _ := cbor.Marshal(allegraHeader) |
| 102 | + return data |
| 103 | +} |
| 104 | + |
| 105 | +func mockMaryCBOR() []byte { |
| 106 | + maryHeader := mary.MaryBlockHeader{ShelleyBlockHeader: ShelleyBlockHeader{}} |
| 107 | + data, _ := cbor.Marshal(maryHeader) |
| 108 | + return data |
| 109 | +} |
| 110 | + |
| 111 | +func mockAlonzoCBOR() []byte { |
| 112 | + alonzoHeader := AlonzoBlockHeader{ShelleyBlockHeader: ShelleyBlockHeader{}} |
| 113 | + data, _ := cbor.Marshal(alonzoHeader) |
| 114 | + return data |
| 115 | +} |
| 116 | + |
| 117 | +func mockBabbageCBOR() []byte { |
| 118 | + babbageHeader := babbage.BabbageBlockHeader{ |
| 119 | + Body: babbage.BabbageBlockHeaderBody{ |
| 120 | + BlockNumber: 54321, |
| 121 | + Slot: 98765, |
| 122 | + PrevHash: common.Blake2b256{}, |
| 123 | + IssuerVkey: common.IssuerVkey{}, |
| 124 | + VrfKey: []byte{0x09, 0x10}, |
| 125 | + VrfResult: common.VrfResult{}, |
| 126 | + BlockBodySize: 1024, |
| 127 | + BlockBodyHash: common.Blake2b256{}, |
| 128 | + OpCert: babbage.BabbageOpCert{ |
| 129 | + HotVkey: []byte{0x11, 0x12}, |
| 130 | + SequenceNumber: 30, |
| 131 | + KesPeriod: 40, |
| 132 | + Signature: []byte{0x13, 0x14}, |
| 133 | + }, |
| 134 | + ProtoVersion: babbage.BabbageProtoVersion{ |
| 135 | + Major: 2, |
| 136 | + Minor: 0, |
| 137 | + }, |
| 138 | + }, |
| 139 | + Signature: []byte{0x15, 0x16}, |
| 140 | + } |
| 141 | + |
| 142 | + // Convert to CBOR |
| 143 | + data, err := cbor.Marshal(babbageHeader) |
| 144 | + if err != nil { |
| 145 | + fmt.Printf("CBOR Encoding Error for Babbage: %v\n", err) |
| 146 | + } |
| 147 | + return data |
| 148 | +} |
| 149 | + |
| 150 | +func mockConwayCBOR() []byte { |
| 151 | + conwayHeader := ConwayBlockHeader{BabbageBlockHeader: BabbageBlockHeader{}} |
| 152 | + data, _ := cbor.Marshal(conwayHeader) |
| 153 | + return data |
| 154 | +} |
| 155 | + |
| 156 | +func TestNewBlockHeaderFromCbor(t *testing.T) { |
| 157 | + tests := []struct { |
| 158 | + name string |
| 159 | + blockType uint |
| 160 | + data []byte |
| 161 | + expectErr bool |
| 162 | + expectedFn string |
| 163 | + }{ |
| 164 | + {"Shelley Block", BlockTypeShelley, mockShelleyCBOR(), false, "NewShelleyBlockHeaderFromCbor"}, |
| 165 | + {"Allegra Block", BlockTypeAllegra, mockAllegraCBOR(), false, "NewAllegraBlockHeaderFromCbor"}, |
| 166 | + {"Mary Block", BlockTypeMary, mockMaryCBOR(), false, "NewMaryBlockHeaderFromCbor"}, |
| 167 | + {"Alonzo Block", BlockTypeAlonzo, mockAlonzoCBOR(), false, "NewAlonzoBlockHeaderFromCbor"}, |
| 168 | + {"Babbage Block", BlockTypeBabbage, mockBabbageCBOR(), false, "NewBabbageBlockHeaderFromCbor"}, |
| 169 | + {"Conway Block", BlockTypeConway, mockConwayCBOR(), false, "NewConwayBlockHeaderFromCbor"}, |
| 170 | + {"Invalid Block Type", 9999, []byte{0xFF, 0x00, 0x00}, true, "UnknownFunction"}, |
| 171 | + } |
| 172 | + |
| 173 | + for _, test := range tests { |
| 174 | + t.Run(test.name, func(t *testing.T) { |
| 175 | + fmt.Printf("\n Running Test: %s\n", test.name) |
| 176 | + |
| 177 | + header, err := NewBlockHeaderFromCbor(test.blockType, test.data) |
| 178 | + |
| 179 | + if test.expectErr { |
| 180 | + if err == nil { |
| 181 | + t.Errorf("Expected error for %s, but got none!", test.name) |
| 182 | + } else { |
| 183 | + fmt.Printf("Expected failure for %s: %v\n", test.name, err) |
| 184 | + } |
| 185 | + } else { |
| 186 | + if err != nil { |
| 187 | + t.Errorf("Unexpected error for %s: %v", test.name, err) |
| 188 | + } else if header == nil { |
| 189 | + t.Errorf("Expected non-nil block header for %s, but got nil", test.name) |
| 190 | + } else { |
| 191 | + fmt.Printf("Test Passed: %s → %s executed successfully!\n", test.name, test.expectedFn) |
| 192 | + } |
| 193 | + } |
| 194 | + }) |
| 195 | + } |
| 196 | +} |
0 commit comments