Function TestReconstructionService

internal/control_plane/control_plane_test.go:492–560

Reconstruction tests

(t *testing.T)

// Reconstruction tests

func TestReconstructionService(t *testing.T) {
	ctrl := gomock.NewController(t)
	defer ctrl.Finish()

	logrus.SetLevel(logrus.FatalLevel)

	persistenceLayer := mock_persistence.NewMockPersistenceLayer(ctrl)

	persistenceLayer.EXPECT().GetWorkerNodeInformation(gomock.Any()).DoAndReturn(func(_ context.Context) ([]*proto.WorkerNodeInformation, error) {
		return make([]*proto.WorkerNodeInformation, 0), nil
	}).Times(1)

	persistenceLayer.EXPECT().GetServiceInformation(gomock.Any()).DoAndReturn(func(_ context.Context) ([]*proto.ServiceInfo, error) {
		return []*proto.ServiceInfo{
			{
				Name:              "1",
				Image:             "",
				PortForwarding:    nil,
				AutoscalingConfig: nil,
			},
			{
				Name:              "2",
				Image:             "",
				PortForwarding:    nil,
				AutoscalingConfig: nil,
			},
			{
				Name:              "3",
				Image:             "",
				PortForwarding:    nil,
				AutoscalingConfig: nil,
			},
			{
				Name:              "4",
				Image:             "",
				PortForwarding:    nil,
				AutoscalingConfig: nil,
			},
			{
				Name:              "5",
				Image:             "",
				PortForwarding:    nil,
				AutoscalingConfig: nil,
			},
		}, nil
	}).Times(1)

	persistenceLayer.EXPECT().GetDataPlaneInformation(gomock.Any()).DoAndReturn(func(_ context.Context) ([]*proto.DataplaneInformation, error) {
		return make([]*proto.DataplaneInformation, 0), nil
	}).Times(1)

	persistenceLayer.EXPECT().SetLeader(gomock.Any()).DoAndReturn(func(_ context.Context) error {
		return nil
	}).Times(1)

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

	start := time.Now()