Function TestRegisterWorker

internal/control_plane/control_plane_test.go:108–151

Registration tests

(t *testing.T)

// Registration tests

func TestRegisterWorker(t *testing.T) {
	logrus.SetLevel(logrus.FatalLevel)

	persistenceLayer, ctrl := simplePersistenceLayer(t)
	defer ctrl.Finish()

	nbRegistrations := 100

	persistenceLayer.EXPECT().StoreWorkerNodeInformation(gomock.Any(), gomock.Any()).DoAndReturn(func(_ context.Context, _ *proto.WorkerNodeInformation) error {
		return nil
	}).Times(nbRegistrations)

	controlPlane := NewControlPlane(persistenceLayer, "", placement_policy.NewRandomPlacement(), data_plane.NewDataplaneConnection, workers.NewWorkerNode, &mockConfig)

	err := controlPlane.ReconstructState(context.Background(), mockConfig, nil)

	assert.NoError(t, err, "reconstructing control plane state failed")

	assert.Zero(t, controlPlane.GetNumberConnectedWorkers(), "Number of connected workers should be 0")
	assert.Zero(t, controlPlane.GetNumberDataplanes(), "Number of connected data planes should be 0")
	assert.Zero(t, controlPlane.GetNumberServices(), "Number of registered services should be 0")

	for i := 1; i <= nbRegistrations; i++ {
		name := "mock" + fmt.Sprint(i)

		status, err := controlPlane.RegisterNode(context.Background(), &proto.NodeInfo{
			NodeID: name,
		})

		assert.True(t, status.Success, "status should be successful")
		assert.NoError(t, err, "error should not be nil")

		assert.Equal(t, i, controlPlane.GetNumberConnectedWorkers())

		status, err = controlPlane.RegisterNode(context.Background(), &proto.NodeInfo{
			NodeID: name,
		})

		assert.False(t, status.Success, "status should be unsuccessful")
		assert.NoError(t, err, "error should not be nil")

		assert.Equal(t, i, controlPlane.GetNumberConnectedWorkers())
	}
}

func TestRegisterDataplanes(t *testing.T) {
	logrus.SetLevel(logrus.FatalLevel)