package main import ( "bytes" "fmt" "strings" "testing" "github.com/ipfs-cluster/ipfs-cluster/api" peer "github.com/libp2p/go-libp2p-core/peer" ) func verifyOutput(t *testing.T, outStr string, trueStr string) { outLines := strings.Split(outStr, "\n") trueLines := strings.Split(trueStr, "\n") if len(outLines) != len(trueLines) { fmt.Printf("expected:\n-%s-\n\n\nactual:\n-%s-", trueStr, outStr) t.Fatal("Number of output lines does not match expectation") } for i := range outLines { if outLines[i] != trueLines[i] { t.Errorf("Difference in sorted outputs (%d): %s vs %s", i, outLines[i], trueLines[i]) } } } var simpleIpfs = `digraph cluster { /* The nodes of the connectivity graph */ /* The cluster-service peers */ subgraph { rank="min" C0 [label=<
Qm*eqhEhD > group="QmUBuxVHoNNjfmNpTad36UeaFQv3gXAtCv9r6KhmeqhEhD" color="orange" style="filled" colorscheme="x11" fontcolor="black" fontname="Arial" shape="box3d" peripheries="2" ] C1 [label=<
Qm*cgHDQJ > group="QmV35LjbEGPfN7KfMAJp43VV2enwXqqQf5esx4vUcgHDQJ" color="darkorange3" style="filled" colorscheme="x11" fontcolor="black" fontname="Arial" shape="box3d" ] C2 [label=<
Qm*6MQmJu > group="QmZ2ckU7G35MYyJgMTwMUnicsGqSy3YUxGBX7qny6MQmJu" color="darkorange3" style="filled" colorscheme="x11" fontcolor="black" fontname="Arial" shape="box3d" ] } /* The ipfs peers */ subgraph { rank="max" I0 [label=< IPFS
Qm*N5LSsq > group="QmPFKAGZbUjdzt8BBx8VTWCe9UeUQVcoqHFehSPzN5LSsq" color="turquoise3" style="filled" colorscheme="x11" fontcolor="black" fontname="Arial" shape="cylinder" ] I1 [label=< IPFS
Qm*R3DZDV > group="QmXbiVZd93SLiu9TAm21F2y9JwGiFLydbEVkPBaMR3DZDV" color="turquoise3" style="filled" colorscheme="x11" fontcolor="black" fontname="Arial" shape="cylinder" ] I2 [label=< IPFS
Qm*wbBsuL > group="QmbU7273zH6jxwDe2nqRmEm2rp5PpqP2xeQr2xCmwbBsuL" color="turquoise3" style="filled" colorscheme="x11" fontcolor="black" fontname="Arial" shape="cylinder" ] } /* Edges representing active connections in the cluster */ /* The connections among cluster-service peers */ C0 -> C1 C0 -> C2 C1 -> C0 C1 -> C2 C2 -> C0 C2 -> C1 /* The connections between cluster peers and their ipfs daemons */ C0 -> I1 C1 -> I0 C2 -> I2 /* The swarm peer connections among ipfs daemons in the cluster */ I0 -> I1 I0 -> I2 I1 -> I0 I1 -> I2 I2 -> I1 I2 -> I0 }` var ( pid1, _ = peer.Decode("QmUBuxVHoNNjfmNpTad36UeaFQv3gXAtCv9r6KhmeqhEhD") pid2, _ = peer.Decode("QmV35LjbEGPfN7KfMAJp43VV2enwXqqQf5esx4vUcgHDQJ") pid3, _ = peer.Decode("QmZ2ckU7G35MYyJgMTwMUnicsGqSy3YUxGBX7qny6MQmJu") pid4, _ = peer.Decode("QmXbiVZd93SLiu9TAm21F2y9JwGiFLydbEVkPBaMR3DZDV") pid5, _ = peer.Decode("QmPFKAGZbUjdzt8BBx8VTWCe9UeUQVcoqHFehSPzN5LSsq") pid6, _ = peer.Decode("QmbU7273zH6jxwDe2nqRmEm2rp5PpqP2xeQr2xCmwbBsuL") pid7, _ = peer.Decode("QmQsdAdCHs4PRLi5tcoLfasYppryqQENxgAy4b2aS8xccb") pid8, _ = peer.Decode("QmVV2enwXqqQf5esx4v36UeaFQvFehSPzNfi8aaaaaanM8") pid9, _ = peer.Decode("QmfCHNQ2vbUmAuJZhE2hEpgiJq4sL1XScWEKnUrVtWZdeD") ) func TestSimpleIpfsGraphs(t *testing.T) { cg := api.ConnectGraph{ ClusterID: pid1, ClusterLinks: map[string][]peer.ID{ peer.Encode(pid1): { pid2, pid3, }, peer.Encode(pid2): { pid1, pid3, }, peer.Encode(pid3): { pid1, pid2, }, }, IPFSLinks: map[string][]peer.ID{ peer.Encode(pid4): { pid5, pid6, }, peer.Encode(pid5): { pid4, pid6, }, peer.Encode(pid6): { pid4, pid5, }, }, ClustertoIPFS: map[string]peer.ID{ peer.Encode(pid1): pid4, peer.Encode(pid2): pid5, peer.Encode(pid3): pid6, }, } buf := new(bytes.Buffer) err := makeDot(cg, buf, false) if err != nil { t.Fatal(err) } verifyOutput(t, buf.String(), simpleIpfs) } var allIpfs = `digraph cluster { /* The nodes of the connectivity graph */ /* The cluster-service peers */ subgraph { rank="min" C0 [label=<
Qm*eqhEhD > group="QmUBuxVHoNNjfmNpTad36UeaFQv3gXAtCv9r6KhmeqhEhD" color="orange" style="filled" colorscheme="x11" fontcolor="black" fontname="Arial" shape="box3d" peripheries="2" ] C1 [label=<
Qm*cgHDQJ > group="QmV35LjbEGPfN7KfMAJp43VV2enwXqqQf5esx4vUcgHDQJ" color="darkorange3" style="filled" colorscheme="x11" fontcolor="black" fontname="Arial" shape="box3d" ] C2 [label=<
Qm*6MQmJu > group="QmZ2ckU7G35MYyJgMTwMUnicsGqSy3YUxGBX7qny6MQmJu" color="darkorange3" style="filled" colorscheme="x11" fontcolor="black" fontname="Arial" shape="box3d" ] } /* The ipfs peers */ subgraph { rank="max" I0 [label=< IPFS
Qm*N5LSsq > group="QmPFKAGZbUjdzt8BBx8VTWCe9UeUQVcoqHFehSPzN5LSsq" color="turquoise3" style="filled" colorscheme="x11" fontcolor="black" fontname="Arial" shape="cylinder" ] I1 [label=< IPFS
Qm*S8xccb > group="QmQsdAdCHs4PRLi5tcoLfasYppryqQENxgAy4b2aS8xccb" color="turquoise3" style="filled" colorscheme="x11" fontcolor="black" fontname="Arial" shape="cylinder" ] I2 [label=< IPFS
Qm*aaanM8 > group="QmVV2enwXqqQf5esx4v36UeaFQvFehSPzNfi8aaaaaanM8" color="turquoise3" style="filled" colorscheme="x11" fontcolor="black" fontname="Arial" shape="cylinder" ] I3 [label=< IPFS
Qm*R3DZDV > group="QmXbiVZd93SLiu9TAm21F2y9JwGiFLydbEVkPBaMR3DZDV" color="turquoise3" style="filled" colorscheme="x11" fontcolor="black" fontname="Arial" shape="cylinder" ] I4 [label=< IPFS
Qm*wbBsuL > group="QmbU7273zH6jxwDe2nqRmEm2rp5PpqP2xeQr2xCmwbBsuL" color="turquoise3" style="filled" colorscheme="x11" fontcolor="black" fontname="Arial" shape="cylinder" ] I5 [label=< IPFS
Qm*tWZdeD > group="QmfCHNQ2vbUmAuJZhE2hEpgiJq4sL1XScWEKnUrVtWZdeD" color="turquoise3" style="filled" colorscheme="x11" fontcolor="black" fontname="Arial" shape="cylinder" ] } /* Edges representing active connections in the cluster */ /* The connections among cluster-service peers */ C0 -> C1 C0 -> C2 C1 -> C0 C1 -> C2 C2 -> C0 C2 -> C1 /* The connections between cluster peers and their ipfs daemons */ C0 -> I3 C1 -> I0 C2 -> I4 /* The swarm peer connections among ipfs daemons in the cluster */ I0 -> I3 I0 -> I4 I0 -> I1 I0 -> I2 I0 -> I5 I3 -> I0 I3 -> I4 I3 -> I1 I3 -> I2 I3 -> I5 I4 -> I3 I4 -> I0 I4 -> I1 I4 -> I2 I4 -> I5 }` func TestIpfsAllGraphs(t *testing.T) { cg := api.ConnectGraph{ ClusterID: pid1, ClusterLinks: map[string][]peer.ID{ peer.Encode(pid1): { pid2, pid3, }, peer.Encode(pid2): { pid1, pid3, }, peer.Encode(pid3): { pid1, pid2, }, }, IPFSLinks: map[string][]peer.ID{ peer.Encode(pid4): { pid5, pid6, pid7, pid8, pid9, }, peer.Encode(pid5): { pid4, pid6, pid7, pid8, pid9, }, peer.Encode(pid6): { pid4, pid5, pid7, pid8, pid9, }, }, ClustertoIPFS: map[string]peer.ID{ peer.Encode(pid1): pid4, peer.Encode(pid2): pid5, peer.Encode(pid3): pid6, }, } buf := new(bytes.Buffer) err := makeDot(cg, buf, true) if err != nil { t.Fatal(err) } verifyOutput(t, buf.String(), allIpfs) }