1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129 |
- package libnetwork_test
- import (
- "bytes"
- "flag"
- "fmt"
- "io/ioutil"
- "net"
- "os"
- "runtime"
- "strconv"
- "sync"
- "testing"
- log "github.com/Sirupsen/logrus"
- "github.com/docker/docker/pkg/reexec"
- "github.com/docker/libnetwork"
- "github.com/docker/libnetwork/netutils"
- "github.com/docker/libnetwork/pkg/netlabel"
- "github.com/docker/libnetwork/pkg/options"
- "github.com/vishvananda/netns"
- )
- const (
- bridgeNetType = "bridge"
- bridgeName = "docker0"
- )
- func TestMain(m *testing.M) {
- if reexec.Init() {
- return
- }
- os.Exit(m.Run())
- }
- func createTestNetwork(networkType, networkName string, option options.Generic, netOption options.Generic) (libnetwork.Network, error) {
- controller := libnetwork.New()
- genericOption := make(map[string]interface{})
- genericOption[netlabel.GenericData] = option
- err := controller.ConfigureNetworkDriver(networkType, genericOption)
- if err != nil {
- return nil, err
- }
- network, err := controller.NewNetwork(networkType, networkName,
- libnetwork.NetworkOptionGeneric(netOption))
- if err != nil {
- return nil, err
- }
- return network, nil
- }
- func getEmptyGenericOption() map[string]interface{} {
- genericOption := make(map[string]interface{})
- genericOption[netlabel.GenericData] = options.Generic{}
- return genericOption
- }
- func getPortMapping() []netutils.PortBinding {
- return []netutils.PortBinding{
- netutils.PortBinding{Proto: netutils.TCP, Port: uint16(230), HostPort: uint16(23000)},
- netutils.PortBinding{Proto: netutils.UDP, Port: uint16(200), HostPort: uint16(22000)},
- netutils.PortBinding{Proto: netutils.TCP, Port: uint16(120), HostPort: uint16(12000)},
- }
- }
- func TestNull(t *testing.T) {
- network, err := createTestNetwork("null", "testnetwork", options.Generic{},
- options.Generic{})
- if err != nil {
- t.Fatal(err)
- }
- ep, err := network.CreateEndpoint("testep")
- if err != nil {
- t.Fatal(err)
- }
- _, err = ep.Join("null_container",
- libnetwork.JoinOptionHostname("test"),
- libnetwork.JoinOptionDomainname("docker.io"),
- libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
- if err != nil {
- t.Fatal(err)
- }
- err = ep.Leave("null_container")
- if err != nil {
- t.Fatal(err)
- }
- if err := ep.Delete(); err != nil {
- t.Fatal(err)
- }
- if err := network.Delete(); err != nil {
- t.Fatal(err)
- }
- }
- func TestHost(t *testing.T) {
- network, err := createTestNetwork("host", "testnetwork", options.Generic{}, options.Generic{})
- if err != nil {
- t.Fatal(err)
- }
- ep, err := network.CreateEndpoint("testep")
- if err != nil {
- t.Fatal(err)
- }
- _, err = ep.Join("host_container",
- libnetwork.JoinOptionHostname("test"),
- libnetwork.JoinOptionDomainname("docker.io"),
- libnetwork.JoinOptionExtraHost("web", "192.168.0.1"),
- libnetwork.JoinOptionUseDefaultSandbox())
- if err != nil {
- t.Fatal(err)
- }
- err = ep.Leave("host_container")
- if err != nil {
- t.Fatal(err)
- }
- if err := ep.Delete(); err != nil {
- t.Fatal(err)
- }
- if err := network.Delete(); err != nil {
- t.Fatal(err)
- }
- }
- func TestBridge(t *testing.T) {
- defer netutils.SetupTestNetNS(t)()
- ip, subnet, err := net.ParseCIDR("192.168.100.1/24")
- if err != nil {
- t.Fatal(err)
- }
- subnet.IP = ip
- ip, cidr, err := net.ParseCIDR("192.168.100.2/28")
- if err != nil {
- t.Fatal(err)
- }
- cidr.IP = ip
- ip, cidrv6, err := net.ParseCIDR("fe90::1/96")
- if err != nil {
- t.Fatal(err)
- }
- cidrv6.IP = ip
- log.Debug("Adding a bridge")
- option := options.Generic{
- "EnableIPForwarding": true,
- }
- netOption := options.Generic{
- "BridgeName": bridgeName,
- "AddressIPv4": subnet,
- "FixedCIDR": cidr,
- "FixedCIDRv6": cidrv6,
- "EnableIPv6": true,
- "EnableIPTables": true,
- "EnableIPMasquerade": true,
- "EnableICC": true,
- "AllowNonDefaultBridge": true}
- network, err := createTestNetwork(bridgeNetType, "testnetwork", option, netOption)
- if err != nil {
- t.Fatal(err)
- }
- ep, err := network.CreateEndpoint("testep", libnetwork.CreateOptionPortMapping(getPortMapping()))
- if err != nil {
- t.Fatal(err)
- }
- epInfo, err := ep.Info()
- if err != nil {
- t.Fatal(err)
- }
- pmd, ok := epInfo[netlabel.PortMap]
- if !ok {
- t.Fatalf("Could not find expected info in endpoint data")
- }
- pm, ok := pmd.([]netutils.PortBinding)
- if !ok {
- t.Fatalf("Unexpected format for port mapping in endpoint operational data")
- }
- if len(pm) != 3 {
- t.Fatalf("Incomplete data for port mapping in endpoint operational data: %d", len(pm))
- }
- if err := ep.Delete(); err != nil {
- t.Fatal(err)
- }
- if err := network.Delete(); err != nil {
- t.Fatal(err)
- }
- }
- func TestUnknownDriver(t *testing.T) {
- defer netutils.SetupTestNetNS(t)()
- _, err := createTestNetwork("unknowndriver", "testnetwork", options.Generic{}, options.Generic{})
- if err == nil {
- t.Fatal("Expected to fail. But instead succeeded")
- }
- if _, ok := err.(libnetwork.NetworkTypeError); !ok {
- t.Fatalf("Did not fail with expected error. Actual error: %v", err)
- }
- }
- func TestNilDriver(t *testing.T) {
- controller := libnetwork.New()
- _, err := controller.NewNetwork("framerelay", "dummy",
- libnetwork.NetworkOptionGeneric(getEmptyGenericOption()))
- if err == nil {
- t.Fatal("Expected to fail. But instead succeeded")
- }
- if err != libnetwork.ErrInvalidNetworkDriver {
- t.Fatalf("Did not fail with expected error. Actual error: %v", err)
- }
- }
- func TestNoInitDriver(t *testing.T) {
- controller := libnetwork.New()
- _, err := controller.NewNetwork("ppp", "dummy",
- libnetwork.NetworkOptionGeneric(getEmptyGenericOption()))
- if err == nil {
- t.Fatal("Expected to fail. But instead succeeded")
- }
- if err != libnetwork.ErrInvalidNetworkDriver {
- t.Fatalf("Did not fail with expected error. Actual error: %v", err)
- }
- }
- func TestDuplicateNetwork(t *testing.T) {
- defer netutils.SetupTestNetNS(t)()
- controller := libnetwork.New()
- genericOption := make(map[string]interface{})
- genericOption[netlabel.GenericData] = options.Generic{}
- err := controller.ConfigureNetworkDriver(bridgeNetType, genericOption)
- if err != nil {
- t.Fatal(err)
- }
- _, err = controller.NewNetwork(bridgeNetType, "testnetwork",
- libnetwork.NetworkOptionGeneric(genericOption))
- if err != nil {
- t.Fatal(err)
- }
- _, err = controller.NewNetwork(bridgeNetType, "testnetwork")
- if err == nil {
- t.Fatal("Expected to fail. But instead succeeded")
- }
- if _, ok := err.(libnetwork.NetworkNameError); !ok {
- t.Fatalf("Did not fail with expected error. Actual error: %v", err)
- }
- }
- func TestNetworkName(t *testing.T) {
- defer netutils.SetupTestNetNS(t)()
- networkName := "testnetwork"
- n, err := createTestNetwork(bridgeNetType, networkName, options.Generic{}, options.Generic{})
- if err != nil {
- t.Fatal(err)
- }
- if n.Name() != networkName {
- t.Fatalf("Expected network name %s, got %s", networkName, n.Name())
- }
- }
- func TestNetworkType(t *testing.T) {
- defer netutils.SetupTestNetNS(t)()
- n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
- if err != nil {
- t.Fatal(err)
- }
- if n.Type() != bridgeNetType {
- t.Fatalf("Expected network type %s, got %s", bridgeNetType, n.Type())
- }
- }
- func TestNetworkID(t *testing.T) {
- defer netutils.SetupTestNetNS(t)()
- n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
- if err != nil {
- t.Fatal(err)
- }
- if n.ID() == "" {
- t.Fatal("Expected non-empty network id")
- }
- }
- func TestDeleteNetworkWithActiveEndpoints(t *testing.T) {
- defer netutils.SetupTestNetNS(t)()
- option := options.Generic{
- "BridgeName": bridgeName,
- "AllowNonDefaultBridge": true}
- network, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, option)
- if err != nil {
- t.Fatal(err)
- }
- ep, err := network.CreateEndpoint("testep")
- if err != nil {
- t.Fatal(err)
- }
- err = network.Delete()
- if err == nil {
- t.Fatal("Expected to fail. But instead succeeded")
- }
- if _, ok := err.(*libnetwork.ActiveEndpointsError); !ok {
- t.Fatalf("Did not fail with expected error. Actual error: %v", err)
- }
- // Done testing. Now cleanup.
- if err := ep.Delete(); err != nil {
- t.Fatal(err)
- }
- if err := network.Delete(); err != nil {
- t.Fatal(err)
- }
- }
- func TestUnknownNetwork(t *testing.T) {
- defer netutils.SetupTestNetNS(t)()
- option := options.Generic{
- "BridgeName": bridgeName,
- "AllowNonDefaultBridge": true}
- network, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, option)
- if err != nil {
- t.Fatal(err)
- }
- err = network.Delete()
- if err != nil {
- t.Fatal(err)
- }
- err = network.Delete()
- if err == nil {
- t.Fatal("Expected to fail. But instead succeeded")
- }
- if _, ok := err.(*libnetwork.UnknownNetworkError); !ok {
- t.Fatalf("Did not fail with expected error. Actual error: %v", err)
- }
- }
- func TestUnknownEndpoint(t *testing.T) {
- defer netutils.SetupTestNetNS(t)()
- ip, subnet, err := net.ParseCIDR("192.168.100.1/24")
- if err != nil {
- t.Fatal(err)
- }
- subnet.IP = ip
- option := options.Generic{
- "BridgeName": bridgeName,
- "AddressIPv4": subnet,
- "AllowNonDefaultBridge": true}
- network, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, option)
- if err != nil {
- t.Fatal(err)
- }
- ep, err := network.CreateEndpoint("testep")
- if err != nil {
- t.Fatal(err)
- }
- err = ep.Delete()
- if err != nil {
- t.Fatal(err)
- }
- err = ep.Delete()
- if err == nil {
- t.Fatal("Expected to fail. But instead succeeded")
- }
- if _, ok := err.(*libnetwork.UnknownEndpointError); !ok {
- t.Fatalf("Did not fail with expected error. Actual error: %v", err)
- }
- // Done testing. Now cleanup
- if err := network.Delete(); err != nil {
- t.Fatal(err)
- }
- }
- func TestNetworkEndpointsWalkers(t *testing.T) {
- defer netutils.SetupTestNetNS(t)()
- controller := libnetwork.New()
- err := controller.ConfigureNetworkDriver(bridgeNetType, getEmptyGenericOption())
- if err != nil {
- t.Fatal(err)
- }
- // Create network 1 and add 2 endpoint: ep11, ep12
- net1, err := controller.NewNetwork(bridgeNetType, "network1")
- if err != nil {
- t.Fatal(err)
- }
- ep11, err := net1.CreateEndpoint("ep11")
- if err != nil {
- t.Fatal(err)
- }
- ep12, err := net1.CreateEndpoint("ep12")
- if err != nil {
- t.Fatal(err)
- }
- // Test list methods on net1
- epList1 := net1.Endpoints()
- if len(epList1) != 2 {
- t.Fatalf("Endpoints() returned wrong number of elements: %d instead of 2", len(epList1))
- }
- // endpoint order is not guaranteed
- for _, e := range epList1 {
- if e != ep11 && e != ep12 {
- t.Fatal("Endpoints() did not return all the expected elements")
- }
- }
- // Test Endpoint Walk method
- var epName string
- var epWanted libnetwork.Endpoint
- wlk := func(ep libnetwork.Endpoint) bool {
- if ep.Name() == epName {
- epWanted = ep
- return true
- }
- return false
- }
- // Look for ep1 on network1
- epName = "ep11"
- net1.WalkEndpoints(wlk)
- if epWanted == nil {
- t.Fatal(err)
- }
- if ep11 != epWanted {
- t.Fatal(err)
- }
- // Test Network Walk method
- var netName string
- var netWanted libnetwork.Network
- nwWlk := func(nw libnetwork.Network) bool {
- if nw.Name() == netName {
- netWanted = nw
- return true
- }
- return false
- }
- // Look for network named "network1"
- netName = "network1"
- controller.WalkNetworks(nwWlk)
- if netWanted == nil {
- t.Fatal(err)
- }
- if net1 != netWanted {
- t.Fatal(err)
- }
- }
- func TestControllerQuery(t *testing.T) {
- defer netutils.SetupTestNetNS(t)()
- controller := libnetwork.New()
- err := controller.ConfigureNetworkDriver(bridgeNetType, getEmptyGenericOption())
- if err != nil {
- t.Fatal(err)
- }
- // Create network 1
- net1, err := controller.NewNetwork(bridgeNetType, "network1")
- if err != nil {
- t.Fatal(err)
- }
- g := controller.NetworkByName("")
- if g != nil {
- t.Fatalf("NetworkByName() succeeded with invalid target name")
- }
- g = controller.NetworkByID("")
- if g != nil {
- t.Fatalf("NetworkByID() succeeded with invalid target id: %v", g)
- }
- g = controller.NetworkByID("network1")
- if g != nil {
- t.Fatalf("NetworkByID() succeeded with invalid target name")
- }
- g = controller.NetworkByName("network1")
- if g == nil {
- t.Fatalf("NetworkByName() did not find the network")
- }
- if g != net1 {
- t.Fatalf("NetworkByName() returned the wrong network")
- }
- g = controller.NetworkByID(net1.ID())
- if net1 != g {
- t.Fatalf("NetworkByID() returned unexpected element: %v", g)
- }
- }
- func TestNetworkQuery(t *testing.T) {
- defer netutils.SetupTestNetNS(t)()
- controller := libnetwork.New()
- err := controller.ConfigureNetworkDriver(bridgeNetType, getEmptyGenericOption())
- if err != nil {
- t.Fatal(err)
- }
- // Create network 1 and add 2 endpoint: ep11, ep12
- net1, err := controller.NewNetwork(bridgeNetType, "network1")
- if err != nil {
- t.Fatal(err)
- }
- ep11, err := net1.CreateEndpoint("ep11")
- if err != nil {
- t.Fatal(err)
- }
- ep12, err := net1.CreateEndpoint("ep12")
- if err != nil {
- t.Fatal(err)
- }
- e := net1.EndpointByName("ep11")
- if ep11 != e {
- t.Fatalf("EndpointByName() returned %v instead of %v", e, ep11)
- }
- e = net1.EndpointByName("")
- if e != nil {
- t.Fatalf("EndpointByName(): expected nil, got %v", e)
- }
- e = net1.EndpointByName("IamNotAnEndpoint")
- if e != nil {
- t.Fatalf("EndpointByName(): expected nil, got %v", e)
- }
- e = net1.EndpointByID(ep12.ID())
- if ep12 != e {
- t.Fatalf("EndpointByID() returned %v instead of %v", e, ep12)
- }
- e = net1.EndpointByID("")
- if e != nil {
- t.Fatalf("EndpointByID(): expected nil, got %v", e)
- }
- }
- const containerID = "valid_container"
- func TestEndpointJoin(t *testing.T) {
- defer netutils.SetupTestNetNS(t)()
- n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
- if err != nil {
- t.Fatal(err)
- }
- ep, err := n.CreateEndpoint("ep1")
- if err != nil {
- t.Fatal(err)
- }
- _, err = ep.Join(containerID,
- libnetwork.JoinOptionHostname("test"),
- libnetwork.JoinOptionDomainname("docker.io"),
- libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
- if err != nil {
- t.Fatal(err)
- }
- defer func() {
- err = ep.Leave(containerID)
- if err != nil {
- t.Fatal(err)
- }
- }()
- }
- func TestEndpointJoinInvalidContainerId(t *testing.T) {
- defer netutils.SetupTestNetNS(t)()
- n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
- if err != nil {
- t.Fatal(err)
- }
- ep, err := n.CreateEndpoint("ep1")
- if err != nil {
- t.Fatal(err)
- }
- _, err = ep.Join("")
- if err == nil {
- t.Fatal("Expected to fail join with empty container id string")
- }
- if _, ok := err.(libnetwork.InvalidContainerIDError); !ok {
- t.Fatalf("Failed for unexpected reason: %v", err)
- }
- }
- func TestEndpointDeleteWithActiveContainer(t *testing.T) {
- defer netutils.SetupTestNetNS(t)()
- n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
- if err != nil {
- t.Fatal(err)
- }
- ep, err := n.CreateEndpoint("ep1")
- if err != nil {
- t.Fatal(err)
- }
- _, err = ep.Join(containerID,
- libnetwork.JoinOptionHostname("test"),
- libnetwork.JoinOptionDomainname("docker.io"),
- libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
- if err != nil {
- t.Fatal(err)
- }
- defer func() {
- err = ep.Leave(containerID)
- if err != nil {
- t.Fatal(err)
- }
- err = ep.Delete()
- if err != nil {
- t.Fatal(err)
- }
- }()
- err = ep.Delete()
- if err == nil {
- t.Fatal("Expected to fail. But instead succeeded")
- }
- if _, ok := err.(*libnetwork.ActiveContainerError); !ok {
- t.Fatalf("Did not fail with expected error. Actual error: %v", err)
- }
- }
- func TestEndpointMultipleJoins(t *testing.T) {
- defer netutils.SetupTestNetNS(t)()
- n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
- if err != nil {
- t.Fatal(err)
- }
- ep, err := n.CreateEndpoint("ep1")
- if err != nil {
- t.Fatal(err)
- }
- _, err = ep.Join(containerID,
- libnetwork.JoinOptionHostname("test"),
- libnetwork.JoinOptionDomainname("docker.io"),
- libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
- if err != nil {
- t.Fatal(err)
- }
- defer func() {
- err = ep.Leave(containerID)
- if err != nil {
- t.Fatal(err)
- }
- }()
- _, err = ep.Join("container2")
- if err == nil {
- t.Fatal("Expected to fail multiple joins for the same endpoint")
- }
- if err != libnetwork.ErrInvalidJoin {
- t.Fatalf("Failed for unexpected reason: %v", err)
- }
- }
- func TestEndpointInvalidLeave(t *testing.T) {
- defer netutils.SetupTestNetNS(t)()
- n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
- if err != nil {
- t.Fatal(err)
- }
- ep, err := n.CreateEndpoint("ep1")
- if err != nil {
- t.Fatal(err)
- }
- err = ep.Leave(containerID)
- if err == nil {
- t.Fatal("Expected to fail leave from an endpoint which has no active join")
- }
- if _, ok := err.(libnetwork.InvalidContainerIDError); !ok {
- if err != libnetwork.ErrNoContainer {
- t.Fatalf("Failed for unexpected reason: %v", err)
- }
- }
- _, err = ep.Join(containerID,
- libnetwork.JoinOptionHostname("test"),
- libnetwork.JoinOptionDomainname("docker.io"),
- libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
- if err != nil {
- t.Fatal(err)
- }
- defer func() {
- err = ep.Leave(containerID)
- if err != nil {
- t.Fatal(err)
- }
- }()
- err = ep.Leave("")
- if err == nil {
- t.Fatal("Expected to fail leave with empty container id")
- }
- if _, ok := err.(libnetwork.InvalidContainerIDError); !ok {
- t.Fatalf("Failed for unexpected reason: %v", err)
- }
- err = ep.Leave("container2")
- if err == nil {
- t.Fatal("Expected to fail leave with wrong container id")
- }
- if _, ok := err.(libnetwork.InvalidContainerIDError); !ok {
- t.Fatalf("Failed for unexpected reason: %v", err)
- }
- }
- func TestEndpointUpdateParent(t *testing.T) {
- defer netutils.SetupTestNetNS(t)()
- n, err := createTestNetwork("bridge", "testnetwork", options.Generic{}, options.Generic{})
- if err != nil {
- t.Fatal(err)
- }
- ep1, err := n.CreateEndpoint("ep1", nil)
- if err != nil {
- t.Fatal(err)
- }
- _, err = ep1.Join(containerID,
- libnetwork.JoinOptionHostname("test1"),
- libnetwork.JoinOptionDomainname("docker.io"),
- libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
- if err != nil {
- t.Fatal(err)
- }
- defer func() {
- err = ep1.Leave(containerID)
- if err != nil {
- t.Fatal(err)
- }
- }()
- ep2, err := n.CreateEndpoint("ep2", nil)
- if err != nil {
- t.Fatal(err)
- }
- _, err = ep2.Join("container2",
- libnetwork.JoinOptionHostname("test2"),
- libnetwork.JoinOptionDomainname("docker.io"),
- libnetwork.JoinOptionHostsPath("/var/lib/docker/test_network/container2/hosts"),
- libnetwork.JoinOptionParentUpdate(ep1.ID(), "web", "192.168.0.2"))
- if err != nil {
- t.Fatal(err)
- }
- defer func() {
- err = ep2.Leave("container2")
- if err != nil {
- t.Fatal(err)
- }
- }()
- }
- func TestEnableIPv6(t *testing.T) {
- defer netutils.SetupTestNetNS(t)()
- tmpResolvConf := []byte("search pommesfrites.fr\nnameserver 12.34.56.78\nnameserver 2001:4860:4860::8888")
- //take a copy of resolv.conf for restoring after test completes
- resolvConfSystem, err := ioutil.ReadFile("/etc/resolv.conf")
- if err != nil {
- t.Fatal(err)
- }
- //cleanup
- defer func() {
- if err := ioutil.WriteFile("/etc/resolv.conf", resolvConfSystem, 0644); err != nil {
- t.Fatal(err)
- }
- }()
- netOption := options.Generic{
- netlabel.EnableIPv6: true,
- }
- n, err := createTestNetwork("bridge", "testnetwork", options.Generic{}, netOption)
- if err != nil {
- t.Fatal(err)
- }
- ep1, err := n.CreateEndpoint("ep1", nil)
- if err != nil {
- t.Fatal(err)
- }
- if err := ioutil.WriteFile("/etc/resolv.conf", tmpResolvConf, 0644); err != nil {
- t.Fatal(err)
- }
- resolvConfPath := "/tmp/libnetwork_test/resolv.conf"
- _, err = ep1.Join(containerID,
- libnetwork.JoinOptionResolvConfPath(resolvConfPath))
- if err != nil {
- t.Fatal(err)
- }
- defer func() {
- err = ep1.Leave(containerID)
- if err != nil {
- t.Fatal(err)
- }
- }()
- content, err := ioutil.ReadFile(resolvConfPath)
- if err != nil {
- t.Fatal(err)
- }
- if !bytes.Equal(content, tmpResolvConf) {
- t.Fatalf("Expected %s, Got %s", string(tmpResolvConf), string(content))
- }
- if err != nil {
- t.Fatal(err)
- }
- }
- func TestNoEnableIPv6(t *testing.T) {
- defer netutils.SetupTestNetNS(t)()
- tmpResolvConf := []byte("search pommesfrites.fr\nnameserver 12.34.56.78\nnameserver 2001:4860:4860::8888")
- expectedResolvConf := []byte("search pommesfrites.fr\nnameserver 12.34.56.78\n")
- //take a copy of resolv.conf for restoring after test completes
- resolvConfSystem, err := ioutil.ReadFile("/etc/resolv.conf")
- if err != nil {
- t.Fatal(err)
- }
- //cleanup
- defer func() {
- if err := ioutil.WriteFile("/etc/resolv.conf", resolvConfSystem, 0644); err != nil {
- t.Fatal(err)
- }
- }()
- n, err := createTestNetwork("bridge", "testnetwork", options.Generic{}, options.Generic{})
- if err != nil {
- t.Fatal(err)
- }
- ep1, err := n.CreateEndpoint("ep1", nil)
- if err != nil {
- t.Fatal(err)
- }
- if err := ioutil.WriteFile("/etc/resolv.conf", tmpResolvConf, 0644); err != nil {
- t.Fatal(err)
- }
- resolvConfPath := "/tmp/libnetwork_test/resolv.conf"
- _, err = ep1.Join(containerID,
- libnetwork.JoinOptionResolvConfPath(resolvConfPath))
- if err != nil {
- t.Fatal(err)
- }
- defer func() {
- err = ep1.Leave(containerID)
- if err != nil {
- t.Fatal(err)
- }
- }()
- content, err := ioutil.ReadFile(resolvConfPath)
- if err != nil {
- t.Fatal(err)
- }
- if !bytes.Equal(content, expectedResolvConf) {
- t.Fatalf("Expected %s, Got %s", string(expectedResolvConf), string(content))
- }
- if err != nil {
- t.Fatal(err)
- }
- }
- var (
- once sync.Once
- ctrlr libnetwork.NetworkController
- start = make(chan struct{})
- done = make(chan chan struct{}, numThreads-1)
- origns = netns.None()
- testns = netns.None()
- )
- const (
- iterCnt = 25
- numThreads = 3
- first = 1
- last = numThreads
- debug = false
- )
- func createGlobalInstance(t *testing.T) {
- var err error
- defer close(start)
- origns, err = netns.Get()
- if err != nil {
- t.Fatal(err)
- }
- //testns = origns
- testns, err = netns.New()
- if err != nil {
- t.Fatal(err)
- }
- ctrlr = libnetwork.New()
- err = ctrlr.ConfigureNetworkDriver(bridgeNetType, getEmptyGenericOption())
- if err != nil {
- t.Fatal("configure driver")
- }
- net, err := ctrlr.NewNetwork(bridgeNetType, "network1")
- if err != nil {
- t.Fatal("new network")
- }
- _, err = net.CreateEndpoint("ep1")
- if err != nil {
- t.Fatal("createendpoint")
- }
- }
- func debugf(format string, a ...interface{}) (int, error) {
- if debug {
- return fmt.Printf(format, a...)
- }
- return 0, nil
- }
- func parallelJoin(t *testing.T, ep libnetwork.Endpoint, thrNumber int) {
- debugf("J%d.", thrNumber)
- _, err := ep.Join("racing_container")
- runtime.LockOSThread()
- if err != nil {
- if err != libnetwork.ErrNoContainer && err != libnetwork.ErrInvalidJoin {
- t.Fatal(err)
- }
- debugf("JE%d(%v).", thrNumber, err)
- }
- debugf("JD%d.", thrNumber)
- }
- func parallelLeave(t *testing.T, ep libnetwork.Endpoint, thrNumber int) {
- debugf("L%d.", thrNumber)
- err := ep.Leave("racing_container")
- runtime.LockOSThread()
- if err != nil {
- if err != libnetwork.ErrNoContainer && err != libnetwork.ErrInvalidJoin {
- t.Fatal(err)
- }
- debugf("LE%d(%v).", thrNumber, err)
- }
- debugf("LD%d.", thrNumber)
- }
- func runParallelTests(t *testing.T, thrNumber int) {
- var err error
- t.Parallel()
- pTest := flag.Lookup("test.parallel")
- if pTest == nil {
- t.Skip("Skipped because test.parallel flag not set;")
- }
- numParallel, err := strconv.Atoi(pTest.Value.String())
- if err != nil {
- t.Fatal(err)
- }
- if numParallel < numThreads {
- t.Skip("Skipped because t.parallel was less than ", numThreads)
- }
- runtime.LockOSThread()
- defer runtime.UnlockOSThread()
- if thrNumber == first {
- createGlobalInstance(t)
- }
- if thrNumber != first {
- select {
- case <-start:
- }
- thrdone := make(chan struct{})
- done <- thrdone
- defer close(thrdone)
- if thrNumber == last {
- defer close(done)
- }
- err = netns.Set(testns)
- if err != nil {
- t.Fatal(err)
- }
- }
- defer netns.Set(origns)
- net := ctrlr.NetworkByName("network1")
- if net == nil {
- t.Fatal("Could not find network1")
- }
- ep := net.EndpointByName("ep1")
- if ep == nil {
- t.Fatal("Could not find ep1")
- }
- for i := 0; i < iterCnt; i++ {
- parallelJoin(t, ep, thrNumber)
- parallelLeave(t, ep, thrNumber)
- }
- debugf("\n")
- if thrNumber == first {
- for thrdone := range done {
- select {
- case <-thrdone:
- }
- }
- testns.Close()
- err = ep.Delete()
- if err != nil {
- t.Fatal(err)
- }
- }
- }
- func TestParallel1(t *testing.T) {
- runParallelTests(t, 1)
- }
- func TestParallel2(t *testing.T) {
- runParallelTests(t, 2)
- }
- func TestParallel3(t *testing.T) {
- runParallelTests(t, 3)
- }
|