github.com/cozy/cozy-stack@v0.0.0-20240327093429-939e4a21320e/model/instance/instance_test.go

package instance_test

import (
	"encoding/json"
	"testing"

	"github.com/cozy/cozy-stack/model/instance"
	"github.com/cozy/cozy-stack/pkg/config/config"
	"github.com/cozy/cozy-stack/pkg/crypto"
	jwt "github.com/golang-jwt/jwt/v5"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestInstance(t *testing.T) {
	config.UseTestFile(t)

	t.Run("Subdomain", func(t *testing.T) {
		inst := &instance.Instance{
			Domain: "foo.example.com",
		}
		cfg := config.GetConfig()
		was := cfg.Subdomains
		defer func() { cfg.Subdomains = was }()

		cfg.Subdomains = config.NestedSubdomains
		u := inst.SubDomain("calendar")
		assert.Equal(t, "https://calendar.foo.example.com/", u.String())

		cfg.Subdomains = config.FlatSubdomains
		u = inst.SubDomain("calendar")
		assert.Equal(t, "https://foo-calendar.example.com/", u.String())
	})

	t.Run("BuildAppToken", func(t *testing.T) {
		inst := &instance.Instance{
			Domain:     "test-ctx-token.example.com",
			SessSecret: crypto.GenerateRandomBytes(64),
		}

		tokenString := inst.BuildAppToken("my-app", "sessionid")
		token, err := jwt.Parse(tokenString, func(token *jwt.Token) (interface{}, error) {
			_, ok := token.Method.(*jwt.SigningMethodHMAC)
			assert.True(t, ok, "The signing method should be HMAC")
			return inst.SessionSecret(), nil
		})
		assert.NoError(t, err)
		assert.True(t, token.Valid)

		claims, ok := token.Claims.(jwt.MapClaims)
		assert.True(t, ok, "Claims can be parsed as standard claims")
		assert.Equal(t, []interface{}{"app"}, claims["aud"])
		assert.Equal(t, "test-ctx-token.example.com", claims["iss"])
		assert.Equal(t, "my-app", claims["sub"])
	})

	t.Run("GetContextWithSponsorships", func(t *testing.T) {
		cfg := config.GetConfig()
		was := cfg.Contexts
		defer func() { cfg.Contexts = was }()

		cfg.Contexts = map[string]interface{}{
			"context": map[string]interface{}{
				"manager_url": "http://manager.example.org",
				"logos": map[string]interface{}{
					"coachco2": map[string]interface{}{
						"light": []interface{}{
							map[string]interface{}{"src": "/logos/main_cozy.png", "alt": "Cozy Cloud"},
						},
					},
					"home": map[string]interface{}{
						"light": []interface{}{
							map[string]interface{}{"src": "/logos/main_cozy.png", "alt": "Cozy Cloud", "type": "main"},
							map[string]interface{}{"src": "/logos/partner1.png", "alt": "Partner1", "type": "secondary"},
						},
						"dark": []interface{}{
							// no main
							map[string]interface{}{"src": "/logos/partner1.png", "alt": "Partner1", "type": "secondary"},
						},
					},
				},
			},
			"sponsor1": map[string]interface{}{
				"move_url": "http://move.cozy.beta/",
				"logos": map[string]interface{}{
					"coachco2": map[string]interface{}{
						"dark": []interface{}{
							map[string]interface{}{"src": "/logos/main_cozy.png", "alt": "Cozy Cloud"},
						},
					},
					"home": map[string]interface{}{
						"light": []interface{}{
							map[string]interface{}{"src": "/logos/main_cozy.png", "alt": "Cozy Cloud", "type": "main"},
							map[string]interface{}{"src": "/logos/partner1.png", "alt": "Partner1", "type": "secondary"},
							map[string]interface{}{"src": "/logos/partner2.png", "alt": "Partner2", "type": "secondary"},
						},
						"dark": []interface{}{
							map[string]interface{}{"src": "/logos/main_cozy.png", "alt": "Cozy Cloud", "type": "main"},
							map[string]interface{}{"src": "/logos/partner2.png", "alt": "Partner2"},
							map[string]interface{}{"src": "/logos/partner1.png", "alt": "Partner1"},
						},
					},
				},
			},
			"sponsor2": map[string]interface{}{
				"logos": map[string]interface{}{
					"mespapiers": map[string]interface{}{
						"dark": []interface{}{
							map[string]interface{}{"src": "/logos/main_cozy.png", "alt": "Cozy Cloud"},
						},
					},
					"home": map[string]interface{}{
						"light": []interface{}{
							map[string]interface{}{"src": "/logos/main_cozy.png", "alt": "Cozy Cloud", "type": "main"},
							map[string]interface{}{"src": "/logos/partner3.png", "alt": "Partner3", "type": "secondary"},
							map[string]interface{}{"src": "/logos/partner2.png", "alt": "Partner2", "type": "secondary"},
						},
					},
				},
			},
		}

		inst := &instance.Instance{
			Domain:       "foo.example.com",
			ContextName:  "context",
			Sponsorships: []string{"sponsor1", "sponsor2"},
		}
		result := inst.GetContextWithSponsorships()
		bytes, err := json.MarshalIndent(result, "", "  ")
		require.NoError(t, err)
		expected := `{
  "logos": {
    "coachco2": {
      "light": [
        {
          "src": "/logos/main_cozy.png",
          "alt": "Cozy Cloud"
        }
      ]
    },
    "home": {
      "light": [
        {
          "src": "/logos/main_cozy.png",
          "alt": "Cozy Cloud",
          "type": "main"
        },
        {
          "src": "/logos/partner1.png",
          "alt": "Partner1",
          "type": "secondary"
        },
        {
          "src": "/ext/sponsor1/logos/partner2.png",
          "alt": "Partner2",
          "type": "secondary"
        },
        {
          "src": "/ext/sponsor2/logos/partner3.png",
          "alt": "Partner3",
          "type": "secondary"
        }
      ],
      "dark": [
        {
          "src": "/logos/partner1.png",
          "alt": "Partner1",
          "type": "secondary"
        },
        {
          "src": "/ext/sponsor1/logos/partner2.png",
          "alt": "Partner2"
        }
      ]
    }
  },
  "manager_url": "http://manager.example.org"
}`
		assert.Equal(t, expected, string(bytes))
	})
}