package identify

import (
	"context"
	"testing"

	"github.com/libp2p/go-libp2p-core/network"
	"github.com/libp2p/go-libp2p-core/peer"

	blhost "github.com/libp2p/go-libp2p-blankhost"
	swarmt "github.com/libp2p/go-libp2p-swarm/testing"
	pb "github.com/libp2p/go-libp2p/p2p/protocol/identify/pb"

	"github.com/stretchr/testify/require"
)

func doeval(t *testing.T, ph *peerHandler, f func()) {
	done := make(chan struct{}, 1)
	ph.evalTestCh <- func() {
		f()
		done <- struct{}{}
	}
	<-done
}

func TestMakeApplyDelta(t *testing.T) {
	isTesting = true
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

	h1 := blhost.NewBlankHost(swarmt.GenSwarm(t, ctx))
	defer h1.Close()
	ids1 := NewIDService(h1)
	ph := newPeerHandler(h1.ID(), ids1, &pb.Identify{})
	ph.start()
	defer ph.close()

	m1 := ph.mkDelta()
	require.NotNil(t, m1)
	// all the Id protocols must have been added
	require.NotEmpty(t, m1.AddedProtocols)
	doeval(t, ph, func() {
		ph.applyDelta(m1)
	})

	h1.SetStreamHandler("p1", func(network.Stream) {})
	m2 := ph.mkDelta()
	require.Len(t, m2.AddedProtocols, 1)
	require.Contains(t, m2.AddedProtocols, "p1")
	require.Empty(t, m2.RmProtocols)
	doeval(t, ph, func() {
		ph.applyDelta(m2)
	})

	h1.SetStreamHandler("p2", func(network.Stream) {})
	h1.SetStreamHandler("p3", func(stream network.Stream) {})
	m3 := ph.mkDelta()
	require.Len(t, m3.AddedProtocols, 2)
	require.Contains(t, m3.AddedProtocols, "p2")
	require.Contains(t, m3.AddedProtocols, "p3")
	require.Empty(t, m3.RmProtocols)
	doeval(t, ph, func() {
		ph.applyDelta(m3)
	})

	h1.RemoveStreamHandler("p3")
	m4 := ph.mkDelta()
	require.Empty(t, m4.AddedProtocols)
	require.Len(t, m4.RmProtocols, 1)
	require.Contains(t, m4.RmProtocols, "p3")
	doeval(t, ph, func() {
		ph.applyDelta(m4)
	})

	h1.RemoveStreamHandler("p2")
	h1.RemoveStreamHandler("p1")
	m5 := ph.mkDelta()
	require.Empty(t, m5.AddedProtocols)
	require.Len(t, m5.RmProtocols, 2)
	require.Contains(t, m5.RmProtocols, "p2")
	require.Contains(t, m5.RmProtocols, "p1")
}

func TestHandlerClose(t *testing.T) {
	isTesting = true
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

	h1 := blhost.NewBlankHost(swarmt.GenSwarm(t, ctx))
	defer h1.Close()
	ids1 := NewIDService(h1)
	ph := newPeerHandler(h1.ID(), ids1, nil)
	ph.start()

	require.NoError(t, ph.close())
}

func TestPeerSupportsProto(t *testing.T) {
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

	h1 := blhost.NewBlankHost(swarmt.GenSwarm(t, ctx))
	defer h1.Close()
	ids1 := NewIDService(h1)

	rp := peer.ID("test")
	ph := newPeerHandler(rp, ids1, nil)
	require.NoError(t, h1.Peerstore().AddProtocols(rp, "test"))
	require.True(t, ph.peerSupportsProtos([]string{"test"}))
	require.False(t, ph.peerSupportsProtos([]string{"random"}))

	// remove support for protocol and check
	require.NoError(t, h1.Peerstore().RemoveProtocols(rp, "test"))
	require.False(t, ph.peerSupportsProtos([]string{"test"}))
}