resolve_test.go 1.16 KB
Newer Older
Jeromy's avatar
Jeromy committed
1 2 3 4 5 6 7 8 9 10
package namesys

import (
	"testing"

	ds "github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/jbenet/datastore.go"
	bs "github.com/jbenet/go-ipfs/blockservice"
	ci "github.com/jbenet/go-ipfs/crypto"
	mdag "github.com/jbenet/go-ipfs/merkledag"
	"github.com/jbenet/go-ipfs/peer"
11
	mock "github.com/jbenet/go-ipfs/routing/mock"
Jeromy's avatar
Jeromy committed
12 13 14 15 16 17 18 19
	u "github.com/jbenet/go-ipfs/util"
)

func TestRoutingResolve(t *testing.T) {
	local := &peer.Peer{
		ID: []byte("testID"),
	}
	lds := ds.NewMapDatastore()
20
	d := mock.NewMockRouter(local, lds)
Jeromy's avatar
Jeromy committed
21 22 23 24 25 26 27 28 29 30

	bserv, err := bs.NewBlockService(lds, nil)
	if err != nil {
		t.Fatal(err)
	}

	dag := &mdag.DAGService{Blocks: bserv}

	resolve := NewMasterResolver(d, dag)

31
	pub := ipnsPublisher{
Jeromy's avatar
Jeromy committed
32 33 34 35 36 37 38 39 40
		dag:     dag,
		routing: d,
	}

	privk, pubk, err := ci.GenerateKeyPair(ci.RSA, 512)
	if err != nil {
		t.Fatal(err)
	}

41
	err = pub.Publish(privk, "Hello")
Jeromy's avatar
Jeromy committed
42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64
	if err != nil {
		t.Fatal(err)
	}

	pubkb, err := pubk.Bytes()
	if err != nil {
		t.Fatal(err)
	}

	pkhash, err := u.Hash(pubkb)
	if err != nil {
		t.Fatal(err)
	}

	res, err := resolve.Resolve(u.Key(pkhash).Pretty())
	if err != nil {
		t.Fatal(err)
	}

	if res != "Hello" {
		t.Fatal("Got back incorrect value.")
	}
}