Function TestDeleteUser

backend/tests/user_test.go:19–126

(t *testing.T)

)

func TestDeleteUser(t *testing.T) {
	a := require.New(t)
	ctx := context.Background()
	ctl := &controller{}
	ctx, err := ctl.StartServerWithExternalPg(ctx)
	a.NoError(err)
	defer ctl.Close(ctx)

	expectErrorMsg := "workspace must have at least one admin"

	memberResp, err := ctl.userServiceClient.CreateUser(ctx, connect.NewRequest(&v1pb.CreateUserRequest{
		User: &v1pb.User{
			Title:    "member",
			Email:    "member@bytebase.com",
			Password: "1024bytebase",
		},
	}))
	a.NoError(err)
	member := memberResp.Msg

	_, err = ctl.userServiceClient.DeleteUser(ctx, connect.NewRequest(&v1pb.DeleteUserRequest{
		Name: member.Name,
	}))
	a.NoError(err)

	// Test: cannot delete the last admin.
	_, err = ctl.userServiceClient.DeleteUser(ctx, connect.NewRequest(&v1pb.DeleteUserRequest{
		Name: ctl.principalName,
	}))
	a.Error(err)
	a.ErrorContains(err, expectErrorMsg)

	actuator, err := ctl.actuatorServiceClient.GetActuatorInfo(ctx, connect.NewRequest(&v1pb.GetActuatorInfoRequest{
		Name: memberResp.Msg.Workspace,
	}))
	a.NoError(err)

	serviceAccountResp, err := ctl.serviceAccountServiceClient.CreateServiceAccount(ctx, connect.NewRequest(&v1pb.CreateServiceAccountRequest{
		Parent:           actuator.Msg.Workspace,
		ServiceAccountId: "bot",
		ServiceAccount: &v1pb.ServiceAccount{
			Title: "bot",
		},
	}))
	a.NoError(err)
	serviceAccount := serviceAccountResp.Msg

	_, err = ctl.addMemberToWorkspaceIAM(ctx, actuator.Msg.Workspace, fmt.Sprintf("serviceAccount:%v", serviceAccount.Email), "roles/workspaceAdmin")
	a.NoError(err)

	_, err = ctl.userServiceClient.DeleteUser(ctx, connect.NewRequest(&v1pb.DeleteUserRequest{
		Name: ctl.principalName,
	}))
	a.Error(err)
	a.ErrorContains(err, expectErrorMsg)

	_, err = ctl.userServiceClient.UndeleteUser(ctx, connect.NewRequest(&v1pb.UndeleteUserRequest{
		Name: member.Name,