From d96fb27ab8adef666c5b0a1d2ce94e5f9bc58922 Mon Sep 17 00:00:00 2001
From: Aleksander Mistewicz <a.mistewicz@gmail.com>
Date: Wed, 25 Sep 2019 15:58:19 +0200
Subject: [PATCH] Add simple connection test

---
 src/yggdrasil/core_test.go | 47 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 src/yggdrasil/core_test.go

diff --git a/src/yggdrasil/core_test.go b/src/yggdrasil/core_test.go
new file mode 100644
index 00000000..4b283d4d
--- /dev/null
+++ b/src/yggdrasil/core_test.go
@@ -0,0 +1,47 @@
+package yggdrasil
+
+import (
+	"os"
+	"testing"
+
+	"github.com/gologme/log"
+
+	"github.com/yggdrasil-network/yggdrasil-go/src/config"
+)
+
+// GenerateConfig is modification
+func GenerateConfig() *config.NodeConfig {
+	cfg := config.GenerateConfig()
+	cfg.AdminListen = "none"
+	cfg.Listen = []string{"tcp://127.0.0.1:0"}
+	cfg.IfName = "none"
+
+	return cfg
+}
+
+func GetLoggerWithPrefix(prefix string) *log.Logger {
+	l := log.New(os.Stderr, prefix, log.Flags())
+	l.EnableLevel("info")
+	l.EnableLevel("warn")
+	l.EnableLevel("error")
+	return l
+}
+
+func TestCore_Start(t *testing.T) {
+	nodeA := Core{}
+	_, err := nodeA.Start(GenerateConfig(), GetLoggerWithPrefix("A: "))
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	nodeB := Core{}
+	_, err = nodeB.Start(GenerateConfig(), GetLoggerWithPrefix("B: "))
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = nodeB.AddPeer("tcp://"+nodeA.link.tcp.getAddr().String(), "")
+	if err != nil {
+		t.Fatal(err)
+	}
+}