package main

import "testing"

func TestIsCLI(t *testing.T) {
	userAgents := []string{"curl/7.26.0", "Wget/1.13.4 (linux-gnu)",
		"fetch libfetch/2.0"}

	for _, userAgent := range userAgents {
		if !isCLI(userAgent) {
			t.Errorf("Expected true for %s", userAgent)
		}
	}

	browserUserAgent := "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_8_4) " +
		"AppleWebKit/537.36 (KHTML, like Gecko) Chrome/30.0.1599.28 " +
		"Safari/537.36"
	if isCLI(browserUserAgent) {
		t.Errorf("Expected false for %s", browserUserAgent)
	}
}

func TestPathToKey(t *testing.T) {
	if key := pathToKey("/ip"); key != "ip" {
		t.Fatalf("Expected 'ip', got '%s'", key)
	}
	if key := pathToKey("/User-Agent"); key != "user-agent" {
		t.Fatalf("Expected 'user-agent', got '%s'", key)
	}
	if key := pathToKey("/all.json"); key != "all" {
		t.Fatalf("Expected 'all', got '%s'", key)
	}
}