depot/packages/networking/ipfs-cluster/pnet_test.go

90 lines
2.5 KiB
Go

package ipfscluster
import (
"context"
"testing"
)
func TestClusterSecretFormat(t *testing.T) {
goodSecret := "0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef"
emptySecret := ""
tooShort := "0123456789abcdef"
tooLong := "0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0"
unsupportedChars := "0123456789abcdef0123456789!!!!!!0123456789abcdef0123456789abcdef"
_, err := DecodeClusterSecret(goodSecret)
if err != nil {
t.Fatal("Failed to decode well-formatted secret.")
}
decodedEmptySecret, err := DecodeClusterSecret(emptySecret)
if decodedEmptySecret != nil || err != nil {
t.Fatal("Unsuspected output of decoding empty secret.")
}
_, err = DecodeClusterSecret(tooShort)
if err == nil {
t.Fatal("Successfully decoded secret that should haved failed (too short).")
}
_, err = DecodeClusterSecret(tooLong)
if err == nil {
t.Fatal("Successfully decoded secret that should haved failed (too long).")
}
_, err = DecodeClusterSecret(unsupportedChars)
if err == nil {
t.Fatal("Successfully decoded secret that should haved failed (unsupported chars).")
}
}
func TestSimplePNet(t *testing.T) {
ctx := context.Background()
clusters, mocks, boot := peerManagerClusters(t)
defer cleanState()
defer shutdownClusters(t, clusters, mocks)
defer boot.Close()
if len(clusters) < 2 {
t.Skip("need at least 2 nodes for this test")
}
_, err := clusters[0].PeerAdd(ctx, clusters[1].id)
if err != nil {
t.Fatal(err)
}
ttlDelay()
if len(peers(ctx, t, clusters[0])) != len(peers(ctx, t, clusters[1])) {
t.Fatal("Expected same number of peers")
}
if len(peers(ctx, t, clusters[0])) < 2 {
// crdt mode has auto discovered all peers at this point.
// Raft mode has 2 peers only.
t.Fatal("Expected at least 2 peers")
}
}
// // Adds one minute to tests. Disabled for the moment.
// func TestClusterSecretRequired(t *testing.T) {
// cl1Secret, err := pnet.GenerateV1Bytes()
// if err != nil {
// t.Fatal("Unable to generate cluster secret.")
// }
// cl1, _ := createOnePeerCluster(t, 1, (*cl1Secret)[:])
// cl2, _ := createOnePeerCluster(t, 2, testingClusterSecret)
// defer cleanState()
// defer cl1.Shutdown()
// defer cl2.Shutdown()
// peers1 := cl1.Peers()
// peers2 := cl2.Peers()
//
// _, err = cl1.PeerAdd(clusterAddr(cl2))
// if err == nil {
// t.Fatal("Peer entered private cluster without key.")
// }
// if len(peers1) != len(peers2) {
// t.Fatal("Expected same number of peers")
// }
// if len(peers1) != 1 {
// t.Fatal("Expected no peers other than self")
// }
// }