-
Notifications
You must be signed in to change notification settings - Fork 95
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
a8ba63f
commit be19305
Showing
1 changed file
with
126 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,126 @@ | ||
package main | ||
|
||
import ( | ||
"os" | ||
"path/filepath" | ||
"testing" | ||
|
||
"github.com/stretchr/testify/require" | ||
"golang.org/x/exp/maps" | ||
) | ||
|
||
func TestDiffer(t *testing.T) { | ||
input := `package main | ||
func (r *ProposerRunner) ProcessPostConsensus(signedMsg *types.SignedPartialSignatureMessage) error { | ||
quorum, roots, err := r.BaseRunner.basePostConsensusMsgProcessing(r, signedMsg) | ||
if err != nil { | ||
return errors.Wrap(err, "failed processing post consensus message") | ||
} | ||
if !quorum { | ||
return nil | ||
} | ||
for _, root := range roots { | ||
sig, err := r.GetState().ReconstructBeaconSig(r.GetState().PostConsensusContainer, root, r.GetShare().ValidatorPubKey) | ||
if err != nil { | ||
// If the reconstructed signature verification failed, fall back to verifying each partial signature | ||
for _, root := range roots { | ||
r.BaseRunner.FallBackAndVerifyEachSignature(r.GetState().PostConsensusContainer, root) | ||
} | ||
return errors.Wrap(err, "got post-consensus quorum but it has invalid signatures") | ||
} | ||
specSig := phase0.BLSSignature{} | ||
copy(specSig[:], sig) | ||
if r.decidedBlindedBlock() { | ||
vBlindedBlk, _, err := r.GetState().DecidedValue.GetBlindedBlockData() | ||
if err != nil { | ||
return errors.Wrap(err, "could not get blinded block") | ||
} | ||
if err := r.GetBeaconNode().SubmitBlindedBeaconBlock(vBlindedBlk, specSig); err != nil { | ||
return errors.Wrap(err, "could not submit to Beacon chain reconstructed signed blinded Beacon block") | ||
} | ||
} else { | ||
vBlk, _, err := r.GetState().DecidedValue.GetBlockData() | ||
if err != nil { | ||
return errors.Wrap(err, "could not get block") | ||
} | ||
if err := r.GetBeaconNode().SubmitBeaconBlock(vBlk, specSig); err != nil { | ||
return errors.Wrap(err, "could not submit to Beacon chain reconstructed signed Beacon block") | ||
} | ||
} | ||
} | ||
r.GetState().Finished = true | ||
return nil | ||
}` | ||
|
||
expectedOutput := `func (r *ProposerRunner) ProcessPostConsensus(signedMsg *SignedPartialSignatureMessage) error { | ||
quorum, roots, err := r.BaseRunner.basePostConsensusMsgProcessing(r, signedMsg) | ||
if err != nil { | ||
return errors.Wrap(err, "failed processing post consensus message") | ||
} | ||
if !quorum { | ||
return nil | ||
} | ||
for _, root := range roots { | ||
sig, err := r.GetState().ReconstructBeaconSig(r.GetState().PostConsensusContainer, root, r.GetShare().ValidatorPubKey) | ||
if err != nil { | ||
for _, root := range roots { | ||
r.BaseRunner.FallBackAndVerifyEachSignature(r.GetState().PostConsensusContainer, root) | ||
} | ||
return errors.Wrap(err, "got post-consensus quorum but it has invalid signatures") | ||
} | ||
specSig := phase0.BLSSignature{} | ||
copy(specSig[:], sig) | ||
if r.decidedBlindedBlock() { | ||
vBlindedBlk, _, err := r.GetState().DecidedValue.GetBlindedBlockData() | ||
if err != nil { | ||
return errors.Wrap(err, "could not get blinded block") | ||
} | ||
if err := r.GetBeaconNode().SubmitBlindedBeaconBlock(vBlindedBlk, specSig); err != nil { | ||
return errors.Wrap(err, "could not submit to Beacon chain reconstructed signed blinded Beacon block") | ||
} | ||
} else { | ||
vBlk, _, err := r.GetState().DecidedValue.GetBlockData() | ||
if err != nil { | ||
return errors.Wrap(err, "could not get block") | ||
} | ||
if err := r.GetBeaconNode().SubmitBeaconBlock(vBlk, specSig); err != nil { | ||
return errors.Wrap(err, "could not submit to Beacon chain reconstructed signed Beacon block") | ||
} | ||
} | ||
} | ||
r.GetState().Finished = true | ||
return nil | ||
}` | ||
|
||
// Create a temporary directory with this code in it. | ||
dir, err := os.MkdirTemp("", "differ-test") | ||
require.NoError(t, err) | ||
defer os.RemoveAll(dir) | ||
|
||
err = os.WriteFile(filepath.Join(dir, "main.go"), []byte(input), 0644) | ||
require.NoError(t, err) | ||
|
||
// Run the differ on the file. | ||
elems, err := NewParser(dir, []string{"./"}, []string{"logger"}).Parse() | ||
require.NoError(t, err) | ||
require.NotNil(t, elems) | ||
require.Equal(t, 1, len(elems)) | ||
require.Equal(t, []string{"ProposerRunner.ProcessPostConsensus"}, maps.Keys(elems)) | ||
|
||
elem := elems["ProposerRunner.ProcessPostConsensus"] | ||
require.Equal(t, "ProposerRunner.ProcessPostConsensus", elem.Name) | ||
|
||
transformers := Transformers{ | ||
NoPackageNames([]string{"types", "qbft"}), | ||
NoComments(), | ||
NoEmptyLines(), | ||
} | ||
code := transformers.Transform(elem.Code) | ||
require.Equal(t, expectedOutput, code) | ||
} |