55import com .bloxbean .cardano .client .plutus .spec .Language ;
66import org .junit .jupiter .api .Test ;
77
8- import java .util .Map ;
8+ import java .util .LinkedHashMap ;
99import java .util .Optional ;
1010
1111import static org .assertj .core .api .Assertions .assertThat ;
@@ -14,48 +14,53 @@ class CostModelUtilTest {
1414
1515 @ Test
1616 void getCostModelFromProtocolParams () {
17- Map <String , Long > costModel1 = Map .of ("verifyEd25519Signature-cpu-arguments-slope" , 1021L ,
18- "addInteger-cpu-arguments-intercept" , 205665L ,
19- "verifyEd25519Signature-memory-arguments" , 10L ,
20- "unBData-cpu-arguments" , 31220L );
21-
22- Map <String , Long > costModel2 = Map .of ("addInteger-cpu-arguments-slope" , 812L ,
23- "addInteger-cpu-arguments-intercept" , 205665L ,
24- "verifyEd25519Signature-memory-arguments" , 10L ,
25- "bData-cpu-arguments" , 1000L );
26-
27- Map <String , Map <String , Long >> costModels = Map .of ("PlutusV1" , costModel1 ,
28- "PlutusV2" , costModel2 );
17+ LinkedHashMap <String , Long > costModel1 = new LinkedHashMap <>();
18+ costModel1 .put ("verifyEd25519Signature-cpu-arguments-slope" , 1021L );
19+ costModel1 .put ("addInteger-cpu-arguments-intercept" , 205665L );
20+ costModel1 .put ("verifyEd25519Signature-memory-arguments" , 10L );
21+ costModel1 .put ("unBData-cpu-arguments" , 31220L );
22+
23+ LinkedHashMap <String , Long > costModel2 = new LinkedHashMap <>();
24+ costModel2 .put ("addInteger-cpu-arguments-slope" , 812L );
25+ costModel2 .put ("addInteger-cpu-arguments-intercept" , 205665L );
26+ costModel2 .put ("verifyEd25519Signature-memory-arguments" , 10L );
27+ costModel2 .put ("bData-cpu-arguments" , 1000L );
28+
29+ LinkedHashMap <String , LinkedHashMap <String , Long >> costModels = new LinkedHashMap <>();
30+ costModels .put ("PlutusV1" , costModel1 );
31+ costModels .put ("PlutusV2" , costModel2 );
2932 ProtocolParams protocolParams = new ProtocolParams ();
3033 protocolParams .setCostModels (costModels );
3134
3235 Optional <CostModel > v1CostModel = CostModelUtil .getCostModelFromProtocolParams (protocolParams , Language .PLUTUS_V1 );
3336 Optional <CostModel > v2CostModel = CostModelUtil .getCostModelFromProtocolParams (protocolParams , Language .PLUTUS_V2 );
3437
35- assertThat (v1CostModel .get ().getCosts ()).isEqualTo (new long []{205665L , 31220L , 1021L , 10L });
36- assertThat (v2CostModel .get ().getCosts ()).isEqualTo (new long []{205665L , 812L , 1000L , 10L });
38+ assertThat (v1CostModel .get ().getCosts ()).isEqualTo (new long []{1021L , 205665L , 10L , 31220L });
39+ assertThat (v2CostModel .get ().getCosts ()).isEqualTo (new long []{812L , 205665L , 10L , 1000L });
3740 }
3841
3942 @ Test
4043 void getCostModelFromProtocolParams_whenNotExists_returnsNull () {
41- Map <String , Long > costModel1 = Map .of ("verifyEd25519Signature-cpu-arguments-slope" , 1021L ,
42- "addInteger-cpu-arguments-intercept" , 205665L ,
43- "verifyEd25519Signature-memory-arguments" , 10L ,
44- "unBData-cpu-arguments" , 31220L );
45-
46- Map <String , Map <String , Long >> costModels = Map .of ("PlutusV1" , costModel1 );
44+ LinkedHashMap <String , Long > costModel1 = new LinkedHashMap <>();
45+ costModel1 .put ("verifyEd25519Signature-cpu-arguments-slope" , 1021L );
46+ costModel1 .put ("addInteger-cpu-arguments-intercept" , 205665L );
47+ costModel1 .put ("verifyEd25519Signature-memory-arguments" , 10L );
48+ costModel1 .put ("unBData-cpu-arguments" , 31220L );
49+
50+ LinkedHashMap <String , LinkedHashMap <String , Long >> costModels = new LinkedHashMap <>();
51+ costModels .put ("PlutusV1" , costModel1 );
4752 ProtocolParams protocolParams = new ProtocolParams ();
4853 protocolParams .setCostModels (costModels );
4954
50- Optional <CostModel > v2CostModel = CostModelUtil .getCostModelFromProtocolParams (protocolParams , Language .PLUTUS_V2 );
55+ Optional <CostModel > v2CostModel = CostModelUtil .getCostModelFromProtocolParams (protocolParams , Language .PLUTUS_V2 );
5156
5257 assertThat (v2CostModel ).isEmpty ();
5358 }
5459
5560 @ Test
5661 void getCostModelFromProtocolParams_whenNoCostModelExists_returnsNull () {
5762 ProtocolParams protocolParams = new ProtocolParams ();
58- Optional <CostModel > v2CostModel = CostModelUtil .getCostModelFromProtocolParams (protocolParams , Language .PLUTUS_V2 );
63+ Optional <CostModel > v2CostModel = CostModelUtil .getCostModelFromProtocolParams (protocolParams , Language .PLUTUS_V2 );
5964
6065 assertThat (v2CostModel ).isEmpty ();
6166 }
0 commit comments