package main

import (
	"log"
	"testing"
)

func TestCartTcpHelpers(t *testing.T) {

	server, err := CartListen("localhost:51337")
	if err != nil {
		t.Errorf("Error listening: %v\n", err)
	}
	client, err := CartDial("localhost:51337")
	if err != nil {
		t.Errorf("Error dialing: %v\n", err)
	}
	var messageData string
	server.ListenFor(1, func(id CartId, data []byte) error {
		log.Printf("Received message: %s\n", string(data))
		messageData = string(data)
		return nil
	})
	server.HandleCall(2, func(id CartId, data []byte) (uint32, []byte, error) {
		log.Printf("Received call: %s\n", string(data))
		return 3, []byte("Hello, client!"), nil
	})
	server.HandleCall(Ping, func(id CartId, data []byte) (uint32, []byte, error) {
		return Pong, nil, nil
	})
	id := ToCartId("kalle")
	client.SendPacket(1, id, []byte("Hello, world!"))
	answer, err := client.Call(2, id, 3, []byte("Hello, server!"))
	if err != nil {
		t.Errorf("Error calling: %v\n", err)
	}
	for i := 0; i < 100; i++ {
		_, err = client.Call(Ping, id, Pong, nil)
		if err != nil {
			t.Errorf("Error calling: %v\n", err)
		}
	}
	_, err = client.Call(Ping, id, Pong, nil)
	if err != nil {
		t.Errorf("Error calling: %v\n", err)
	}
	if string(answer) != "Hello, client!" {
		t.Errorf("Expected answer 'Hello, client!', got %s\n", string(answer))
	}
	if messageData != "Hello, world!" {
		t.Errorf("Expected message 'Hello, world!', got %s\n", messageData)
	}
}