more tests

Author: guumaster

pkg/docker/docker_test.go

@@ -3,15 +3,46 @@ package docker
 import (
 	"bytes"
 	"context"
+	"errors"
 	"io/ioutil"
 	"net/http"
 	"testing"
 
-	"github.com/docker/docker/api/types"
+	dtypes "github.com/docker/docker/api/types"
 	"github.com/docker/docker/client"
 	"github.com/stretchr/testify/assert"
+
+	"github.com/guumaster/hostctl/pkg/types"
 )
 
+func TestGetNetworkID(t *testing.T) {
+	cli := newClientWithResponse(t, map[string]string{
+		"/v1.22/networks": `[
+{"Id": "networkId1", "Name": "networkName1" },
+{"Id": "networkId2", "Name": "networkName2" }
+]`,
+	})
+
+	t.Run("By Name", func(t *testing.T) {
+		net, err := GetNetworkID(context.Background(), cli, "networkName2")
+		assert.NoError(t, err)
+
+		assert.Equal(t, "networkId2", net)
+	})
+
+	t.Run("By Name error", func(t *testing.T) {
+		_, err := GetNetworkID(context.Background(), cli, "invent")
+		assert.True(t, errors.Is(err, types.ErrUnknownNetworkID))
+	})
+
+	t.Run("By Name empty", func(t *testing.T) {
+		list, err := GetNetworkID(context.Background(), cli, "")
+		assert.NoError(t, err)
+		assert.Empty(t, list)
+	})
+
+}
+
 func TestGetContainerList(t *testing.T) {
 	cli := newClientWithResponse(t, map[string]string{
 		"/v1.22/networks": `[
@@ -32,17 +63,17 @@ func TestGetContainerList(t *testing.T) {
 
 	assert.Len(t, list, 2)
 
-	assert.IsType(t, types.Container{}, list[0])
-	assert.IsType(t, types.Container{}, list[1])
+	assert.IsType(t, dtypes.Container{}, list[0])
+	assert.IsType(t, dtypes.Container{}, list[1])
 	assert.Equal(t, "networkID1", list[1].NetworkSettings.Networks["networkName1"].NetworkID)
 
-	assert.Equal(t, types.Container{
+	assert.Equal(t, dtypes.Container{
 		ID:    "container_id1",
 		Names: []string{"container1"},
 
 		NetworkSettings: list[0].NetworkSettings, // simplify the comparison
 	}, list[0])
-	assert.Equal(t, types.Container{
+	assert.Equal(t, dtypes.Container{
 		ID:    "container_id2",
 		Names: []string{"container2"},