satellite/console: remove entitlements on project/account delete

Change-Id: I26525325feb8da0b407052584b1496c945d9e65b

Author: VitaliiShpital

satellite/console/service.go

@@ -2843,6 +2843,14 @@ func (s *Service) handleDeleteProjectStep(ctx context.Context, user *User, proje
 		)
 	}
 
+	err = s.entitlementsService.Projects().DeleteByPublicID(ctx, publicProjectID)
+	if err != nil {
+		s.log.Error("failed to delete project entitlements",
+			zap.String("project_public_id", publicProjectID.String()),
+			zap.Error(err),
+		)
+	}
+
 	// We update status to disabled instead of deleting the project
 	// to not lose the historical project/user usage data.
 	err = s.store.Projects().UpdateStatus(ctx, projectID, ProjectDisabled)
@@ -2899,6 +2907,14 @@ func (s *Service) handleDeleteAccountStep(ctx context.Context, user *User) (err
 			)
 		}
 
+		err = s.entitlementsService.Projects().DeleteByPublicID(ctx, p.PublicID)
+		if err != nil {
+			s.log.Error("failed to delete project entitlements",
+				zap.String("project_public_id", p.PublicID.String()),
+				zap.Error(err),
+			)
+		}
+
 		// We update status to disabled instead of deleting the project
 		// to not lose the historical project/user usage data.
 		err = s.store.Projects().UpdateStatus(ctx, p.ID, ProjectDisabled)

satellite/console/service_test.go

@@ -2698,6 +2698,23 @@ func TestDeleteProject(t *testing.T) {
 		require.NoError(t, err)
 		require.NotNil(t, project)
 
+		scope := entitlements.ConvertPublicIDToProjectScope(project.PublicID)
+		feats := entitlements.ProjectFeatures{NewBucketPlacements: []storj.PlacementConstraint{storj.DefaultPlacement}}
+		featBytes, err := json.Marshal(feats)
+		require.NoError(t, err)
+		require.NotNil(t, featBytes)
+
+		_, err = sat.API.DB.Console().Entitlements().UpsertByScope(ctx, &entitlements.Entitlement{
+			Scope:    scope,
+			Features: featBytes,
+		})
+		require.NoError(t, err)
+
+		entitlement, err := sat.API.DB.Console().Entitlements().GetByScope(ctx, scope)
+		require.NoError(t, err)
+		require.NotNil(t, entitlement)
+		require.NotNil(t, entitlement.Features)
+
 		_, err = sat.DB.Console().Domains().Create(ctx, console.Domain{
 			Subdomain:       "example.test",
 			ProjectID:       p2.ID,
@@ -2720,6 +2737,9 @@ func TestDeleteProject(t *testing.T) {
 		require.NoError(t, err)
 		require.Len(t, domains, 0)
 
+		_, err = sat.API.DB.Console().Entitlements().GetByScope(ctx, scope)
+		require.True(t, entitlements.ErrNotFound.Has(err))
+
 		projects, err := db.Console().Projects().GetOwnActive(ctx, user.ID)
 		require.NoError(t, err)
 		require.Zero(t, len(projects))
@@ -3624,6 +3644,23 @@ func TestDeleteAccount(t *testing.T) {
 			require.NoError(t, err)
 			require.Len(t, cards, 1)
 
+			scope := entitlements.ConvertPublicIDToProjectScope(p2.PublicID)
+			feats := entitlements.ProjectFeatures{NewBucketPlacements: []storj.PlacementConstraint{storj.DefaultPlacement}}
+			featBytes, err := json.Marshal(feats)
+			require.NoError(t, err)
+			require.NotNil(t, featBytes)
+
+			_, err = sat.API.DB.Console().Entitlements().UpsertByScope(ctx, &entitlements.Entitlement{
+				Scope:    scope,
+				Features: featBytes,
+			})
+			require.NoError(t, err)
+
+			entitlement, err := sat.API.DB.Console().Entitlements().GetByScope(ctx, scope)
+			require.NoError(t, err)
+			require.NotNil(t, entitlement)
+			require.NotNil(t, entitlement.Features)
+
 			_, err = sat.DB.Console().Domains().Create(ctx, console.Domain{
 				Subdomain:       "example.test",
 				ProjectID:       p2.ID,
@@ -3649,6 +3686,9 @@ func TestDeleteAccount(t *testing.T) {
 			require.Empty(t, user.ActivationCode)
 			require.Contains(t, user.Email, "deactivated")
 
+			_, err = sat.API.DB.Console().Entitlements().GetByScope(ctx, scope)
+			require.True(t, entitlements.ErrNotFound.Has(err))
+
 			domains, err = sat.DB.Console().Domains().GetAllDomainNamesByProjectID(ctx, p2.ID)
 			require.NoError(t, err)
 			require.Len(t, domains, 0)