// Copyright 2026 The frp Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package registry

import (
	"testing"
	"time"

	clocktesting "k8s.io/utils/clock/testing"

	"kanhole/pkg/proto/wire"
)

func TestClientRegistryRegisterStoresWireProtocol(t *testing.T) {
	registry := NewClientRegistry()
	key, conflict := registry.Register("user", "client-id", "run-id", "host", "1.0.0", "127.0.0.1", wire.ProtocolV2)
	if conflict {
		t.Fatal("unexpected client conflict")
	}

	info, ok := registry.GetByKey(key)
	if !ok {
		t.Fatalf("client %q not found", key)
	}
	if info.WireProtocol != wire.ProtocolV2 {
		t.Fatalf("wire protocol mismatch, want %q got %q", wire.ProtocolV2, info.WireProtocol)
	}
}

func TestClientRegistryUsesClockForTimestamps(t *testing.T) {
	start := time.Date(2026, time.May, 8, 12, 30, 0, 0, time.UTC)
	clk := clocktesting.NewFakeClock(start)
	registry := newClientRegistryWithClock(clk)

	key, conflict := registry.Register("user", "client-id", "run-id", "host", "1.0.0", "127.0.0.1", wire.ProtocolV2)
	if conflict {
		t.Fatal("unexpected client conflict")
	}

	info, ok := registry.GetByKey(key)
	if !ok {
		t.Fatalf("client %q not found", key)
	}
	if !info.FirstConnectedAt.Equal(start) {
		t.Fatalf("first connected time mismatch, want %s got %s", start, info.FirstConnectedAt)
	}
	if !info.LastConnectedAt.Equal(start) {
		t.Fatalf("last connected time mismatch, want %s got %s", start, info.LastConnectedAt)
	}

	disconnectedAt := start.Add(time.Minute)
	clk.SetTime(disconnectedAt)
	registry.MarkOfflineByRunID("run-id")

	info, ok = registry.GetByKey(key)
	if !ok {
		t.Fatalf("client %q not found after disconnect", key)
	}
	if !info.DisconnectedAt.Equal(disconnectedAt) {
		t.Fatalf("disconnected time mismatch, want %s got %s", disconnectedAt, info.DisconnectedAt)
	}
}