added server tests

Author: radutopala

server.go

@@ -38,7 +38,7 @@ func (s *server) Key(file string) *server {
 }
 
 func (s *server) GetUser() string {
-	return s.Dsn[:strings.Index(s.Dsn, "@")-1]
+	return s.Dsn[:strings.Index(s.Dsn, "@")]
 }
 
 func (s *server) GetHost() string {

server_test.go

@@ -1,8 +1,8 @@
 package exec
 
 import (
-    "github.com/stretchr/testify/require"
-    "testing"
+	"github.com/stretchr/testify/require"
+	"testing"
 )
 
 func TestServer_AddRole(t *testing.T) {
@@ -16,12 +16,54 @@ func TestServer_AddRole(t *testing.T) {
 }
 
 func TestServer_HasRole(t *testing.T) {
-    s := &server{
-        Name: "qa",
-        Dsn:  "root@domain.com",
-    }
-    s.AddRole("qa")
-
-    require.True(t, s.HasRole("qa"))
-    require.False(t, s.HasRole("test"))
-}
+	s := &server{
+		Name: "qa",
+		Dsn:  "root@domain.com",
+	}
+	s.AddRole("qa")
+
+	require.True(t, s.HasRole("qa"))
+	require.False(t, s.HasRole("test"))
+}
+
+func TestServer_Set(t *testing.T) {
+	s := &server{
+		Name:    "qa",
+		Dsn:     "root@domain.com",
+		Configs: make(map[string]*config),
+	}
+	s.Set("config", "value")
+
+	require.Contains(t, s.Configs, "config")
+	require.Equal(t, s.Configs["config"].Value(), "value")
+}
+
+func TestServer_Key(t *testing.T) {
+	s := &server{
+		Name:      "qa",
+		Dsn:       "root@domain.com",
+		Configs:   make(map[string]*config),
+		sshClient: &sshClient{},
+	}
+	s.Key("key")
+
+	require.Contains(t, s.sshClient.keys, "key")
+}
+
+func TestServer_GetUser(t *testing.T) {
+	s := &server{
+		Name: "qa",
+		Dsn:  "root@domain.com",
+	}
+
+	require.Equal(t, s.GetUser(), "root")
+}
+
+func TestServer_GetHost(t *testing.T) {
+	s := &server{
+		Name: "qa",
+		Dsn:  "root@domain.com",
+	}
+
+	require.Equal(t, s.GetHost(), "domain.com")
+}