Skip to content

Commit f1bc178

Browse files
committed
Update PeerStatus to use generic record format
1 parent 255f649 commit f1bc178

File tree

2 files changed

+10
-6
lines changed

2 files changed

+10
-6
lines changed

convex-core/src/main/java/convex/core/cvm/PeerStatus.java

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -265,12 +265,12 @@ public PeerStatus withPeerStake(long newStake) {
265265
if (peerStake == newStake) return this;
266266
long stakeChange=newStake-peerStake;
267267

268-
return new PeerStatus(controller, newStake, stakes, delegatedStake, metadata,timestamp,balance+stakeChange);
268+
return new PeerStatus(controller, newStake, getStakes(), delegatedStake, getMetadata(),timestamp,balance+stakeChange);
269269
}
270270

271271
public PeerStatus withPeerData(AHashMap<ACell,ACell> newMeta) {
272272
if (metadata==newMeta) return this;
273-
return new PeerStatus(controller, peerStake, stakes, delegatedStake, newMeta,timestamp,balance);
273+
return new PeerStatus(controller, peerStake, getStakes(), delegatedStake, newMeta,timestamp,balance);
274274
}
275275

276276
@Override
@@ -283,12 +283,12 @@ public void validateCell() throws InvalidDataException {
283283
@Override
284284
public ACell get(Keyword key) {
285285
if (Keywords.CONTROLLER.equals(key)) return controller;
286-
if (Keywords.STAKE.equals(key)) return CVMLong.create(peerStake);
286+
if (Keywords.STAKE.equals(key)) return values.get(1);
287287
if (Keywords.STAKES.equals(key)) return getStakes();
288-
if (Keywords.DELEGATED_STAKE.equals(key)) return CVMLong.create(delegatedStake);
288+
if (Keywords.DELEGATED_STAKE.equals(key)) return values.get(3);
289289
if (Keywords.METADATA.equals(key)) return getMetadata();
290-
if (Keywords.TIMESTAMP.equals(key)) return CVMLong.create(timestamp);
291-
if (Keywords.BALANCE.equals(key)) return CVMLong.create(balance);
290+
if (Keywords.TIMESTAMP.equals(key)) return values.get(5);
291+
if (Keywords.BALANCE.equals(key)) return values.get(6);
292292

293293
return null;
294294
}

convex-core/src/test/java/convex/core/data/PeerStatusTest.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,23 @@
11
package convex.core.data;
22

3+
import static org.junit.jupiter.api.Assertions.assertEquals;
34
import static org.junit.jupiter.api.Assertions.assertTrue;
45

56
import org.junit.jupiter.api.Test;
67

78
import convex.core.cvm.Address;
89
import convex.core.cvm.Keywords;
910
import convex.core.cvm.PeerStatus;
11+
import convex.core.data.prim.CVMLong;
1012
import convex.core.exceptions.InvalidDataException;
1113
import convex.core.util.Utils;
1214

1315
public class PeerStatusTest {
1416

1517
@Test public void testEmpty() {
1618
PeerStatus ps=PeerStatus.create(null, 0);
19+
20+
assertEquals(CVMLong.ZERO,ps.get(Keywords.DELEGATED_STAKE));
1721
doPeerStatusTest(ps);
1822
}
1923

0 commit comments

Comments
 (0)