package headscale import ( "sync" "testing" "github.com/coreos/go-oidc/v3/oidc" "github.com/patrickmn/go-cache" "golang.org/x/oauth2" "gorm.io/gorm" "tailscale.com/tailcfg" "tailscale.com/types/wgkey" ) func TestHeadscale_getNamespaceFromEmail(t *testing.T) { type fields struct { cfg Config db *gorm.DB dbString string dbType string dbDebug bool publicKey *wgkey.Key privateKey *wgkey.Private aclPolicy *ACLPolicy aclRules []tailcfg.FilterRule lastStateChange sync.Map oidcProvider *oidc.Provider oauth2Config *oauth2.Config oidcStateCache *cache.Cache } type args struct { email string } tests := []struct { name string fields fields args args want string want1 bool }{ { name: "match all", fields: fields{ cfg: Config{ OIDC: OIDCConfig{ MatchMap: map[string]string{ ".*": "space", }, }, }, }, args: args{ email: "test@example.no", }, want: "space", want1: true, }, { name: "match user", fields: fields{ cfg: Config{ OIDC: OIDCConfig{ MatchMap: map[string]string{ "specific@user\\.no": "user-namespace", }, }, }, }, args: args{ email: "specific@user.no", }, want: "user-namespace", want1: true, }, { name: "match domain", fields: fields{ cfg: Config{ OIDC: OIDCConfig{ MatchMap: map[string]string{ ".*@example\\.no": "example", }, }, }, }, args: args{ email: "test@example.no", }, want: "example", want1: true, }, { name: "multi match domain", fields: fields{ cfg: Config{ OIDC: OIDCConfig{ MatchMap: map[string]string{ ".*@example\\.no": "exammple", ".*@gmail\\.com": "gmail", }, }, }, }, args: args{ email: "someuser@gmail.com", }, want: "gmail", want1: true, }, { name: "no match domain", fields: fields{ cfg: Config{ OIDC: OIDCConfig{ MatchMap: map[string]string{ ".*@dontknow.no": "never", }, }, }, }, args: args{ email: "test@wedontknow.no", }, want: "", want1: false, }, { name: "multi no match domain", fields: fields{ cfg: Config{ OIDC: OIDCConfig{ MatchMap: map[string]string{ ".*@dontknow.no": "never", ".*@wedontknow.no": "other", ".*\\.no": "stuffy", }, }, }, }, args: args{ email: "tasy@nonofthem.com", }, want: "", want1: false, }, } //nolint for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { h := &Headscale{ cfg: tt.fields.cfg, db: tt.fields.db, dbString: tt.fields.dbString, dbType: tt.fields.dbType, dbDebug: tt.fields.dbDebug, publicKey: tt.fields.publicKey, privateKey: tt.fields.privateKey, aclPolicy: tt.fields.aclPolicy, aclRules: tt.fields.aclRules, lastStateChange: tt.fields.lastStateChange, oidcProvider: tt.fields.oidcProvider, oauth2Config: tt.fields.oauth2Config, oidcStateCache: tt.fields.oidcStateCache, } got, got1 := h.getNamespaceFromEmail(tt.args.email) if got != tt.want { t.Errorf("Headscale.getNamespaceFromEmail() got = %v, want %v", got, tt.want) } if got1 != tt.want1 { t.Errorf("Headscale.getNamespaceFromEmail() got1 = %v, want %v", got1, tt.want1) } }) } }