pkg/clean/auth_test.go - develop - githubmirror/photoprism

pkg/clean/auth_test.go
236 строк7 KB
Michael Mayer
AI: Set OpenAI API service key via OPENAI_API_KEY(_FILE) variable #5322
14 ноя 2025, 14:04
14 ноя 2025, 14:04d35dd08
package clean

import (
	"strings"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestAuth(t *testing.T) {
	t.Run("Admin", func(t *testing.T) {
		assert.Equal(t, "Admin", Auth("Admin "))
	})
	t.Run("At", func(t *testing.T) {
		assert.Equal(t, "Admin@foo", Auth(" Admin@foo "))
	})
	t.Run("Spaces", func(t *testing.T) {
		assert.Equal(t, "Admin foo", Auth(" Admin foo "))
	})
	t.Run("Padding", func(t *testing.T) {
		assert.Equal(t, "admin", Auth(" admin "))
	})
	t.Run("Flash", func(t *testing.T) {
		assert.Equal(t, "admin/user", Auth("admin/user"))
	})
	t.Run("Windows", func(t *testing.T) {
		assert.Equal(t, "DOMAIN\\Jens Mander", Auth("DOMAIN\\Jens Mander "))
	})
	t.Run("Empty", func(t *testing.T) {
		assert.Equal(t, "", Auth("  "))
	})
	t.Run("ControlCharacter", func(t *testing.T) {
		assert.Equal(t, "admin!", Auth("admin!"+string(rune(1))))
	})
	t.Run("Clip", func(t *testing.T) {
		assert.Equal(t,
			"a34fd47a7ecd9967a89330a3f92cb55513d5eca79b6c4999dc910818c29d5b9925a3a04ed91a4e57a2c25cbfdab3a751bb8d7f3635092b9242d154f389d9700aa34fd47a7ecd9967a89330a3f92cb55513d5eca79b6c4999dc910818c29d5b9925a3a04ed91a4e57a2c25cbfdab3a751bb8d7f3635092b9242d154f389d9700",
			Auth("a34fd47a7ecd9967a89330a3f92cb55513d5eca79b6c4999dc910818c29d5b9925a3a04ed91a4e57a2c25cbfdab3a751bb8d7f3635092b9242d154f389d9700aa34fd47a7ecd9967a89330a3f92cb55513d5eca79b6c4999dc910818c29d5b9925a3a04ed91a4e57a2c25cbfdab3a751bb8d7f3635092b9242d154f389d9700a"))
	})
	t.Run("Empty", func(t *testing.T) {
		assert.Equal(t, "", Auth(""))
	})
	t.Run("TeLessThanSGreaterThanT", func(t *testing.T) {
		assert.Equal(t, "Test", Auth("Te<s>t"))
	})
	t.Run("ApiKey", func(t *testing.T) {
		assert.Equal(t,
			"ab-prot-keech1aqu8quamiNaecuisuem1ahg7dieph8eitohzo7hoo7pe-Chohzu4eaA-Chohzu4ea-soh7Seic8eig9joojaeshe4Ahsu8zeibooCh9ooquaaleev3poLeev0su9jei2yeich3ahsi9quar1oqueic",
			Auth("ab-prot-keech1aqu8quamiNaecuisuem1ahg7dieph8eitohzo7hoo7pe-Chohzu4eaA-Chohzu4ea-soh7Seic8eig9joojaeshe4Ahsu8zeibooCh9ooquaaleev3poLeev0su9jei2yeich3ahsi9quar1oqueic"),
		)
	})
}

func TestHandle(t *testing.T) {
	t.Run("Admin", func(t *testing.T) {
		assert.Equal(t, "admin", Handle("Admin "))
	})
	t.Run("Admin", func(t *testing.T) {
		assert.Equal(t, "admin", Handle(" Admin@foo "))
	})
	t.Run("Admin", func(t *testing.T) {
		assert.Equal(t, "admin.foo", Handle(" Admin foo "))
	})
	t.Run("Admin", func(t *testing.T) {
		assert.Equal(t, "admin", Handle(" admin "))
	})
	t.Run("AdminUser", func(t *testing.T) {
		assert.Equal(t, "admin.user", Handle("admin/user"))
	})
	t.Run("Windows", func(t *testing.T) {
		assert.Equal(t, "jens.mander", Handle("DOMAIN\\Jens Mander "))
	})
	t.Run("Empty", func(t *testing.T) {
		assert.Equal(t, "", Handle("  "))
	})
	t.Run("ControlCharacter", func(t *testing.T) {
		assert.Equal(t, "admin!", Handle("admin!"+string(rune(1))))
	})
}

func TestUsername(t *testing.T) {
	t.Run("Admin", func(t *testing.T) {
		assert.Equal(t, "admin", Username("Admin "))
	})
	t.Run("Admin", func(t *testing.T) {
		assert.Equal(t, "admin@foo", Username(" Admin@foo "))
	})
	t.Run("Admin", func(t *testing.T) {
		assert.Equal(t, "admin foo", Username(" Admin foo "))
	})
	t.Run("Admin", func(t *testing.T) {
		assert.Equal(t, "admin", Username(" admin "))
	})
	t.Run("AdminUser", func(t *testing.T) {
		assert.Equal(t, "adminuser", Username("admin/user"))
	})
	t.Run("Windows", func(t *testing.T) {
		assert.Equal(t, "domain\\jens mander", Username("DOMAIN\\Jens Mander "))
	})
	t.Run("Empty", func(t *testing.T) {
		assert.Equal(t, "", Username("   "))
	})
	t.Run("ControlCharacter", func(t *testing.T) {
		assert.Equal(t, "admin!", Username("admin!"+string(rune(1))))
	})
}

func TestEmail(t *testing.T) {
	t.Run("ValidExamples", func(t *testing.T) {
		valid := []string{
			"user@example.com",
			"user+news@example.com",
			"user.name@sub-domain.example",
			"user_name@example.co.uk",
			"user@localhost",
			" User@Example.COM ",
		}

		for _, addr := range valid {
			assert.Equal(t, strings.ToLower(strings.TrimSpace(addr)), Email(addr), addr)
		}
	})
	t.Run("Invalid", func(t *testing.T) {
		invalid := []string{
			"userexample.com",
			"user@@example.com",
			"user@",
			"@example.com",
			"user example@example.com",
			"user@-example.com",
			"user@example..com",
		}

		for _, addr := range invalid {
			assert.Equal(t, "", Email(addr), addr)
		}
	})
	t.Run("Empty", func(t *testing.T) {
		assert.Equal(t, "", Email(""))
	})
}

func TestDomain(t *testing.T) {
	t.Run("Valid", func(t *testing.T) {
		assert.Equal(t, "photoprism.app", Domain("photoprism.app"))
	})
	t.Run("Whitespace", func(t *testing.T) {
		assert.Equal(t, "photoprism.app", Domain(" photoprism.app "))
	})
	t.Run("Hostname", func(t *testing.T) {
		assert.Equal(t, "foo.example.com", Domain(" FOO.example.Com   "))
	})
	t.Run("Example", func(t *testing.T) {
		assert.Equal(t, "", Domain("example"))
	})
	t.Run("Invalid", func(t *testing.T) {
		assert.Equal(t, "", Domain(" hello-photoprism "))
	})
	t.Run("Empty", func(t *testing.T) {
		assert.Equal(t, "", Domain(""))
	})
	t.Run("Match", func(t *testing.T) {
		email := "john.doe@example.com"
		domain := Domain("example.com")

		_, emailDomain, _ := strings.Cut(Email(email), "@")

		assert.True(t, strings.HasSuffix("."+emailDomain, "."+domain))
		assert.False(t, strings.HasSuffix(".my-"+emailDomain, "."+domain))
		assert.True(t, strings.HasSuffix("my-"+emailDomain, domain))
	})
}

func TestRole(t *testing.T) {
	t.Run("Admin", func(t *testing.T) {
		assert.Equal(t, "admin", Role("Admin "))
	})
	t.Run("Admin", func(t *testing.T) {
		assert.Equal(t, "admin", Role(" Admin "))
	})
	t.Run("Admin", func(t *testing.T) {
		assert.Equal(t, "admin", Role(" admin "))
	})
	t.Run("AdmIn", func(t *testing.T) {
		assert.Equal(t, "admin", Role("adm}in"))
	})
	t.Run("Empty", func(t *testing.T) {
		assert.Equal(t, "", Role(""))
	})
}

func TestFlags(t *testing.T) {
	t.Run("Empty", func(t *testing.T) {
		s := ""
		assert.Equal(t, s, Attr(s))
	})
	t.Run("SlackScope", func(t *testing.T) {
		s := "admin.conversations.removeCustomRetention admin.usergroups:read"
		assert.Equal(t, s, Attr(s))
	})
	t.Run("Random", func(t *testing.T) {
		s := "  admin.conversations.removeCustomRetention   admin.usergroups:read  me:yes FOOt0-2U	6VU #$#%$ cm,Nu"
		cleaned := "6VU FOOt0-2U admin.conversations.removeCustomRetention admin.usergroups:read cmNu me"
		assert.Equal(t, cleaned, Attr(s))
	})
}

func TestPassword(t *testing.T) {
	t.Run("Alnum", func(t *testing.T) {
		assert.Equal(t, "fgdg5yw4y", Password("fgdg5yw4y "))
	})
	t.Run("Upper", func(t *testing.T) {
		assert.Equal(t, "AABDF24245vgfrg", Password(" AABDF24245vgfrg "))
	})
	t.Run("Special", func(t *testing.T) {
		assert.Equal(t, "!#$T#)$%I#J$I", Password("!#$T#)$%I#J$I"))
	})
}

func TestPasscode(t *testing.T) {
	t.Run("Alnum", func(t *testing.T) {
		assert.Equal(t, "fgdg5yw4y", Passcode("fgdg5yw4y "))
	})
	t.Run("Upper", func(t *testing.T) {
		assert.Equal(t, "aabdf24245vgfrg", Passcode(" AABDF24245vgfrg "))
	})
	t.Run("Special", func(t *testing.T) {
		assert.Equal(t, "tiji", Passcode("!#$T#)$%I#J$I"))
	})
	t.Run("Empty", func(t *testing.T) {
		assert.Equal(t, "", Passcode(""))
	})
	t.Run("Space", func(t *testing.T) {
		assert.Equal(t, "", Passcode("    	"))
	})
}