package httpprobe

import "testing"

func TestIsHttpMethods(t *testing.T) {
	cases := map[string][]byte{
		"GET":     []byte("GET /index.html HTTP/1.1\r\nHost: a.com\r\n\r\n"),
		"POST":    []byte("POST /api HTTP/1.1\r\nHost: a.com\r\n\r\n{}"),
		"PUT":     []byte("PUT /api HTTP/1.1\r\nHost: a.com\r\n\r\n"),
		"HEAD":    []byte("HEAD / HTTP/1.1\r\nHost: a.com\r\n\r\n"),
		"DELETE":  []byte("DELETE /api HTTP/1.1\r\nHost: a.com\r\n\r\n"),
		"CONNECT": []byte("CONNECT a.com:443 HTTP/1.1\r\n\r\n"),
		"OPTIONS": []byte("OPTIONS / HTTP/1.1\r\nHost: a.com\r\n\r\n"),
		"TRACE":   []byte("TRACE / HTTP/1.1\r\nHost: a.com\r\n\r\n"),
	}
	for method, data := range cases {
		if !IsHttp(data) {
			t.Errorf("IsHttp(%s...) 应为 true", method)
		}
	}
}

func TestIsHttpNegative(t *testing.T) {
	if IsHttp([]byte("hello world! This is plain data")) {
		t.Error("普通文本不应识别为 HTTP")
	}
	if IsHttp(nil) {
		t.Error("空数据不应识别为 HTTP")
	}
	if IsHttp([]byte("GET")) {
		t.Error("短于 8 字节不应识别为 HTTP")
	}
	// TLS 握手不是 HTTP
	tls := []byte{0x16, 0x03, 0x00, 0x00, 0x01, 0x02, 0x03, 0x04, 0x05}
	if IsHttp(tls) {
		t.Error("TLS 握手不应识别为 HTTP")
	}
}

func TestIsHttpCaseInsensitive(t *testing.T) {
	if !IsHttp([]byte("get / HTTP/1.1")) {
		t.Error("IsHttp 应大小写不敏感")
	}
	if !IsHttp([]byte("trace / HTTP/1.1")) {
		t.Error("小写 trace 应识别为 HTTP")
	}
}

func TestIsHttps(t *testing.T) {
	data := make([]byte, 64)
	data[0] = 0x16
	data[1] = 0x03
	data[3] = 0x01
	copy(data[40:], "http")
	if !IsHttps(data) {
		t.Error("TLS ClientHello 应识别为 HTTPS")
	}
}

func TestIsHttpsNegative(t *testing.T) {
	if IsHttps([]byte("hello world")) {
		t.Error("普通文本不应识别为 HTTPS")
	}
	if IsHttps(nil) {
		t.Error("空数据不应识别为 HTTPS")
	}
	// 非 TLS 握手类型（第一字节不是 0x16）
	wrong := make([]byte, 16)
	wrong[0] = 0x17
	wrong[1] = 0x03
	if IsHttps(wrong) {
		t.Error("非握手记录不应识别为 HTTPS")
	}
}