package client

import (
	"testing"
	"time"

	"github.com/noahlann/nnet/pkg/client"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestTCPClientNew(t *testing.T) {
	config := &client.Config{
		Addr: "tcp://127.0.0.1:0",
	}

	cli := NewTCPClient(config)
	require.NotNil(t, cli, "Expected client to be non-nil")
}

func TestTCPClientNewWithNilConfig(t *testing.T) {
	// NewTCPClient接受nil配置，会使用默认配置
	cli := NewTCPClient(nil)
	require.NotNil(t, cli, "Expected client to be non-nil")
}

func TestTCPClientConnect(t *testing.T) {
	t.Skip("Skipping connect test - requires actual server")

	// 这个测试需要实际的服务器连接
	// 可以放在集成测试中
}

func TestTCPClientIsConnected(t *testing.T) {
	config := &client.Config{
		Addr:         "tcp://127.0.0.1:6995",
		ReadTimeout:  5 * time.Second,
		WriteTimeout: 5 * time.Second,
	}

	cli := NewTCPClient(config)
	require.NotNil(t, cli, "Expected client to be non-nil")

	// 测试初始连接状态
	assert.False(t, cli.IsConnected(), "Expected client to not be connected initially")
}

func TestTCPClientWriteRead(t *testing.T) {
	t.Skip("Skipping write/read test - requires actual server connection")

	// 这个测试需要实际的服务器连接
	// 可以放在集成测试中
}

func TestTCPClientClose(t *testing.T) {
	config := &client.Config{
		Addr: "tcp://127.0.0.1:6995",
	}

	cli := NewTCPClient(config)
	require.NotNil(t, cli)

	// 测试关闭未连接的客户端（应该不报错）
	err := cli.Close()
	require.NoError(t, err, "Expected no error when closing unconnected client")

	// 验证客户端已关闭
	assert.False(t, cli.IsConnected(), "Expected client to not be connected after close")
}

func TestTCPClientDisconnect(t *testing.T) {
	config := &client.Config{
		Addr: "tcp://127.0.0.1:6995",
	}

	cli := NewTCPClient(config)
	require.NotNil(t, cli)

	// 测试断开未连接的客户端（应该不报错）
	err := cli.Disconnect()
	require.NoError(t, err, "Expected no error when disconnecting unconnected client")
}

func TestTCPClientSendNotConnected(t *testing.T) {
	config := &client.Config{
		Addr: "tcp://127.0.0.1:6995",
	}

	cli := NewTCPClient(config)
	require.NotNil(t, cli)

	// 测试未连接时发送数据（应该返回错误）
	err := cli.Send([]byte("test"))
	assert.Error(t, err, "Expected error when sending without connection")
}

func TestTCPClientReceiveNotConnected(t *testing.T) {
	config := &client.Config{
		Addr: "tcp://127.0.0.1:6995",
	}

	cli := NewTCPClient(config)
	require.NotNil(t, cli)

	// 测试未连接时接收数据（应该返回错误）
	_, err := cli.Receive()
	assert.Error(t, err, "Expected error when receiving without connection")
}

func TestTCPClientReconnect(t *testing.T) {
	t.Skip("Skipping reconnect test - requires actual server connection")

	// 这个测试需要实际的服务器连接和重连逻辑
}