kubo/peer/peerstore_test.go

package peer

import (
	"errors"
	"testing"

	ma "github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/jbenet/go-multiaddr"
)

func setupPeer(id string, addr string) (*Peer, error) {
	tcp, err := ma.NewMultiaddr(addr)
	if err != nil {
		return nil, err
	}

	p := &Peer{ID: ID(id)}
	p.AddAddress(tcp)
	return p, nil
}

func TestPeerstore(t *testing.T) {

	ps := NewPeerstore()

	p11, _ := setupPeer("11140beec7b5ea3f0fdbc95d0dd47f3c5bc275da8a31", "/ip4/127.0.0.1/tcp/1234")
	p21, _ := setupPeer("11140beec7b5ea3f0fdbc95d0dd47f3c5bc275da8a32", "/ip4/127.0.0.1/tcp/2345")
	// p31, _ := setupPeer("11140beec7b5ea3f0fdbc95d0dd47f3c5bc275da8a33", "/ip4/127.0.0.1/tcp/3456")
	// p41, _ := setupPeer("11140beec7b5ea3f0fdbc95d0dd47f3c5bc275da8a34", "/ip4/127.0.0.1/tcp/4567")

	err := ps.Put(p11)
	if err != nil {
		t.Error(err)
	}

	p12, err := ps.Get(ID("11140beec7b5ea3f0fdbc95d0dd47f3c5bc275da8a31"))
	if err != nil {
		t.Error(err)
	}

	if p11 != p12 {
		t.Error(errors.New("peers should be the same"))
	}

	err = ps.Put(p21)
	if err != nil {
		t.Error(err)
	}

	p22, err := ps.Get(ID("11140beec7b5ea3f0fdbc95d0dd47f3c5bc275da8a32"))
	if err != nil {
		t.Error(err)
	}

	if p21 != p22 {
		t.Error(errors.New("peers should be the same"))
	}

	_, err = ps.Get(ID("11140beec7b5ea3f0fdbc95d0dd47f3c5bc275da8a33"))
	if err == nil {
		t.Error(errors.New("should've been an error here"))
	}

	err = ps.Delete(ID("11140beec7b5ea3f0fdbc95d0dd47f3c5bc275da8a31"))
	if err != nil {
		t.Error(err)
	}

	_, err = ps.Get(ID("11140beec7b5ea3f0fdbc95d0dd47f3c5bc275da8a31"))
	if err == nil {
		t.Error(errors.New("should've been an error here"))
	}

	p22, err = ps.Get(ID("11140beec7b5ea3f0fdbc95d0dd47f3c5bc275da8a32"))
	if err != nil {
		t.Error(err)
	}

	if p21 != p22 {
		t.Error(errors.New("peers should be the same"))
	}

}