cwtch/peer/connections/peerpeerconnection_test.go

85 lines
2.3 KiB
Go
Raw Normal View History

package connections
import (
2018-10-05 03:18:34 +00:00
"crypto/rand"
2018-05-28 18:05:06 +00:00
"cwtch.im/cwtch/model"
"cwtch.im/cwtch/peer/peer"
"cwtch.im/cwtch/protocol"
2018-06-23 16:15:36 +00:00
"git.openprivacy.ca/openprivacy/libricochet-go"
2018-10-29 19:19:30 +00:00
"git.openprivacy.ca/openprivacy/libricochet-go/application"
2018-06-23 16:15:36 +00:00
"git.openprivacy.ca/openprivacy/libricochet-go/channels"
"git.openprivacy.ca/openprivacy/libricochet-go/connection"
"git.openprivacy.ca/openprivacy/libricochet-go/identity"
2018-10-05 03:18:34 +00:00
"golang.org/x/crypto/ed25519"
"net"
"testing"
"time"
)
2018-10-05 03:18:34 +00:00
func PeerAuthValid(hostname string, key ed25519.PublicKey) (allowed, known bool) {
return true, true
}
2018-10-05 03:18:34 +00:00
func runtestpeer(t *testing.T, tp *TestPeer, identity identity.Identity) {
ln, _ := net.Listen("tcp", "127.0.0.1:5452")
conn, _ := ln.Accept()
defer conn.Close()
rc, err := goricochet.NegotiateVersionInbound(conn)
if err != nil {
t.Errorf("Negotiate Version Error: %v", err)
}
rc.TraceLog(true)
2018-10-05 03:18:34 +00:00
err = connection.HandleInboundConnection(rc).ProcessAuthAsV3Server(identity, PeerAuthValid)
if err != nil {
t.Errorf("ServerAuth Error: %v", err)
}
2018-03-11 19:05:35 +00:00
tp.RegisterChannelHandler("im.cwtch.peer", func() channels.Handler {
2018-03-14 22:23:35 +00:00
cpc := new(peer.CwtchPeerChannel)
cpc.Handler = tp
return cpc
2018-03-11 19:05:35 +00:00
})
rc.Process(tp)
}
type TestPeer struct {
connection.AutoConnectionHandler
ReceivedIdentityPacket bool
ReceivedGroupInvite bool
2018-03-11 19:05:35 +00:00
}
2018-03-12 18:43:51 +00:00
func (tp *TestPeer) HandleGroupInvite(gci *protocol.GroupChatInvite) {
tp.ReceivedGroupInvite = true
}
func TestPeerPeerConnection(t *testing.T) {
2018-10-05 03:18:34 +00:00
pub, priv, _ := ed25519.GenerateKey(rand.Reader)
identity := identity.InitializeV3("", &priv, &pub)
profile := model.GenerateNewProfile("alice")
2018-10-05 03:18:34 +00:00
hostname := identity.Hostname()
2018-10-29 19:19:30 +00:00
ppc := NewPeerPeerConnection("127.0.0.1:5452|"+hostname, profile, nil, application.ApplicationInstanceFactory{})
2018-10-05 03:18:34 +00:00
2018-03-11 19:05:35 +00:00
tp := new(TestPeer)
tp.Init()
2018-10-05 03:18:34 +00:00
go runtestpeer(t, tp, identity)
state := ppc.GetState()
if state != DISCONNECTED {
t.Errorf("new connections should start in disconnected state")
}
go ppc.Run()
2018-03-11 19:05:35 +00:00
time.Sleep(time.Second * 5)
state = ppc.GetState()
2018-03-11 19:05:35 +00:00
if state != AUTHENTICATED {
t.Errorf("connection state should be authenticated(3), was instead %v", state)
}
2018-05-16 20:20:46 +00:00
_, invite, _ := profile.StartGroup("aaa.onion")
ppc.SendGroupInvite(invite)
time.Sleep(time.Second * 3)
if tp.ReceivedGroupInvite == false {
t.Errorf("should have received an group invite packet")
}
}