libnetwork_test.go 26 KB

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