[Go] สร้าง test context type เพื่อให้โค้ดของ test เรียบง่ายขึ้น

func TestGetUserByUID(t *testing.T) {
	ctx := context.Background()
	fbApp, err := firebase.NewApp(ctx, &firebase.Config{
		ProjectID: "test",
	})
	require.NoError(t, err)

	fbAuth, err := New(ctx, fbApp)
	require.NoError(t, err)
	client := fbAuth.client

	userRecord, err := createUser(ctx, client, "admin@example.com", "password1234", "admin")
	require.NoError(t, err)
	defer func() {
		err := client.DeleteUser(ctx, userRecord.UID)
		require.NoError(t, err)
	}()

	user, err := fbAuth.GetUserByUID(ctx, userRecord.UID)
	require.NoError(t, err)

	expectedUser := &User{
		UID:      userRecord.UID,
		Email:    "admin@example.com",
		Role:     "admin",
		Activate: true,
	}
	require.Equal(t, expectedUser, user)
}

func TestActivateUser(t *testing.T) {
	ctx := context.Background()
	fbApp, err := firebase.NewApp(ctx, &firebase.Config{
		ProjectID: "test",
	})
	require.NoError(t, err)

	fbAuth, err := New(ctx, fbApp)
	require.NoError(t, err)
	client := fbAuth.client

	userRecord, err := createUser(ctx, client, "planner@example.com", "password1234", "planner")
	require.NoError(t, err)
	defer func() {
		err := client.DeleteUser(ctx, userRecord.UID)
		require.NoError(t, err)
	}()

	userRecord, err = client.UpdateUser(ctx, userRecord.UID, (&auth.UserToUpdate{}).Disabled(true))
	require.NoError(t, err)
	require.True(t, userRecord.Disabled)

	user, err := fbAuth.ActivateUser(ctx, userRecord.UID)
	require.NoError(t, err)

	expectedUser := &User{
		UID:      userRecord.UID,
		Email:    "planner@example.com",
		Role:     "planner",
		Activate: true,
	}
	require.Equal(t, expectedUser, user)
}

func TestDeactivateUser(t *testing.T) {
	ctx := context.Background()
	fbApp, err := firebase.NewApp(ctx, &firebase.Config{
		ProjectID: "test",
	})
	require.NoError(t, err)

	fbAuth, err := New(ctx, fbApp)
	require.NoError(t, err)
	client := fbAuth.client

	userRecord, err := createUser(ctx, client, "planner@example.com", "password1234", "planner")
	require.NoError(t, err)
	defer func() {
		err := client.DeleteUser(ctx, userRecord.UID)
		require.NoError(t, err)
	}()
	require.False(t, userRecord.Disabled)

	user, err := fbAuth.DeactivateUser(ctx, userRecord.UID)
	require.NoError(t, err)

	expectedUser := &User{
		UID:      userRecord.UID,
		Email:    "planner@example.com",
		Role:     "planner",
		Activate: false,
	}
	require.Equal(t, expectedUser, user)
}

func createUser(ctx context.Context, client *auth.Client, email, password, role string) (*auth.UserRecord, error) {
	createUserParam := &auth.UserToCreate{}
	createUserParam.Email(email)
	createUserParam.Password(password)
	userRecord, err := client.CreateUser(ctx, createUserParam)
	if err != nil {
		return nil, err
	}

	err = client.SetCustomUserClaims(ctx, userRecord.UID, map[string]any{"role": role})
	if err != nil {
		return nil, err
	}

	return client.GetUser(ctx, userRecord.UID)
}

type testContext struct {
	ctx        context.Context
	fba        *firebaseAuth
	authClient *auth.Client
	t          *testing.T
}

func newTestContext(t *testing.T) *testContext {
	ctx := context.Background()
	fbApp, err := firebase.NewApp(ctx, &firebase.Config{
		ProjectID: "test",
	})
	require.NoError(t, err)

	fbAuth, err := New(ctx, fbApp)
	require.NoError(t, err)

	return &testContext{
		ctx:        ctx,
		authClient: fbAuth.client,
		t:          t,
		fba:        fbAuth,
	}
}

func (testCtx *testContext) createUser(email, password, role string) *auth.UserRecord {
	createUserParam := &auth.UserToCreate{}
	createUserParam.Email(email)
	createUserParam.Password(password)
	userRecord, err := testCtx.authClient.CreateUser(testCtx.ctx, createUserParam)
	require.NoError(testCtx.t, err)

	err = testCtx.authClient.SetCustomUserClaims(testCtx.ctx, userRecord.UID, map[string]any{"role": role})
	require.NoError(testCtx.t, err)

	userRecord, err = testCtx.authClient.GetUser(testCtx.ctx, userRecord.UID)
	require.NoError(testCtx.t, err)

	testCtx.t.Cleanup(func() {
		err := testCtx.authClient.DeleteUser(testCtx.ctx, userRecord.UID)
		require.NoError(testCtx.t, err)
	})
	return userRecord
}

func (testCtx *testContext) createDisabledUser(email, password, role string) *auth.UserRecord {
	user := testCtx.createUser(email, password, role)
	userRecord, err := testCtx.authClient.UpdateUser(testCtx.ctx, user.UID, (&auth.UserToUpdate{}).Disabled(true))
	require.NoError(testCtx.t, err)
	require.True(testCtx.t, userRecord.Disabled)
	return userRecord
}

func TestGetUserByUID(t *testing.T) {
	testCtx := newTestContext(t)
	userRecord := testCtx.createUser("admin@example.com", "password1234", "admin")

	user, err := testCtx.fba.GetUserByUID(testCtx.ctx, userRecord.UID)
	require.NoError(t, err)

	expectedUser := &User{
		UID:      userRecord.UID,
		Email:    "admin@example.com",
		Role:     "admin",
		Activate: true,
	}
	require.Equal(t, expectedUser, user)
}

func TestActivateUser(t *testing.T) {
	testCtx := newTestContext(t)
	userRecord := testCtx.createDisabledUser("planner@example.com", "password1234", "planner")

	user, err := testCtx.fba.ActivateUser(testCtx.ctx, userRecord.UID)
	require.NoError(t, err)

	expectedUser := &User{
		UID:      userRecord.UID,
		Email:    "planner@example.com",
		Role:     "planner",
		Activate: true,
	}
	require.Equal(t, expectedUser, user)
}

func TestDeactivateUser(t *testing.T) {
	testCtx := newTestContext(t)
	userRecord := testCtx.createUser("planner@example.com", "password1234", "planner")

	user, err := testCtx.fba.DeactivateUser(testCtx.ctx, userRecord.UID)
	require.NoError(t, err)

	expectedUser := &User{
		UID:      userRecord.UID,
		Email:    "planner@example.com",
		Role:     "planner",
		Activate: false,
	}
	require.Equal(t, expectedUser, user)
}

// testContext encapsulates the state required to run tests that simulate
// an istio like environment.
type testContext struct {

	testCtx := SetupTestContext(
		context.Background(),
		t,
		TestConfig{
			Clusters: []KubeClusterConfig{{Name: kubeCluster, APIEndpoint: kubeMock.URL}},
		},
	)