test dkg status metrics

Author: jbsv

services/dkg/pedersen/mod_test.go

@@ -54,6 +54,7 @@ func TestActor_MarshalJSON(t *testing.T) {
 	actor1, err := p.NewActor([]byte("deadbeef"), &fake.Pool{},
 		fake.Manager{}, NewHandlerData())
 	require.NoError(t, err)
+	require.Equal(t, float64(dkg.Initialized), testutil.ToFloat64(evoting.PromElectionDkgStatus))
 
 	// Serialize its persistent data
 	actor1Buf, err := actor1.MarshalJSON()
@@ -64,8 +65,11 @@ func TestActor_MarshalJSON(t *testing.T) {
 	err = handlerData.UnmarshalJSON(actor1Buf)
 	require.NoError(t, err)
 
+	initMetrics()
+
 	actor2, err := p.NewActor([]byte("beefdead"), &fake.Pool{}, fake.Manager{}, handlerData)
 	require.NoError(t, err)
+	require.Equal(t, float64(dkg.Initialized), testutil.ToFloat64(evoting.PromElectionDkgStatus))
 
 	// Check that the persistent data is the same for both actors
 	requireActorsEqual(t, actor1, actor2)
@@ -74,6 +78,8 @@ func TestActor_MarshalJSON(t *testing.T) {
 // After initializing a Pedersen when dkgMap is not empty, the actors map should
 // contain the same information as dkgMap
 func TestPedersen_InitNonEmptyMap(t *testing.T) {
+	initMetrics()
+
 	// Create a new DKG map and fill it with data
 	dkgMap := fake.NewInMemoryDB()
 
@@ -143,9 +149,14 @@ func TestPedersen_InitNonEmptyMap(t *testing.T) {
 
 			_, err = p.NewActor(electionIDBuf, &fake.Pool{}, fake.Manager{}, handlerData)
 			if err != nil {
+				require.Equal(t, float64(dkg.Failed), testutil.ToFloat64(evoting.PromElectionDkgStatus))
 				return err
+			} else {
+				require.Equal(t, float64(dkg.Initialized), testutil.ToFloat64(evoting.PromElectionDkgStatus))
 			}
 
+			initMetrics()
+
 			return nil
 		})
 	})
@@ -331,6 +342,8 @@ func TestPedersen_Setup(t *testing.T) {
 	require.EqualError(t, err, "failed to get election: election does not exist: <nil>")
 	require.Equal(t, float64(dkg.Failed), testutil.ToFloat64(evoting.PromElectionDkgStatus))
 
+	initMetrics()
+
 	actor.electionID = electionID
 
 	// RPC is bogus 1