chore: move the go code into a subfolder

2025-08-12 03:47:33 +00:00 · 2025-08-05 15:20:32 -07:00
parent 4ad22ba456
commit cd2921de26
2978 changed files with 373 additions and 300 deletions
--- a/apps/api/internal/query/domain_policy_test.go
+++ b/apps/api/internal/query/domain_policy_test.go
@@ -0,0 +1,127 @@
+package query
+
+import (
+	"database/sql"
+	"database/sql/driver"
+	"errors"
+	"fmt"
+	"regexp"
+	"testing"
+
+	"github.com/zitadel/zitadel/internal/domain"
+	"github.com/zitadel/zitadel/internal/zerrors"
+)
+
+var (
+	prepareDomainPolicyStmt = `SELECT projections.domain_policies2.id,` +
+		` projections.domain_policies2.sequence,` +
+		` projections.domain_policies2.creation_date,` +
+		` projections.domain_policies2.change_date,` +
+		` projections.domain_policies2.resource_owner,` +
+		` projections.domain_policies2.user_login_must_be_domain,` +
+		` projections.domain_policies2.validate_org_domains,` +
+		` projections.domain_policies2.smtp_sender_address_matches_instance_domain,` +
+		` projections.domain_policies2.is_default,` +
+		` projections.domain_policies2.state` +
+		` FROM projections.domain_policies2`
+	prepareDomainPolicyCols = []string{
+		"id",
+		"sequence",
+		"creation_date",
+		"change_date",
+		"resource_owner",
+		"user_login_must_be_domain",
+		"validate_org_domains",
+		"smtp_sender_address_matches_instance_domain",
+		"is_default",
+		"state",
+	}
+)
+
+func Test_DomainPolicyPrepares(t *testing.T) {
+	type want struct {
+		sqlExpectations sqlExpectation
+		err             checkErr
+	}
+	tests := []struct {
+		name    string
+		prepare interface{}
+		want    want
+		object  interface{}
+	}{
+		{
+			name:    "prepareDomainPolicyQuery no result",
+			prepare: prepareDomainPolicyQuery,
+			want: want{
+				sqlExpectations: mockQueriesScanErr(
+					regexp.QuoteMeta(prepareDomainPolicyStmt),
+					nil,
+					nil,
+				),
+				err: func(err error) (error, bool) {
+					if !zerrors.IsNotFound(err) {
+						return fmt.Errorf("err should be zitadel.NotFoundError got: %w", err), false
+					}
+					return nil, true
+				},
+			},
+			object: (*DomainPolicy)(nil),
+		},
+		{
+			name:    "prepareDomainPolicyQuery found",
+			prepare: prepareDomainPolicyQuery,
+			want: want{
+				sqlExpectations: mockQuery(
+					regexp.QuoteMeta(prepareDomainPolicyStmt),
+					prepareDomainPolicyCols,
+					[]driver.Value{
+						"pol-id",
+						uint64(20211109),
+						testNow,
+						testNow,
+						"ro",
+						true,
+						true,
+						true,
+						true,
+						domain.PolicyStateActive,
+					},
+				),
+			},
+			object: &DomainPolicy{
+				ID:                                     "pol-id",
+				CreationDate:                           testNow,
+				ChangeDate:                             testNow,
+				Sequence:                               20211109,
+				ResourceOwner:                          "ro",
+				State:                                  domain.PolicyStateActive,
+				UserLoginMustBeDomain:                  true,
+				ValidateOrgDomains:                     true,
+				SMTPSenderAddressMatchesInstanceDomain: true,
+				IsDefault:                              true,
+			},
+		},
+		{
+			name:    "prepareDomainPolicyQuery sql err",
+			prepare: prepareDomainPolicyQuery,
+			want: want{
+				sqlExpectations: mockQueryErr(
+					regexp.QuoteMeta(prepareDomainPolicyStmt),
+					sql.ErrConnDone,
+				),
+				err: func(err error) (error, bool) {
+					if !errors.Is(err, sql.ErrConnDone) {
+						return fmt.Errorf("err should be sql.ErrConnDone got: %w", err), false
+					}
+					return nil, true
+				},
+			},
+			object: (*DomainPolicy)(nil),
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			assertPrepare(t, tt.prepare, tt.object, tt.want.sqlExpectations, tt.want.err)
+		})
+	}
+}