mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
36 lines
1 KiB
Go
36 lines
1 KiB
Go
|
package network
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/docker/docker/api/types"
|
||
|
|
"github.com/docker/docker/client"
|
||
|
|
"github.com/gotestyourself/gotestyourself/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func createNetwork(ctx context.Context, client client.APIClient, name string, ops ...func(*types.NetworkCreate)) (string, error) {
|
||
|
|
config := types.NetworkCreate{}
|
||
|
|
|
||
|
|
for _, op := range ops {
|
||
|
|
op(&config)
|
||
|
|
}
|
||
|
|
|
||
|
|
n, err := client.NetworkCreate(ctx, name, config)
|
||
|
|
return n.ID, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Create creates a network with the specified options
|
||
|
|
func Create(ctx context.Context, client client.APIClient, name string, ops ...func(*types.NetworkCreate)) (string, error) {
|
||
|
|
return createNetwork(ctx, client, name, ops...)
|
||
|
|
}
|
||
|
|
|
||
|
|
// CreateNoError creates a network with the specified options and verifies there were no errors
|
||
|
|
func CreateNoError(t *testing.T, ctx context.Context, client client.APIClient, name string, ops ...func(*types.NetworkCreate)) string { // nolint: golint
|
||
|
|
t.Helper()
|
||
|
|
|
||
|
|
name, err := createNetwork(ctx, client, name, ops...)
|
||
|
|
assert.NilError(t, err)
|
||
|
|
return name
|
||
|
|
}
|