gitlab.com/postgres-ai/database-lab/v3@v3.0.3/pkg/util/networks/networks_test.go (about)

     1  /*
     2  2021 © Postgres.ai
     3  */
     4  
     5  package networks
     6  
     7  import (
     8  	"testing"
     9  
    10  	"github.com/docker/docker/api/types"
    11  	"github.com/stretchr/testify/assert"
    12  )
    13  
    14  func TestInternalNetworks(t *testing.T) {
    15  	t.Run("test internal network naming", func(t *testing.T) {
    16  		instanceID := "testInstanceID"
    17  
    18  		assert.Equal(t, "dle_network_testInstanceID", getNetworkName(instanceID))
    19  	})
    20  }
    21  
    22  func TestIfContainerConnected(t *testing.T) {
    23  	t.Run("test if container connected", func(t *testing.T) {
    24  		resource := types.NetworkResource{
    25  			Containers: map[string]types.EndpointResource{
    26  				"testID": {Name: "test_server"},
    27  			},
    28  		}
    29  		testCases := []struct {
    30  			containerName string
    31  			result        bool
    32  		}{
    33  			{
    34  				containerName: "test_server",
    35  				result:        true,
    36  			},
    37  			{
    38  				containerName: "not_connected_server",
    39  				result:        false,
    40  			},
    41  		}
    42  
    43  		for _, tc := range testCases {
    44  			assert.Equal(t, tc.result, hasContainerConnected(resource, tc.containerName))
    45  		}
    46  	})
    47  }