12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178 |
- 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)()
- _, err := createTestNetwork(bridgeNetType, "", options.Generic{}, options.Generic{})
- if err == nil {
- t.Fatal("Expected to fail. But instead succeeded")
- }
- if err != libnetwork.ErrInvalidName {
- t.Fatal("Expected to fail with ErrInvalidName error")
- }
- 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)
- }
- _, err = network.CreateEndpoint("")
- if err == nil {
- t.Fatal("Expected to fail. But instead succeeded")
- }
- if err != libnetwork.ErrInvalidName {
- t.Fatal("Expected to fail with ErrInvalidName error")
- }
- 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)
- }
- _, err = controller.NetworkByName("")
- if err == nil {
- t.Fatalf("NetworkByName() succeeded with invalid target name")
- }
- if err != libnetwork.ErrInvalidName {
- t.Fatalf("NetworkByName() failed with unexpected error: %v", err)
- }
- _, err = controller.NetworkByID("")
- if err == nil {
- t.Fatalf("NetworkByID() succeeded with invalid target id")
- }
- if err != libnetwork.ErrInvalidID {
- t.Fatalf("NetworkByID() failed with unexpected error: %v", err)
- }
- g, err := controller.NetworkByID("network1")
- if err != nil {
- t.Fatalf("Unexpected failure for NetworkByID(): %v", err)
- }
- if g != nil {
- t.Fatalf("NetworkByID() succeeded with unknown target id")
- }
- g, err = controller.NetworkByName("network1")
- if err != nil {
- t.Fatalf("Unexpected failure for NetworkByName(): %v", err)
- }
- if g == nil {
- t.Fatalf("NetworkByName() did not find the network")
- }
- if g != net1 {
- t.Fatalf("NetworkByName() returned the wrong network")
- }
- g, err = controller.NetworkByID(net1.ID())
- if err != nil {
- t.Fatalf("Unexpected failure for NetworkByID(): %v", err)
- }
- 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, err := net1.EndpointByName("ep11")
- if err != nil {
- t.Fatal(err)
- }
- if ep11 != e {
- t.Fatalf("EndpointByName() returned %v instead of %v", e, ep11)
- }
- e, err = net1.EndpointByName("")
- if err == nil {
- t.Fatalf("EndpointByName() succeeded with invalid target name")
- }
- if err != libnetwork.ErrInvalidName {
- t.Fatalf("EndpointByName() failed with unexpected error: %v", err)
- }
- e, err = net1.EndpointByName("IamNotAnEndpoint")
- if err != nil {
- t.Fatal(err)
- }
- if e != nil {
- t.Fatalf("EndpointByName(): expected nil, got %v", e)
- }
- e, err = net1.EndpointByID(ep12.ID())
- if ep12 != e {
- t.Fatalf("EndpointByID() returned %v instead of %v", e, ep12)
- }
- e, err = net1.EndpointByID("")
- if err == nil {
- t.Fatalf("EndpointByID() succeeded with invalid target id")
- }
- if err != libnetwork.ErrInvalidID {
- t.Fatalf("EndpointByID() failed with unexpected error: %v", err)
- }
- }
- 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, err := ctrlr.NetworkByName("network1")
- if err != nil {
- t.Fatal(err)
- }
- if net == nil {
- t.Fatal("Could not find network1")
- }
- ep, err := net.EndpointByName("ep1")
- if err != nil {
- t.Fatal(err)
- }
- 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)
- }
|