diff --git a/app/rpc/rpchandlers/get_virtual_selected_parent_blue_score.go b/app/rpc/rpchandlers/get_virtual_selected_parent_blue_score.go index 4500bf09e..3b96e67c7 100644 --- a/app/rpc/rpchandlers/get_virtual_selected_parent_blue_score.go +++ b/app/rpc/rpchandlers/get_virtual_selected_parent_blue_score.go @@ -8,9 +8,14 @@ import ( // HandleGetVirtualSelectedParentBlueScore handles the respectively named RPC command func HandleGetVirtualSelectedParentBlueScore(context *rpccontext.Context, _ *router.Router, _ appmessage.Message) (appmessage.Message, error) { - virtualInfo, err := context.Domain.Consensus().GetVirtualInfo() + c := context.Domain.Consensus() + selectedParent, err := c.GetVirtualSelectedParent() if err != nil { return nil, err } - return appmessage.NewGetVirtualSelectedParentBlueScoreResponseMessage(virtualInfo.BlueScore), nil + blockInfo, err := c.GetBlockInfo(selectedParent) + if err != nil { + return nil, err + } + return appmessage.NewGetVirtualSelectedParentBlueScoreResponseMessage(blockInfo.BlueScore), nil } diff --git a/testing/integration/virtual_selected_parent_blue_score_test.go b/testing/integration/virtual_selected_parent_blue_score_test.go index 2d22289bd..651fc4041 100644 --- a/testing/integration/virtual_selected_parent_blue_score_test.go +++ b/testing/integration/virtual_selected_parent_blue_score_test.go @@ -17,14 +17,14 @@ func TestVirtualSelectedParentBlueScoreAndVirtualDAAScore(t *testing.T) { kaspad, teardown := setupHarness(t, harnessParams) defer teardown() - // Make sure that the initial blue score is 1 + // Make sure that the initial selected parent blue score is 0 response, err := kaspad.rpcClient.GetVirtualSelectedParentBlueScore() if err != nil { t.Fatalf("Error getting virtual selected parent blue score: %s", err) } - if response.BlueScore != 1 { + if response.BlueScore != 0 { t.Fatalf("Unexpected virtual selected parent blue score. Want: %d, got: %d", - 1, response.BlueScore) + 0, response.BlueScore) } // Register to virtual selected parent blue score changes @@ -70,8 +70,8 @@ func TestVirtualSelectedParentBlueScoreAndVirtualDAAScore(t *testing.T) { if err != nil { t.Fatalf("Error getting virtual selected parent blue score: %s", err) } - if response.BlueScore != 1+blockAmountToMine { + if response.BlueScore != blockAmountToMine { t.Fatalf("Unexpected virtual selected parent blue score. Want: %d, got: %d", - 1+blockAmountToMine, response.BlueScore) + blockAmountToMine, response.BlueScore) } }