From 7a5edb3894e31c9c84d8bab1aab681f1ba6e927c Mon Sep 17 00:00:00 2001
From: Maycon Santos <mlsmaycon@gmail.com>
Date: Tue, 1 Jul 2025 23:25:52 +0200
Subject: [PATCH] create accounts with pending onboarding

---
 management/server/account.go      |  4 +++
 management/server/account_test.go | 57 +++++++++++++++++++++++++++++++
 2 files changed, 61 insertions(+)

diff --git a/management/server/account.go b/management/server/account.go
index dd3a72f60..4f3220de0 100644
--- a/management/server/account.go
+++ b/management/server/account.go
@@ -1778,6 +1778,10 @@ func newAccountWithId(ctx context.Context, accountID, userID, domain string) *ty
 			PeerInactivityExpiration:        types.DefaultPeerInactivityExpiration,
 			RoutingPeerDNSResolutionEnabled: true,
 		},
+		Onboarding: types.AccountOnboarding{
+			OnboardingFlowPending: true,
+			SignupFormPending:     true,
+		},
 	}
 
 	if err := acc.AddAllGroup(); err != nil {
diff --git a/management/server/account_test.go b/management/server/account_test.go
index c3b1f31a6..dc21cf71f 100644
--- a/management/server/account_test.go
+++ b/management/server/account_test.go
@@ -3440,3 +3440,60 @@ func TestPropagateUserGroupMemberships(t *testing.T) {
 		}
 	})
 }
+
+func TestDefaultAccountManager_GetAccountOnboarding(t *testing.T) {
+	manager, err := createManager(t)
+	require.NoError(t, err)
+
+	account, err := manager.GetOrCreateAccountByUser(context.Background(), userID, "")
+	require.NoError(t, err)
+
+	onboarding, err := manager.GetAccountOnboarding(context.Background(), account.Id, userID)
+	require.NoError(t, err)
+	require.NotNil(t, onboarding)
+	assert.Equal(t, account.Id, onboarding.AccountID)
+	assert.Equal(t, true, onboarding.OnboardingFlowPending)
+	assert.Equal(t, true, onboarding.SignupFormPending)
+	if onboarding.UpdatedAt.IsZero() {
+		t.Errorf("Onboarding was not retrieved from the store")
+	}
+}
+
+func TestDefaultAccountManager_UpdateAccountOnboarding(t *testing.T) {
+	manager, err := createManager(t)
+	require.NoError(t, err)
+
+	account, err := manager.GetOrCreateAccountByUser(context.Background(), userID, "")
+	require.NoError(t, err)
+
+	onboarding := &types.AccountOnboarding{
+		OnboardingFlowPending: true,
+		SignupFormPending:     true,
+	}
+
+	t.Run("update onboarding with no change", func(t *testing.T) {
+		updated, err := manager.UpdateAccountOnboarding(context.Background(), account.Id, userID, onboarding)
+		require.NoError(t, err)
+		assert.Equal(t, onboarding.OnboardingFlowPending, updated.OnboardingFlowPending)
+		assert.Equal(t, onboarding.SignupFormPending, updated.SignupFormPending)
+		if updated.UpdatedAt.IsZero() {
+			t.Errorf("Onboarding was updated in the store")
+		}
+	})
+
+	onboarding.OnboardingFlowPending = false
+	onboarding.SignupFormPending = false
+
+	t.Run("update onboarding", func(t *testing.T) {
+		updated, err := manager.UpdateAccountOnboarding(context.Background(), account.Id, userID, onboarding)
+		require.NoError(t, err)
+		require.NotNil(t, updated)
+		assert.Equal(t, onboarding.OnboardingFlowPending, updated.OnboardingFlowPending)
+		assert.Equal(t, onboarding.SignupFormPending, updated.SignupFormPending)
+	})
+
+	t.Run("update onboarding with no change", func(t *testing.T) {
+		_, err = manager.UpdateAccountOnboarding(context.Background(), account.Id, userID, nil)
+		require.Error(t, err)
+	})
+}