fixing linting errors

ChainSafe · Feb 25, 2022 · 0701543 · 0701543
1 parent 48a2a58
commit 0701543
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 5 deletions.
diff --git a/dot/core/messages_integration_test.go b/dot/core/messages_integration_test.go
@@ -81,8 +81,8 @@ func TestService_HandleBlockProduced(t *testing.T) {
 	require.NoError(t, err)
 
 	// simulate block sent from BABE session
-	digest := types.ToPreRuntimeDigest(*types.NewDigest()
-	prd, err := types.NewBabeSecondaryPlainPreDigest(0, 1))
+	digest := types.NewDigest()
+	prd, err := types.ToPreRuntimeDigest(*types.NewBabeSecondaryPlainPreDigest(0, 1))
 	require.NoError(t, err)
 	err = digest.Add(*prd)
 	require.NoError(t, err)

diff --git a/dot/core/service_integration_test.go b/dot/core/service_integration_test.go
@@ -74,8 +74,8 @@ func TestAnnounceBlock(t *testing.T) {
 	defer s.Stop()
 
 	// simulate block sent from BABE session
-	digest := types.ToPreRuntimeDigest(*types.NewDigest()
-	prd, err := types.NewBabeSecondaryPlainPreDigest(0, 1))
+	digest := types.NewDigest()
+	prd, err := types.ToPreRuntimeDigest(*types.NewBabeSecondaryPlainPreDigest(0, 1))
 	require.NoError(t, err)
 	err = digest.Add(*prd)
 	require.NoError(t, err)

diff --git a/dot/types/babe_digest.go b/dot/types/babe_digest.go
@@ -95,6 +95,7 @@ func NewBabeSecondaryVRFPreDigest(authorityIndex uint32,
 // Index Returns VDT index
 func (d BabeSecondaryVRFPreDigest) Index() uint { return 3 }
 
+// ToPreRuntimeDigest returns the VaryingDataTypeValue as a PreRuntimeDigest
 func ToPreRuntimeDigest(value scale.VaryingDataTypeValue) (*PreRuntimeDigest, error) {
 	digest := NewBabeDigest()
 	err := digest.Set(value)

diff --git a/lib/babe/build_integration_test.go b/lib/babe/build_integration_test.go
@@ -171,7 +171,7 @@ func TestApplyExtrinsic(t *testing.T) {
 	}
 	testVRFOutputAndProof := &VrfOutputAndProof{}
 
-	preDigest2, err := types.types.ToPreRuntimeDigest(*types.NewBabePrimaryPreDigest(
+	preDigest2, err := types.ToPreRuntimeDigest(*types.NewBabePrimaryPreDigest(
 		authorityIndex, slot2.number,
 		testVRFOutputAndProof.output,
 		testVRFOutputAndProof.proof,