Daemon and DHT Tests

core/core.go

@@ -56,6 +56,7 @@ type IpfsNode struct {
 
 	// the name system, resolves paths to hashes
 	// Namesys *namesys.Namesys
+
 }
 
 // NewIpfsNode constructs a new IpfsNode based on the given config.
@@ -77,8 +78,8 @@ func NewIpfsNode(cfg *config.Config, online bool) (*IpfsNode, error) {
 	var (
 		net *swarm.Swarm
 		// TODO: refactor so we can use IpfsRouting interface instead of being DHT-specific
-		route* dht.IpfsDHT
-		swap *bitswap.BitSwap
+		route *dht.IpfsDHT
+		swap  *bitswap.BitSwap
 	)
 
 	if online {
@@ -134,7 +135,7 @@ func initIdentity(cfg *config.Config) (*peer.Peer, error) {
 			return nil, err
 		}
 
-		addresses = []*ma.Multiaddr{ maddr }
+		addresses = []*ma.Multiaddr{maddr}
 	}
 
 	skb, err := base64.StdEncoding.DecodeString(cfg.Identity.PrivKey)
@@ -174,3 +175,20 @@ func (n *IpfsNode) PinDagNode(nd *merkledag.Node) error {
 	u.POut("Pinning node. Currently No-Op\n")
 	return nil
 }
+
+func (n *IpfsNode) Close() (*IpfsNode, error) {
+	//n.Swarm.Close()
+
+	return &IpfsNode{
+		Config:    nil,
+		PeerMap:   nil,
+		Datastore: nil,
+		Blocks:    nil,
+		DAG:       nil,
+		Resolver:  nil,
+		BitSwap:   nil,
+		Identity:  nil,
+		Routing:   nil,
+	}, nil
+
+}

core/core_test.go

@@ -43,12 +43,14 @@ func TestInitialization(t *testing.T) {
 		if n == nil || err != nil {
 			t.Error("Should have constructed.", i, err)
 		}
+		n.Close()
 	}
 
 	for i, c := range bad {
 		n, err := NewIpfsNode(c, false)
 		if n != nil || err == nil {
 			t.Error("Should have failed to construct.", i)
 		}
+		n.Close()
 	}
 }

daemon/daemon_test.go

@@ -0,0 +1,63 @@
+package daemon
+
+import (
+	"encoding/base64"
+	"testing"
+
+	config "github.com/jbenet/go-ipfs/config"
+	core "github.com/jbenet/go-ipfs/core"
+	ci "github.com/jbenet/go-ipfs/crypto"
+)
+
+func TestDaemonListener(t *testing.T) {
+
+	priv, _, err := ci.GenerateKeyPair(ci.RSA, 512)
+	if err != nil {
+		t.Fatal(err)
+	}
+	prbytes, err := priv.Bytes()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	privKey := base64.StdEncoding.EncodeToString(prbytes)
+
+	id := &config.Identity{
+		PeerID:  "QmNgdzLieYi8tgfo2WfTUzNVH5hQK9oAYGVf6dxN12NrHt",
+		Address: "/ip4/127.0.0.1/tcp/8000",
+		PrivKey: privKey,
+	}
+
+	nodeConfigs := []*config.Config{
+		&config.Config{
+			Identity: id,
+			Datastore: config.Datastore{
+				Type: "memory",
+			},
+		},
+
+		&config.Config{
+			Identity: id,
+			Datastore: config.Datastore{
+				Type: "leveldb",
+				Path: ".testdb",
+			},
+		},
+	}
+
+	for _, c := range nodeConfigs {
+
+		node, _ := core.NewIpfsNode(c, false)
+		dl, initErr := NewDaemonListener(node, "localhost:1327")
+		if initErr != nil {
+			t.Fatal(initErr)
+		}
+		closeErr := dl.Close()
+		if closeErr != nil {
+			t.Fatal(closeErr)
+		}
+		node.Close()
+
+	}
+
+}

routing/dht/providers_test.go

@@ -12,9 +12,14 @@ func TestProviderManager(t *testing.T) {
 	p := NewProviderManager(mid)
 	a := u.Key("test")
 	p.AddProvider(a, &peer.Peer{})
-	resp := p.GetProviders(a)
-	if len(resp) != 1 {
-		t.Fatal("Could not retrieve provider.")
+	remotePeers := p.GetProviders(a)
+	localPeers := p.GetLocal()
+	if len(remotePeers) != 1 {
+		t.Fatal("Could not retrieve remote provider.")
 	}
+	if len(localPeers) != 1 {
+		t.Fatal("Could not retrieve local provider.")
+	}
+
 	p.Halt()
 }