client_hint_test.go

package devicedetector

import (
	"net/http"
	"testing"
)

func TestClientHint(t *testing.T) {
	headers := http.Header{}
	headers.Add("sec-ch-ua", `"Opera";v="83", " Not;A Brand";v="99", "Chromium";v="98"`)
	headers.Add("sec-ch-ua-mobile", `?0`)
	headers.Add("sec-ch-ua-platform", "Windows")
	headers.Add("sec-ch-ua-platform-version", "14.0.0")

	cache, err := NewEmbeddedCache()
	if err != nil {
		t.Error("error initialising cache:", err)
	}

	ch := NewClientHint(cache, headers)

	if ch.Platform() != "Windows" {
		t.Errorf("expected Platform() to be 'Windows' but was '%s'", ch.Platform())
	}

	if ch.PlatformVersion() != "14.0.0" {
		t.Errorf("expected PlatformVersion() to be '14.0.0' but was '%s'", ch.PlatformVersion())
	}

	if ch.BrowserName() != "Opera" {
		t.Errorf("expected BrowserName() to return 'Opera' but returned '%s'", ch.BrowserName())
	}

	testBrands := map[string]string{
		"Opera":        "83",
		" Not;A Brand": "99",
		"Chromium":     "98",
	}

	for brand, version := range ch.Brands() {
		if item, ok := testBrands[brand]; !ok {
			t.Errorf("expected '%s' to be in the list of brands", item)
		} else {
			if version != item {
				t.Errorf("expected version to be '%s' but was '%s'", item, version)
			}
		}
	}

	headers = http.Header{}
	headers.Add("HTTP_SEC_CH_UA_FULL_VERSION_LIST", `" Not A;Brand";v="99.0.0.0", "Chromium";v="98.0.4758.82", "Opera";v="98.0.4758.82"`)
	headers.Add("HTTP_SEC_CH_UA", `" Not A;Brand";v="99", "Chromium";v="98", "Opera";v="84"`)
	headers.Add("HTTP_SEC_CH_UA_MOBILE", "?1")
	headers.Add("HTTP_SEC_CH_UA_MODEL", "DN2103")
	headers.Add("HTTP_SEC_CH_UA_PLATFORM", "Ubuntu")
	headers.Add("HTTP_SEC_CH_UA_PLATFORM_VERSION", "3.7")
	headers.Add("HTTP_SEC_CH_UA_FULL_VERSION", "98.0.14335.105")

	ch = NewClientHint(cache, headers)

	if ch.OSName() != "Ubuntu" {
		t.Errorf("expected OSName() to return '%s' but returned '%s'", "Ubuntu", ch.OSName())
	}

	if ch.OSVersion() != "3.7" {
		t.Errorf("expected OSVersion() to return '3.7' but returned '%s'", ch.OSVersion())
	}

	if ch.Model() != "DN2103" {
		t.Errorf("expected Model() to return '%s' but returned '%s'", "DN2103", ch.Model())
	}

	testBrands = map[string]string{
		" Not A;Brand": "99.0.0.0",
		"Chromium":     "98.0.4758.82",
		"Opera":        "98.0.4758.82",
	}

	for brand, version := range ch.Brands() {
		if item, ok := testBrands[brand]; !ok {
			t.Errorf("expected '%s' to be in the list of brands", item)
		} else {
			if version != item {
				t.Errorf("expected version to be '%s' but was '%s'", item, version)
			}
		}
	}
}