libnetwork_test.go 25 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178
  1. package libnetwork_test
  2. import (
  3. "bytes"
  4. "flag"
  5. "fmt"
  6. "io/ioutil"
  7. "net"
  8. "os"
  9. "runtime"
  10. "strconv"
  11. "sync"
  12. "testing"
  13. log "github.com/Sirupsen/logrus"
  14. "github.com/docker/docker/pkg/reexec"
  15. "github.com/docker/libnetwork"
  16. "github.com/docker/libnetwork/netutils"
  17. "github.com/docker/libnetwork/pkg/netlabel"
  18. "github.com/docker/libnetwork/pkg/options"
  19. "github.com/vishvananda/netns"
  20. )
  21. const (
  22. bridgeNetType = "bridge"
  23. bridgeName = "docker0"
  24. )
  25. func TestMain(m *testing.M) {
  26. if reexec.Init() {
  27. return
  28. }
  29. os.Exit(m.Run())
  30. }
  31. func createTestNetwork(networkType, networkName string, option options.Generic, netOption options.Generic) (libnetwork.Network, error) {
  32. controller := libnetwork.New()
  33. genericOption := make(map[string]interface{})
  34. genericOption[netlabel.GenericData] = option
  35. err := controller.ConfigureNetworkDriver(networkType, genericOption)
  36. if err != nil {
  37. return nil, err
  38. }
  39. network, err := controller.NewNetwork(networkType, networkName,
  40. libnetwork.NetworkOptionGeneric(netOption))
  41. if err != nil {
  42. return nil, err
  43. }
  44. return network, nil
  45. }
  46. func getEmptyGenericOption() map[string]interface{} {
  47. genericOption := make(map[string]interface{})
  48. genericOption[netlabel.GenericData] = options.Generic{}
  49. return genericOption
  50. }
  51. func getPortMapping() []netutils.PortBinding {
  52. return []netutils.PortBinding{
  53. netutils.PortBinding{Proto: netutils.TCP, Port: uint16(230), HostPort: uint16(23000)},
  54. netutils.PortBinding{Proto: netutils.UDP, Port: uint16(200), HostPort: uint16(22000)},
  55. netutils.PortBinding{Proto: netutils.TCP, Port: uint16(120), HostPort: uint16(12000)},
  56. }
  57. }
  58. func TestNull(t *testing.T) {
  59. network, err := createTestNetwork("null", "testnetwork", options.Generic{},
  60. options.Generic{})
  61. if err != nil {
  62. t.Fatal(err)
  63. }
  64. ep, err := network.CreateEndpoint("testep")
  65. if err != nil {
  66. t.Fatal(err)
  67. }
  68. _, err = ep.Join("null_container",
  69. libnetwork.JoinOptionHostname("test"),
  70. libnetwork.JoinOptionDomainname("docker.io"),
  71. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
  72. if err != nil {
  73. t.Fatal(err)
  74. }
  75. err = ep.Leave("null_container")
  76. if err != nil {
  77. t.Fatal(err)
  78. }
  79. if err := ep.Delete(); err != nil {
  80. t.Fatal(err)
  81. }
  82. if err := network.Delete(); err != nil {
  83. t.Fatal(err)
  84. }
  85. }
  86. func TestHost(t *testing.T) {
  87. network, err := createTestNetwork("host", "testnetwork", options.Generic{}, options.Generic{})
  88. if err != nil {
  89. t.Fatal(err)
  90. }
  91. ep, err := network.CreateEndpoint("testep")
  92. if err != nil {
  93. t.Fatal(err)
  94. }
  95. _, err = ep.Join("host_container",
  96. libnetwork.JoinOptionHostname("test"),
  97. libnetwork.JoinOptionDomainname("docker.io"),
  98. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"),
  99. libnetwork.JoinOptionUseDefaultSandbox())
  100. if err != nil {
  101. t.Fatal(err)
  102. }
  103. err = ep.Leave("host_container")
  104. if err != nil {
  105. t.Fatal(err)
  106. }
  107. if err := ep.Delete(); err != nil {
  108. t.Fatal(err)
  109. }
  110. if err := network.Delete(); err != nil {
  111. t.Fatal(err)
  112. }
  113. }
  114. func TestBridge(t *testing.T) {
  115. defer netutils.SetupTestNetNS(t)()
  116. ip, subnet, err := net.ParseCIDR("192.168.100.1/24")
  117. if err != nil {
  118. t.Fatal(err)
  119. }
  120. subnet.IP = ip
  121. ip, cidr, err := net.ParseCIDR("192.168.100.2/28")
  122. if err != nil {
  123. t.Fatal(err)
  124. }
  125. cidr.IP = ip
  126. ip, cidrv6, err := net.ParseCIDR("fe90::1/96")
  127. if err != nil {
  128. t.Fatal(err)
  129. }
  130. cidrv6.IP = ip
  131. log.Debug("Adding a bridge")
  132. option := options.Generic{
  133. "EnableIPForwarding": true,
  134. }
  135. netOption := options.Generic{
  136. "BridgeName": bridgeName,
  137. "AddressIPv4": subnet,
  138. "FixedCIDR": cidr,
  139. "FixedCIDRv6": cidrv6,
  140. "EnableIPv6": true,
  141. "EnableIPTables": true,
  142. "EnableIPMasquerade": true,
  143. "EnableICC": true,
  144. "AllowNonDefaultBridge": true}
  145. network, err := createTestNetwork(bridgeNetType, "testnetwork", option, netOption)
  146. if err != nil {
  147. t.Fatal(err)
  148. }
  149. ep, err := network.CreateEndpoint("testep", libnetwork.CreateOptionPortMapping(getPortMapping()))
  150. if err != nil {
  151. t.Fatal(err)
  152. }
  153. epInfo, err := ep.Info()
  154. if err != nil {
  155. t.Fatal(err)
  156. }
  157. pmd, ok := epInfo[netlabel.PortMap]
  158. if !ok {
  159. t.Fatalf("Could not find expected info in endpoint data")
  160. }
  161. pm, ok := pmd.([]netutils.PortBinding)
  162. if !ok {
  163. t.Fatalf("Unexpected format for port mapping in endpoint operational data")
  164. }
  165. if len(pm) != 3 {
  166. t.Fatalf("Incomplete data for port mapping in endpoint operational data: %d", len(pm))
  167. }
  168. if err := ep.Delete(); err != nil {
  169. t.Fatal(err)
  170. }
  171. if err := network.Delete(); err != nil {
  172. t.Fatal(err)
  173. }
  174. }
  175. func TestUnknownDriver(t *testing.T) {
  176. defer netutils.SetupTestNetNS(t)()
  177. _, err := createTestNetwork("unknowndriver", "testnetwork", options.Generic{}, options.Generic{})
  178. if err == nil {
  179. t.Fatal("Expected to fail. But instead succeeded")
  180. }
  181. if _, ok := err.(libnetwork.NetworkTypeError); !ok {
  182. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  183. }
  184. }
  185. func TestNilDriver(t *testing.T) {
  186. controller := libnetwork.New()
  187. _, err := controller.NewNetwork("framerelay", "dummy",
  188. libnetwork.NetworkOptionGeneric(getEmptyGenericOption()))
  189. if err == nil {
  190. t.Fatal("Expected to fail. But instead succeeded")
  191. }
  192. if err != libnetwork.ErrInvalidNetworkDriver {
  193. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  194. }
  195. }
  196. func TestNoInitDriver(t *testing.T) {
  197. controller := libnetwork.New()
  198. _, err := controller.NewNetwork("ppp", "dummy",
  199. libnetwork.NetworkOptionGeneric(getEmptyGenericOption()))
  200. if err == nil {
  201. t.Fatal("Expected to fail. But instead succeeded")
  202. }
  203. if err != libnetwork.ErrInvalidNetworkDriver {
  204. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  205. }
  206. }
  207. func TestDuplicateNetwork(t *testing.T) {
  208. defer netutils.SetupTestNetNS(t)()
  209. controller := libnetwork.New()
  210. genericOption := make(map[string]interface{})
  211. genericOption[netlabel.GenericData] = options.Generic{}
  212. err := controller.ConfigureNetworkDriver(bridgeNetType, genericOption)
  213. if err != nil {
  214. t.Fatal(err)
  215. }
  216. _, err = controller.NewNetwork(bridgeNetType, "testnetwork",
  217. libnetwork.NetworkOptionGeneric(genericOption))
  218. if err != nil {
  219. t.Fatal(err)
  220. }
  221. _, err = controller.NewNetwork(bridgeNetType, "testnetwork")
  222. if err == nil {
  223. t.Fatal("Expected to fail. But instead succeeded")
  224. }
  225. if _, ok := err.(libnetwork.NetworkNameError); !ok {
  226. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  227. }
  228. }
  229. func TestNetworkName(t *testing.T) {
  230. defer netutils.SetupTestNetNS(t)()
  231. _, err := createTestNetwork(bridgeNetType, "", options.Generic{}, options.Generic{})
  232. if err == nil {
  233. t.Fatal("Expected to fail. But instead succeeded")
  234. }
  235. if err != libnetwork.ErrInvalidName {
  236. t.Fatal("Expected to fail with ErrInvalidName error")
  237. }
  238. networkName := "testnetwork"
  239. n, err := createTestNetwork(bridgeNetType, networkName, options.Generic{}, options.Generic{})
  240. if err != nil {
  241. t.Fatal(err)
  242. }
  243. if n.Name() != networkName {
  244. t.Fatalf("Expected network name %s, got %s", networkName, n.Name())
  245. }
  246. }
  247. func TestNetworkType(t *testing.T) {
  248. defer netutils.SetupTestNetNS(t)()
  249. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  250. if err != nil {
  251. t.Fatal(err)
  252. }
  253. if n.Type() != bridgeNetType {
  254. t.Fatalf("Expected network type %s, got %s", bridgeNetType, n.Type())
  255. }
  256. }
  257. func TestNetworkID(t *testing.T) {
  258. defer netutils.SetupTestNetNS(t)()
  259. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  260. if err != nil {
  261. t.Fatal(err)
  262. }
  263. if n.ID() == "" {
  264. t.Fatal("Expected non-empty network id")
  265. }
  266. }
  267. func TestDeleteNetworkWithActiveEndpoints(t *testing.T) {
  268. defer netutils.SetupTestNetNS(t)()
  269. option := options.Generic{
  270. "BridgeName": bridgeName,
  271. "AllowNonDefaultBridge": true}
  272. network, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, option)
  273. if err != nil {
  274. t.Fatal(err)
  275. }
  276. ep, err := network.CreateEndpoint("testep")
  277. if err != nil {
  278. t.Fatal(err)
  279. }
  280. err = network.Delete()
  281. if err == nil {
  282. t.Fatal("Expected to fail. But instead succeeded")
  283. }
  284. if _, ok := err.(*libnetwork.ActiveEndpointsError); !ok {
  285. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  286. }
  287. // Done testing. Now cleanup.
  288. if err := ep.Delete(); err != nil {
  289. t.Fatal(err)
  290. }
  291. if err := network.Delete(); err != nil {
  292. t.Fatal(err)
  293. }
  294. }
  295. func TestUnknownNetwork(t *testing.T) {
  296. defer netutils.SetupTestNetNS(t)()
  297. option := options.Generic{
  298. "BridgeName": bridgeName,
  299. "AllowNonDefaultBridge": true}
  300. network, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, option)
  301. if err != nil {
  302. t.Fatal(err)
  303. }
  304. err = network.Delete()
  305. if err != nil {
  306. t.Fatal(err)
  307. }
  308. err = network.Delete()
  309. if err == nil {
  310. t.Fatal("Expected to fail. But instead succeeded")
  311. }
  312. if _, ok := err.(*libnetwork.UnknownNetworkError); !ok {
  313. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  314. }
  315. }
  316. func TestUnknownEndpoint(t *testing.T) {
  317. defer netutils.SetupTestNetNS(t)()
  318. ip, subnet, err := net.ParseCIDR("192.168.100.1/24")
  319. if err != nil {
  320. t.Fatal(err)
  321. }
  322. subnet.IP = ip
  323. option := options.Generic{
  324. "BridgeName": bridgeName,
  325. "AddressIPv4": subnet,
  326. "AllowNonDefaultBridge": true}
  327. network, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, option)
  328. if err != nil {
  329. t.Fatal(err)
  330. }
  331. _, err = network.CreateEndpoint("")
  332. if err == nil {
  333. t.Fatal("Expected to fail. But instead succeeded")
  334. }
  335. if err != libnetwork.ErrInvalidName {
  336. t.Fatal("Expected to fail with ErrInvalidName error")
  337. }
  338. ep, err := network.CreateEndpoint("testep")
  339. if err != nil {
  340. t.Fatal(err)
  341. }
  342. err = ep.Delete()
  343. if err != nil {
  344. t.Fatal(err)
  345. }
  346. err = ep.Delete()
  347. if err == nil {
  348. t.Fatal("Expected to fail. But instead succeeded")
  349. }
  350. if _, ok := err.(*libnetwork.UnknownEndpointError); !ok {
  351. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  352. }
  353. // Done testing. Now cleanup
  354. if err := network.Delete(); err != nil {
  355. t.Fatal(err)
  356. }
  357. }
  358. func TestNetworkEndpointsWalkers(t *testing.T) {
  359. defer netutils.SetupTestNetNS(t)()
  360. controller := libnetwork.New()
  361. err := controller.ConfigureNetworkDriver(bridgeNetType, getEmptyGenericOption())
  362. if err != nil {
  363. t.Fatal(err)
  364. }
  365. // Create network 1 and add 2 endpoint: ep11, ep12
  366. net1, err := controller.NewNetwork(bridgeNetType, "network1")
  367. if err != nil {
  368. t.Fatal(err)
  369. }
  370. ep11, err := net1.CreateEndpoint("ep11")
  371. if err != nil {
  372. t.Fatal(err)
  373. }
  374. ep12, err := net1.CreateEndpoint("ep12")
  375. if err != nil {
  376. t.Fatal(err)
  377. }
  378. // Test list methods on net1
  379. epList1 := net1.Endpoints()
  380. if len(epList1) != 2 {
  381. t.Fatalf("Endpoints() returned wrong number of elements: %d instead of 2", len(epList1))
  382. }
  383. // endpoint order is not guaranteed
  384. for _, e := range epList1 {
  385. if e != ep11 && e != ep12 {
  386. t.Fatal("Endpoints() did not return all the expected elements")
  387. }
  388. }
  389. // Test Endpoint Walk method
  390. var epName string
  391. var epWanted libnetwork.Endpoint
  392. wlk := func(ep libnetwork.Endpoint) bool {
  393. if ep.Name() == epName {
  394. epWanted = ep
  395. return true
  396. }
  397. return false
  398. }
  399. // Look for ep1 on network1
  400. epName = "ep11"
  401. net1.WalkEndpoints(wlk)
  402. if epWanted == nil {
  403. t.Fatal(err)
  404. }
  405. if ep11 != epWanted {
  406. t.Fatal(err)
  407. }
  408. // Test Network Walk method
  409. var netName string
  410. var netWanted libnetwork.Network
  411. nwWlk := func(nw libnetwork.Network) bool {
  412. if nw.Name() == netName {
  413. netWanted = nw
  414. return true
  415. }
  416. return false
  417. }
  418. // Look for network named "network1"
  419. netName = "network1"
  420. controller.WalkNetworks(nwWlk)
  421. if netWanted == nil {
  422. t.Fatal(err)
  423. }
  424. if net1 != netWanted {
  425. t.Fatal(err)
  426. }
  427. }
  428. func TestControllerQuery(t *testing.T) {
  429. defer netutils.SetupTestNetNS(t)()
  430. controller := libnetwork.New()
  431. err := controller.ConfigureNetworkDriver(bridgeNetType, getEmptyGenericOption())
  432. if err != nil {
  433. t.Fatal(err)
  434. }
  435. // Create network 1
  436. net1, err := controller.NewNetwork(bridgeNetType, "network1")
  437. if err != nil {
  438. t.Fatal(err)
  439. }
  440. _, err = controller.NetworkByName("")
  441. if err == nil {
  442. t.Fatalf("NetworkByName() succeeded with invalid target name")
  443. }
  444. if err != libnetwork.ErrInvalidName {
  445. t.Fatalf("NetworkByName() failed with unexpected error: %v", err)
  446. }
  447. _, err = controller.NetworkByID("")
  448. if err == nil {
  449. t.Fatalf("NetworkByID() succeeded with invalid target id")
  450. }
  451. if err != libnetwork.ErrInvalidID {
  452. t.Fatalf("NetworkByID() failed with unexpected error: %v", err)
  453. }
  454. g, err := controller.NetworkByID("network1")
  455. if err != nil {
  456. t.Fatalf("Unexpected failure for NetworkByID(): %v", err)
  457. }
  458. if g != nil {
  459. t.Fatalf("NetworkByID() succeeded with unknown target id")
  460. }
  461. g, err = controller.NetworkByName("network1")
  462. if err != nil {
  463. t.Fatalf("Unexpected failure for NetworkByName(): %v", err)
  464. }
  465. if g == nil {
  466. t.Fatalf("NetworkByName() did not find the network")
  467. }
  468. if g != net1 {
  469. t.Fatalf("NetworkByName() returned the wrong network")
  470. }
  471. g, err = controller.NetworkByID(net1.ID())
  472. if err != nil {
  473. t.Fatalf("Unexpected failure for NetworkByID(): %v", err)
  474. }
  475. if net1 != g {
  476. t.Fatalf("NetworkByID() returned unexpected element: %v", g)
  477. }
  478. }
  479. func TestNetworkQuery(t *testing.T) {
  480. defer netutils.SetupTestNetNS(t)()
  481. controller := libnetwork.New()
  482. err := controller.ConfigureNetworkDriver(bridgeNetType, getEmptyGenericOption())
  483. if err != nil {
  484. t.Fatal(err)
  485. }
  486. // Create network 1 and add 2 endpoint: ep11, ep12
  487. net1, err := controller.NewNetwork(bridgeNetType, "network1")
  488. if err != nil {
  489. t.Fatal(err)
  490. }
  491. ep11, err := net1.CreateEndpoint("ep11")
  492. if err != nil {
  493. t.Fatal(err)
  494. }
  495. ep12, err := net1.CreateEndpoint("ep12")
  496. if err != nil {
  497. t.Fatal(err)
  498. }
  499. e, err := net1.EndpointByName("ep11")
  500. if err != nil {
  501. t.Fatal(err)
  502. }
  503. if ep11 != e {
  504. t.Fatalf("EndpointByName() returned %v instead of %v", e, ep11)
  505. }
  506. e, err = net1.EndpointByName("")
  507. if err == nil {
  508. t.Fatalf("EndpointByName() succeeded with invalid target name")
  509. }
  510. if err != libnetwork.ErrInvalidName {
  511. t.Fatalf("EndpointByName() failed with unexpected error: %v", err)
  512. }
  513. e, err = net1.EndpointByName("IamNotAnEndpoint")
  514. if err != nil {
  515. t.Fatal(err)
  516. }
  517. if e != nil {
  518. t.Fatalf("EndpointByName(): expected nil, got %v", e)
  519. }
  520. e, err = net1.EndpointByID(ep12.ID())
  521. if ep12 != e {
  522. t.Fatalf("EndpointByID() returned %v instead of %v", e, ep12)
  523. }
  524. e, err = net1.EndpointByID("")
  525. if err == nil {
  526. t.Fatalf("EndpointByID() succeeded with invalid target id")
  527. }
  528. if err != libnetwork.ErrInvalidID {
  529. t.Fatalf("EndpointByID() failed with unexpected error: %v", err)
  530. }
  531. }
  532. const containerID = "valid_container"
  533. func TestEndpointJoin(t *testing.T) {
  534. defer netutils.SetupTestNetNS(t)()
  535. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  536. if err != nil {
  537. t.Fatal(err)
  538. }
  539. ep, err := n.CreateEndpoint("ep1")
  540. if err != nil {
  541. t.Fatal(err)
  542. }
  543. _, err = ep.Join(containerID,
  544. libnetwork.JoinOptionHostname("test"),
  545. libnetwork.JoinOptionDomainname("docker.io"),
  546. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
  547. if err != nil {
  548. t.Fatal(err)
  549. }
  550. defer func() {
  551. err = ep.Leave(containerID)
  552. if err != nil {
  553. t.Fatal(err)
  554. }
  555. }()
  556. }
  557. func TestEndpointJoinInvalidContainerId(t *testing.T) {
  558. defer netutils.SetupTestNetNS(t)()
  559. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  560. if err != nil {
  561. t.Fatal(err)
  562. }
  563. ep, err := n.CreateEndpoint("ep1")
  564. if err != nil {
  565. t.Fatal(err)
  566. }
  567. _, err = ep.Join("")
  568. if err == nil {
  569. t.Fatal("Expected to fail join with empty container id string")
  570. }
  571. if _, ok := err.(libnetwork.InvalidContainerIDError); !ok {
  572. t.Fatalf("Failed for unexpected reason: %v", err)
  573. }
  574. }
  575. func TestEndpointDeleteWithActiveContainer(t *testing.T) {
  576. defer netutils.SetupTestNetNS(t)()
  577. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  578. if err != nil {
  579. t.Fatal(err)
  580. }
  581. ep, err := n.CreateEndpoint("ep1")
  582. if err != nil {
  583. t.Fatal(err)
  584. }
  585. _, err = ep.Join(containerID,
  586. libnetwork.JoinOptionHostname("test"),
  587. libnetwork.JoinOptionDomainname("docker.io"),
  588. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
  589. if err != nil {
  590. t.Fatal(err)
  591. }
  592. defer func() {
  593. err = ep.Leave(containerID)
  594. if err != nil {
  595. t.Fatal(err)
  596. }
  597. err = ep.Delete()
  598. if err != nil {
  599. t.Fatal(err)
  600. }
  601. }()
  602. err = ep.Delete()
  603. if err == nil {
  604. t.Fatal("Expected to fail. But instead succeeded")
  605. }
  606. if _, ok := err.(*libnetwork.ActiveContainerError); !ok {
  607. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  608. }
  609. }
  610. func TestEndpointMultipleJoins(t *testing.T) {
  611. defer netutils.SetupTestNetNS(t)()
  612. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  613. if err != nil {
  614. t.Fatal(err)
  615. }
  616. ep, err := n.CreateEndpoint("ep1")
  617. if err != nil {
  618. t.Fatal(err)
  619. }
  620. _, err = ep.Join(containerID,
  621. libnetwork.JoinOptionHostname("test"),
  622. libnetwork.JoinOptionDomainname("docker.io"),
  623. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
  624. if err != nil {
  625. t.Fatal(err)
  626. }
  627. defer func() {
  628. err = ep.Leave(containerID)
  629. if err != nil {
  630. t.Fatal(err)
  631. }
  632. }()
  633. _, err = ep.Join("container2")
  634. if err == nil {
  635. t.Fatal("Expected to fail multiple joins for the same endpoint")
  636. }
  637. if err != libnetwork.ErrInvalidJoin {
  638. t.Fatalf("Failed for unexpected reason: %v", err)
  639. }
  640. }
  641. func TestEndpointInvalidLeave(t *testing.T) {
  642. defer netutils.SetupTestNetNS(t)()
  643. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  644. if err != nil {
  645. t.Fatal(err)
  646. }
  647. ep, err := n.CreateEndpoint("ep1")
  648. if err != nil {
  649. t.Fatal(err)
  650. }
  651. err = ep.Leave(containerID)
  652. if err == nil {
  653. t.Fatal("Expected to fail leave from an endpoint which has no active join")
  654. }
  655. if _, ok := err.(libnetwork.InvalidContainerIDError); !ok {
  656. if err != libnetwork.ErrNoContainer {
  657. t.Fatalf("Failed for unexpected reason: %v", err)
  658. }
  659. }
  660. _, err = ep.Join(containerID,
  661. libnetwork.JoinOptionHostname("test"),
  662. libnetwork.JoinOptionDomainname("docker.io"),
  663. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
  664. if err != nil {
  665. t.Fatal(err)
  666. }
  667. defer func() {
  668. err = ep.Leave(containerID)
  669. if err != nil {
  670. t.Fatal(err)
  671. }
  672. }()
  673. err = ep.Leave("")
  674. if err == nil {
  675. t.Fatal("Expected to fail leave with empty container id")
  676. }
  677. if _, ok := err.(libnetwork.InvalidContainerIDError); !ok {
  678. t.Fatalf("Failed for unexpected reason: %v", err)
  679. }
  680. err = ep.Leave("container2")
  681. if err == nil {
  682. t.Fatal("Expected to fail leave with wrong container id")
  683. }
  684. if _, ok := err.(libnetwork.InvalidContainerIDError); !ok {
  685. t.Fatalf("Failed for unexpected reason: %v", err)
  686. }
  687. }
  688. func TestEndpointUpdateParent(t *testing.T) {
  689. defer netutils.SetupTestNetNS(t)()
  690. n, err := createTestNetwork("bridge", "testnetwork", options.Generic{}, options.Generic{})
  691. if err != nil {
  692. t.Fatal(err)
  693. }
  694. ep1, err := n.CreateEndpoint("ep1", nil)
  695. if err != nil {
  696. t.Fatal(err)
  697. }
  698. _, err = ep1.Join(containerID,
  699. libnetwork.JoinOptionHostname("test1"),
  700. libnetwork.JoinOptionDomainname("docker.io"),
  701. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
  702. if err != nil {
  703. t.Fatal(err)
  704. }
  705. defer func() {
  706. err = ep1.Leave(containerID)
  707. if err != nil {
  708. t.Fatal(err)
  709. }
  710. }()
  711. ep2, err := n.CreateEndpoint("ep2", nil)
  712. if err != nil {
  713. t.Fatal(err)
  714. }
  715. _, err = ep2.Join("container2",
  716. libnetwork.JoinOptionHostname("test2"),
  717. libnetwork.JoinOptionDomainname("docker.io"),
  718. libnetwork.JoinOptionHostsPath("/var/lib/docker/test_network/container2/hosts"),
  719. libnetwork.JoinOptionParentUpdate(ep1.ID(), "web", "192.168.0.2"))
  720. if err != nil {
  721. t.Fatal(err)
  722. }
  723. defer func() {
  724. err = ep2.Leave("container2")
  725. if err != nil {
  726. t.Fatal(err)
  727. }
  728. }()
  729. }
  730. func TestEnableIPv6(t *testing.T) {
  731. defer netutils.SetupTestNetNS(t)()
  732. tmpResolvConf := []byte("search pommesfrites.fr\nnameserver 12.34.56.78\nnameserver 2001:4860:4860::8888")
  733. //take a copy of resolv.conf for restoring after test completes
  734. resolvConfSystem, err := ioutil.ReadFile("/etc/resolv.conf")
  735. if err != nil {
  736. t.Fatal(err)
  737. }
  738. //cleanup
  739. defer func() {
  740. if err := ioutil.WriteFile("/etc/resolv.conf", resolvConfSystem, 0644); err != nil {
  741. t.Fatal(err)
  742. }
  743. }()
  744. netOption := options.Generic{
  745. netlabel.EnableIPv6: true,
  746. }
  747. n, err := createTestNetwork("bridge", "testnetwork", options.Generic{}, netOption)
  748. if err != nil {
  749. t.Fatal(err)
  750. }
  751. ep1, err := n.CreateEndpoint("ep1", nil)
  752. if err != nil {
  753. t.Fatal(err)
  754. }
  755. if err := ioutil.WriteFile("/etc/resolv.conf", tmpResolvConf, 0644); err != nil {
  756. t.Fatal(err)
  757. }
  758. resolvConfPath := "/tmp/libnetwork_test/resolv.conf"
  759. _, err = ep1.Join(containerID,
  760. libnetwork.JoinOptionResolvConfPath(resolvConfPath))
  761. if err != nil {
  762. t.Fatal(err)
  763. }
  764. defer func() {
  765. err = ep1.Leave(containerID)
  766. if err != nil {
  767. t.Fatal(err)
  768. }
  769. }()
  770. content, err := ioutil.ReadFile(resolvConfPath)
  771. if err != nil {
  772. t.Fatal(err)
  773. }
  774. if !bytes.Equal(content, tmpResolvConf) {
  775. t.Fatalf("Expected %s, Got %s", string(tmpResolvConf), string(content))
  776. }
  777. if err != nil {
  778. t.Fatal(err)
  779. }
  780. }
  781. func TestNoEnableIPv6(t *testing.T) {
  782. defer netutils.SetupTestNetNS(t)()
  783. tmpResolvConf := []byte("search pommesfrites.fr\nnameserver 12.34.56.78\nnameserver 2001:4860:4860::8888")
  784. expectedResolvConf := []byte("search pommesfrites.fr\nnameserver 12.34.56.78\n")
  785. //take a copy of resolv.conf for restoring after test completes
  786. resolvConfSystem, err := ioutil.ReadFile("/etc/resolv.conf")
  787. if err != nil {
  788. t.Fatal(err)
  789. }
  790. //cleanup
  791. defer func() {
  792. if err := ioutil.WriteFile("/etc/resolv.conf", resolvConfSystem, 0644); err != nil {
  793. t.Fatal(err)
  794. }
  795. }()
  796. n, err := createTestNetwork("bridge", "testnetwork", options.Generic{}, options.Generic{})
  797. if err != nil {
  798. t.Fatal(err)
  799. }
  800. ep1, err := n.CreateEndpoint("ep1", nil)
  801. if err != nil {
  802. t.Fatal(err)
  803. }
  804. if err := ioutil.WriteFile("/etc/resolv.conf", tmpResolvConf, 0644); err != nil {
  805. t.Fatal(err)
  806. }
  807. resolvConfPath := "/tmp/libnetwork_test/resolv.conf"
  808. _, err = ep1.Join(containerID,
  809. libnetwork.JoinOptionResolvConfPath(resolvConfPath))
  810. if err != nil {
  811. t.Fatal(err)
  812. }
  813. defer func() {
  814. err = ep1.Leave(containerID)
  815. if err != nil {
  816. t.Fatal(err)
  817. }
  818. }()
  819. content, err := ioutil.ReadFile(resolvConfPath)
  820. if err != nil {
  821. t.Fatal(err)
  822. }
  823. if !bytes.Equal(content, expectedResolvConf) {
  824. t.Fatalf("Expected %s, Got %s", string(expectedResolvConf), string(content))
  825. }
  826. if err != nil {
  827. t.Fatal(err)
  828. }
  829. }
  830. var (
  831. once sync.Once
  832. ctrlr libnetwork.NetworkController
  833. start = make(chan struct{})
  834. done = make(chan chan struct{}, numThreads-1)
  835. origns = netns.None()
  836. testns = netns.None()
  837. )
  838. const (
  839. iterCnt = 25
  840. numThreads = 3
  841. first = 1
  842. last = numThreads
  843. debug = false
  844. )
  845. func createGlobalInstance(t *testing.T) {
  846. var err error
  847. defer close(start)
  848. origns, err = netns.Get()
  849. if err != nil {
  850. t.Fatal(err)
  851. }
  852. //testns = origns
  853. testns, err = netns.New()
  854. if err != nil {
  855. t.Fatal(err)
  856. }
  857. ctrlr = libnetwork.New()
  858. err = ctrlr.ConfigureNetworkDriver(bridgeNetType, getEmptyGenericOption())
  859. if err != nil {
  860. t.Fatal("configure driver")
  861. }
  862. net, err := ctrlr.NewNetwork(bridgeNetType, "network1")
  863. if err != nil {
  864. t.Fatal("new network")
  865. }
  866. _, err = net.CreateEndpoint("ep1")
  867. if err != nil {
  868. t.Fatal("createendpoint")
  869. }
  870. }
  871. func debugf(format string, a ...interface{}) (int, error) {
  872. if debug {
  873. return fmt.Printf(format, a...)
  874. }
  875. return 0, nil
  876. }
  877. func parallelJoin(t *testing.T, ep libnetwork.Endpoint, thrNumber int) {
  878. debugf("J%d.", thrNumber)
  879. _, err := ep.Join("racing_container")
  880. runtime.LockOSThread()
  881. if err != nil {
  882. if err != libnetwork.ErrNoContainer && err != libnetwork.ErrInvalidJoin {
  883. t.Fatal(err)
  884. }
  885. debugf("JE%d(%v).", thrNumber, err)
  886. }
  887. debugf("JD%d.", thrNumber)
  888. }
  889. func parallelLeave(t *testing.T, ep libnetwork.Endpoint, thrNumber int) {
  890. debugf("L%d.", thrNumber)
  891. err := ep.Leave("racing_container")
  892. runtime.LockOSThread()
  893. if err != nil {
  894. if err != libnetwork.ErrNoContainer && err != libnetwork.ErrInvalidJoin {
  895. t.Fatal(err)
  896. }
  897. debugf("LE%d(%v).", thrNumber, err)
  898. }
  899. debugf("LD%d.", thrNumber)
  900. }
  901. func runParallelTests(t *testing.T, thrNumber int) {
  902. var err error
  903. t.Parallel()
  904. pTest := flag.Lookup("test.parallel")
  905. if pTest == nil {
  906. t.Skip("Skipped because test.parallel flag not set;")
  907. }
  908. numParallel, err := strconv.Atoi(pTest.Value.String())
  909. if err != nil {
  910. t.Fatal(err)
  911. }
  912. if numParallel < numThreads {
  913. t.Skip("Skipped because t.parallel was less than ", numThreads)
  914. }
  915. runtime.LockOSThread()
  916. defer runtime.UnlockOSThread()
  917. if thrNumber == first {
  918. createGlobalInstance(t)
  919. }
  920. if thrNumber != first {
  921. select {
  922. case <-start:
  923. }
  924. thrdone := make(chan struct{})
  925. done <- thrdone
  926. defer close(thrdone)
  927. if thrNumber == last {
  928. defer close(done)
  929. }
  930. err = netns.Set(testns)
  931. if err != nil {
  932. t.Fatal(err)
  933. }
  934. }
  935. defer netns.Set(origns)
  936. net, err := ctrlr.NetworkByName("network1")
  937. if err != nil {
  938. t.Fatal(err)
  939. }
  940. if net == nil {
  941. t.Fatal("Could not find network1")
  942. }
  943. ep, err := net.EndpointByName("ep1")
  944. if err != nil {
  945. t.Fatal(err)
  946. }
  947. if ep == nil {
  948. t.Fatal("Could not find ep1")
  949. }
  950. for i := 0; i < iterCnt; i++ {
  951. parallelJoin(t, ep, thrNumber)
  952. parallelLeave(t, ep, thrNumber)
  953. }
  954. debugf("\n")
  955. if thrNumber == first {
  956. for thrdone := range done {
  957. select {
  958. case <-thrdone:
  959. }
  960. }
  961. testns.Close()
  962. err = ep.Delete()
  963. if err != nil {
  964. t.Fatal(err)
  965. }
  966. }
  967. }
  968. func TestParallel1(t *testing.T) {
  969. runParallelTests(t, 1)
  970. }
  971. func TestParallel2(t *testing.T) {
  972. runParallelTests(t, 2)
  973. }
  974. func TestParallel3(t *testing.T) {
  975. runParallelTests(t, 3)
  976. }