[integration-test] allow mark user as unleashed plan

Author: iQQBot

test/pkg/integration/apis.go

@@ -533,6 +533,53 @@ func (c *ComponentAPI) CreateUser(username string, token string) (string, error)
 	return userId, nil
 }
 
+func (c *ComponentAPI) MakeUserUnleashedPlan(username string) error {
+	db, err := c.DB()
+	if err != nil {
+		return err
+	}
+	defer db.Close()
+
+	var userId string
+	err = db.QueryRow(`SELECT id FROM d_b_user WHERE name = ?`, username).Scan(&userId)
+	if err != nil {
+		return err
+	}
+
+	var subId string
+	err = db.QueryRow(`SELECT uid FROM d_b_subscription WHERE userId = ? and planId = ?`, username, "professional-eur").Scan(&subId)
+	if err != nil && !errors.Is(err, sql.ErrNoRows) {
+		return err
+	}
+	if subId != "" {
+		return nil
+	}
+
+	// reset all of this user subscription
+	_, err = db.Exec(`DELETE from d_b_subscription WHERE userId = ?`, userId)
+	if err != nil {
+		return err
+	}
+
+	uid, err := uuid.NewRandom()
+	if err != nil {
+		return err
+	}
+
+	_, err = db.Exec(`INSERT INTO d_b_subscription (uid, userId, startDate, amount, planId) VALUES (?, ?, ?, ?, ?)`,
+		uid,
+		userId,
+		"2022-10-19T00:00:00.000Z",
+		11904,
+		"professional-eur",
+	)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
 func (c *ComponentAPI) createGitpodToken(user string, scopes []string) (tkn string, err error) {
 	id, err := c.GetUserId(user)
 	if err != nil {

test/tests/ide/jetbrains/gateway_test.go

@@ -102,6 +102,10 @@ func TestJetBrainsGatewayWorkspace(t *testing.T) {
 				if err != nil {
 					t.Fatal(err)
 				}
+				err = api.MakeUserUnleashedPlan(username)
+				if err != nil {
+					t.Fatal(err)
+				}
 
 				t.Logf("connecting to server...")
 				server, err := api.GitpodServer(integration.WithGitpodUser(username))