libnetwork_test.go 34 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527
  1. package libnetwork_test
  2. import (
  3. "bytes"
  4. "flag"
  5. "fmt"
  6. "io/ioutil"
  7. "net"
  8. "net/http"
  9. "net/http/httptest"
  10. "os"
  11. "runtime"
  12. "strconv"
  13. "sync"
  14. "testing"
  15. log "github.com/Sirupsen/logrus"
  16. "github.com/docker/docker/pkg/plugins"
  17. "github.com/docker/docker/pkg/reexec"
  18. "github.com/docker/libnetwork"
  19. "github.com/docker/libnetwork/driverapi"
  20. "github.com/docker/libnetwork/netlabel"
  21. "github.com/docker/libnetwork/netutils"
  22. "github.com/docker/libnetwork/options"
  23. "github.com/docker/libnetwork/types"
  24. "github.com/vishvananda/netlink"
  25. "github.com/vishvananda/netns"
  26. )
  27. const (
  28. bridgeNetType = "bridge"
  29. bridgeName = "docker0"
  30. )
  31. func TestMain(m *testing.M) {
  32. if reexec.Init() {
  33. return
  34. }
  35. os.Exit(m.Run())
  36. }
  37. func createTestNetwork(networkType, networkName string, option options.Generic, netOption options.Generic) (libnetwork.Network, error) {
  38. controller, err := libnetwork.New()
  39. if err != nil {
  40. return nil, err
  41. }
  42. genericOption := make(map[string]interface{})
  43. genericOption[netlabel.GenericData] = option
  44. err = controller.ConfigureNetworkDriver(networkType, genericOption)
  45. if err != nil {
  46. return nil, err
  47. }
  48. network, err := controller.NewNetwork(networkType, networkName,
  49. libnetwork.NetworkOptionGeneric(netOption))
  50. if err != nil {
  51. return nil, err
  52. }
  53. return network, nil
  54. }
  55. func getEmptyGenericOption() map[string]interface{} {
  56. genericOption := make(map[string]interface{})
  57. genericOption[netlabel.GenericData] = options.Generic{}
  58. return genericOption
  59. }
  60. func getPortMapping() []types.PortBinding {
  61. return []types.PortBinding{
  62. types.PortBinding{Proto: types.TCP, Port: uint16(230), HostPort: uint16(23000)},
  63. types.PortBinding{Proto: types.UDP, Port: uint16(200), HostPort: uint16(22000)},
  64. types.PortBinding{Proto: types.TCP, Port: uint16(120), HostPort: uint16(12000)},
  65. }
  66. }
  67. func TestNull(t *testing.T) {
  68. network, err := createTestNetwork("null", "testnetwork", options.Generic{},
  69. options.Generic{})
  70. if err != nil {
  71. t.Fatal(err)
  72. }
  73. ep, err := network.CreateEndpoint("testep")
  74. if err != nil {
  75. t.Fatal(err)
  76. }
  77. _, err = ep.Join("null_container",
  78. libnetwork.JoinOptionHostname("test"),
  79. libnetwork.JoinOptionDomainname("docker.io"),
  80. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
  81. if err != nil {
  82. t.Fatal(err)
  83. }
  84. err = ep.Leave("null_container")
  85. if err != nil {
  86. t.Fatal(err)
  87. }
  88. if err := ep.Delete(); err != nil {
  89. t.Fatal(err)
  90. }
  91. if err := network.Delete(); err != nil {
  92. t.Fatal(err)
  93. }
  94. }
  95. func TestHost(t *testing.T) {
  96. network, err := createTestNetwork("host", "testnetwork", options.Generic{}, options.Generic{})
  97. if err != nil {
  98. t.Fatal(err)
  99. }
  100. ep1, err := network.CreateEndpoint("testep1")
  101. if err != nil {
  102. t.Fatal(err)
  103. }
  104. _, err = ep1.Join("host_container1",
  105. libnetwork.JoinOptionHostname("test1"),
  106. libnetwork.JoinOptionDomainname("docker.io"),
  107. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"),
  108. libnetwork.JoinOptionUseDefaultSandbox())
  109. if err != nil {
  110. t.Fatal(err)
  111. }
  112. ep2, err := network.CreateEndpoint("testep2")
  113. if err != nil {
  114. t.Fatal(err)
  115. }
  116. _, err = ep2.Join("host_container2",
  117. libnetwork.JoinOptionHostname("test2"),
  118. libnetwork.JoinOptionDomainname("docker.io"),
  119. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"),
  120. libnetwork.JoinOptionUseDefaultSandbox())
  121. if err != nil {
  122. t.Fatal(err)
  123. }
  124. err = ep1.Leave("host_container1")
  125. if err != nil {
  126. t.Fatal(err)
  127. }
  128. err = ep2.Leave("host_container2")
  129. if err != nil {
  130. t.Fatal(err)
  131. }
  132. if err := ep1.Delete(); err != nil {
  133. t.Fatal(err)
  134. }
  135. if err := ep2.Delete(); err != nil {
  136. t.Fatal(err)
  137. }
  138. // Try to create another host endpoint and join/leave that.
  139. ep3, err := network.CreateEndpoint("testep3")
  140. if err != nil {
  141. t.Fatal(err)
  142. }
  143. _, err = ep3.Join("host_container3",
  144. libnetwork.JoinOptionHostname("test3"),
  145. libnetwork.JoinOptionDomainname("docker.io"),
  146. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"),
  147. libnetwork.JoinOptionUseDefaultSandbox())
  148. if err != nil {
  149. t.Fatal(err)
  150. }
  151. err = ep3.Leave("host_container3")
  152. if err != nil {
  153. t.Fatal(err)
  154. }
  155. if err := ep3.Delete(); err != nil {
  156. t.Fatal(err)
  157. }
  158. if err := network.Delete(); err != nil {
  159. t.Fatal(err)
  160. }
  161. }
  162. func TestBridge(t *testing.T) {
  163. if !netutils.IsRunningInContainer() {
  164. defer netutils.SetupTestNetNS(t)()
  165. }
  166. ip, subnet, err := net.ParseCIDR("192.168.100.1/24")
  167. if err != nil {
  168. t.Fatal(err)
  169. }
  170. subnet.IP = ip
  171. ip, cidr, err := net.ParseCIDR("192.168.100.2/28")
  172. if err != nil {
  173. t.Fatal(err)
  174. }
  175. cidr.IP = ip
  176. ip, cidrv6, err := net.ParseCIDR("fe90::1/96")
  177. if err != nil {
  178. t.Fatal(err)
  179. }
  180. cidrv6.IP = ip
  181. log.Debug("Adding a bridge")
  182. option := options.Generic{
  183. "EnableIPForwarding": true,
  184. }
  185. netOption := options.Generic{
  186. "BridgeName": bridgeName,
  187. "AddressIPv4": subnet,
  188. "FixedCIDR": cidr,
  189. "FixedCIDRv6": cidrv6,
  190. "EnableIPv6": true,
  191. "EnableIPTables": true,
  192. "EnableIPMasquerade": true,
  193. "EnableICC": true,
  194. "AllowNonDefaultBridge": true}
  195. network, err := createTestNetwork(bridgeNetType, "testnetwork", option, netOption)
  196. if err != nil {
  197. t.Fatal(err)
  198. }
  199. ep, err := network.CreateEndpoint("testep", libnetwork.CreateOptionPortMapping(getPortMapping()))
  200. if err != nil {
  201. t.Fatal(err)
  202. }
  203. epInfo, err := ep.DriverInfo()
  204. if err != nil {
  205. t.Fatal(err)
  206. }
  207. pmd, ok := epInfo[netlabel.PortMap]
  208. if !ok {
  209. t.Fatalf("Could not find expected info in endpoint data")
  210. }
  211. pm, ok := pmd.([]types.PortBinding)
  212. if !ok {
  213. t.Fatalf("Unexpected format for port mapping in endpoint operational data")
  214. }
  215. if len(pm) != 3 {
  216. t.Fatalf("Incomplete data for port mapping in endpoint operational data: %d", len(pm))
  217. }
  218. if err := ep.Delete(); err != nil {
  219. t.Fatal(err)
  220. }
  221. if err := network.Delete(); err != nil {
  222. t.Fatal(err)
  223. }
  224. }
  225. func TestUnknownDriver(t *testing.T) {
  226. if !netutils.IsRunningInContainer() {
  227. defer netutils.SetupTestNetNS(t)()
  228. }
  229. _, err := createTestNetwork("unknowndriver", "testnetwork", options.Generic{}, options.Generic{})
  230. if err == nil {
  231. t.Fatal("Expected to fail. But instead succeeded")
  232. }
  233. if _, ok := err.(libnetwork.NetworkTypeError); !ok {
  234. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  235. }
  236. }
  237. func TestNilRemoteDriver(t *testing.T) {
  238. controller, err := libnetwork.New()
  239. if err != nil {
  240. t.Fatal(err)
  241. }
  242. _, err = controller.NewNetwork("framerelay", "dummy",
  243. libnetwork.NetworkOptionGeneric(getEmptyGenericOption()))
  244. if err == nil {
  245. t.Fatal("Expected to fail. But instead succeeded")
  246. }
  247. if _, ok := err.(types.NotFoundError); !ok {
  248. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  249. }
  250. }
  251. func TestDuplicateNetwork(t *testing.T) {
  252. if !netutils.IsRunningInContainer() {
  253. defer netutils.SetupTestNetNS(t)()
  254. }
  255. controller, err := libnetwork.New()
  256. if err != nil {
  257. t.Fatal(err)
  258. }
  259. genericOption := make(map[string]interface{})
  260. genericOption[netlabel.GenericData] = options.Generic{}
  261. err = controller.ConfigureNetworkDriver(bridgeNetType, genericOption)
  262. if err != nil {
  263. t.Fatal(err)
  264. }
  265. _, err = controller.NewNetwork(bridgeNetType, "testnetwork", nil)
  266. if err != nil {
  267. t.Fatal(err)
  268. }
  269. _, err = controller.NewNetwork(bridgeNetType, "testnetwork")
  270. if err == nil {
  271. t.Fatal("Expected to fail. But instead succeeded")
  272. }
  273. if _, ok := err.(libnetwork.NetworkNameError); !ok {
  274. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  275. }
  276. }
  277. func TestNetworkName(t *testing.T) {
  278. if !netutils.IsRunningInContainer() {
  279. defer netutils.SetupTestNetNS(t)()
  280. }
  281. _, err := createTestNetwork(bridgeNetType, "", options.Generic{}, options.Generic{})
  282. if err == nil {
  283. t.Fatal("Expected to fail. But instead succeeded")
  284. }
  285. if _, ok := err.(libnetwork.ErrInvalidName); !ok {
  286. t.Fatalf("Expected to fail with ErrInvalidName error. Got %v", err)
  287. }
  288. networkName := "testnetwork"
  289. n, err := createTestNetwork(bridgeNetType, networkName, options.Generic{}, options.Generic{})
  290. if err != nil {
  291. t.Fatal(err)
  292. }
  293. if n.Name() != networkName {
  294. t.Fatalf("Expected network name %s, got %s", networkName, n.Name())
  295. }
  296. }
  297. func TestNetworkType(t *testing.T) {
  298. if !netutils.IsRunningInContainer() {
  299. defer netutils.SetupTestNetNS(t)()
  300. }
  301. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  302. if err != nil {
  303. t.Fatal(err)
  304. }
  305. if n.Type() != bridgeNetType {
  306. t.Fatalf("Expected network type %s, got %s", bridgeNetType, n.Type())
  307. }
  308. }
  309. func TestNetworkID(t *testing.T) {
  310. if !netutils.IsRunningInContainer() {
  311. defer netutils.SetupTestNetNS(t)()
  312. }
  313. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  314. if err != nil {
  315. t.Fatal(err)
  316. }
  317. if n.ID() == "" {
  318. t.Fatal("Expected non-empty network id")
  319. }
  320. }
  321. func TestDeleteNetworkWithActiveEndpoints(t *testing.T) {
  322. if !netutils.IsRunningInContainer() {
  323. defer netutils.SetupTestNetNS(t)()
  324. }
  325. option := options.Generic{
  326. "BridgeName": bridgeName,
  327. "AllowNonDefaultBridge": true}
  328. network, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, option)
  329. if err != nil {
  330. t.Fatal(err)
  331. }
  332. ep, err := network.CreateEndpoint("testep")
  333. if err != nil {
  334. t.Fatal(err)
  335. }
  336. err = network.Delete()
  337. if err == nil {
  338. t.Fatal("Expected to fail. But instead succeeded")
  339. }
  340. if _, ok := err.(*libnetwork.ActiveEndpointsError); !ok {
  341. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  342. }
  343. // Done testing. Now cleanup.
  344. if err := ep.Delete(); err != nil {
  345. t.Fatal(err)
  346. }
  347. if err := network.Delete(); err != nil {
  348. t.Fatal(err)
  349. }
  350. }
  351. func TestUnknownNetwork(t *testing.T) {
  352. if !netutils.IsRunningInContainer() {
  353. defer netutils.SetupTestNetNS(t)()
  354. }
  355. option := options.Generic{
  356. "BridgeName": bridgeName,
  357. "AllowNonDefaultBridge": true}
  358. network, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, option)
  359. if err != nil {
  360. t.Fatal(err)
  361. }
  362. err = network.Delete()
  363. if err != nil {
  364. t.Fatal(err)
  365. }
  366. err = network.Delete()
  367. if err == nil {
  368. t.Fatal("Expected to fail. But instead succeeded")
  369. }
  370. if _, ok := err.(*libnetwork.UnknownNetworkError); !ok {
  371. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  372. }
  373. }
  374. func TestUnknownEndpoint(t *testing.T) {
  375. if !netutils.IsRunningInContainer() {
  376. defer netutils.SetupTestNetNS(t)()
  377. }
  378. ip, subnet, err := net.ParseCIDR("192.168.100.1/24")
  379. if err != nil {
  380. t.Fatal(err)
  381. }
  382. subnet.IP = ip
  383. option := options.Generic{
  384. "BridgeName": bridgeName,
  385. "AddressIPv4": subnet,
  386. "AllowNonDefaultBridge": true}
  387. network, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, option)
  388. if err != nil {
  389. t.Fatal(err)
  390. }
  391. _, err = network.CreateEndpoint("")
  392. if err == nil {
  393. t.Fatal("Expected to fail. But instead succeeded")
  394. }
  395. if _, ok := err.(libnetwork.ErrInvalidName); !ok {
  396. t.Fatalf("Expected to fail with ErrInvalidName error. Actual error: %v", err)
  397. }
  398. ep, err := network.CreateEndpoint("testep")
  399. if err != nil {
  400. t.Fatal(err)
  401. }
  402. err = ep.Delete()
  403. if err != nil {
  404. t.Fatal(err)
  405. }
  406. err = ep.Delete()
  407. if err == nil {
  408. t.Fatal("Expected to fail. But instead succeeded")
  409. }
  410. if _, ok := err.(*libnetwork.UnknownEndpointError); !ok {
  411. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  412. }
  413. // Done testing. Now cleanup
  414. if err := network.Delete(); err != nil {
  415. t.Fatal(err)
  416. }
  417. }
  418. func TestNetworkEndpointsWalkers(t *testing.T) {
  419. if !netutils.IsRunningInContainer() {
  420. defer netutils.SetupTestNetNS(t)()
  421. }
  422. controller, err := libnetwork.New()
  423. if err != nil {
  424. t.Fatal(err)
  425. }
  426. err = controller.ConfigureNetworkDriver(bridgeNetType, getEmptyGenericOption())
  427. if err != nil {
  428. t.Fatal(err)
  429. }
  430. // Create network 1 and add 2 endpoint: ep11, ep12
  431. net1, err := controller.NewNetwork(bridgeNetType, "network1")
  432. if err != nil {
  433. t.Fatal(err)
  434. }
  435. ep11, err := net1.CreateEndpoint("ep11")
  436. if err != nil {
  437. t.Fatal(err)
  438. }
  439. ep12, err := net1.CreateEndpoint("ep12")
  440. if err != nil {
  441. t.Fatal(err)
  442. }
  443. // Test list methods on net1
  444. epList1 := net1.Endpoints()
  445. if len(epList1) != 2 {
  446. t.Fatalf("Endpoints() returned wrong number of elements: %d instead of 2", len(epList1))
  447. }
  448. // endpoint order is not guaranteed
  449. for _, e := range epList1 {
  450. if e != ep11 && e != ep12 {
  451. t.Fatal("Endpoints() did not return all the expected elements")
  452. }
  453. }
  454. // Test Endpoint Walk method
  455. var epName string
  456. var epWanted libnetwork.Endpoint
  457. wlk := func(ep libnetwork.Endpoint) bool {
  458. if ep.Name() == epName {
  459. epWanted = ep
  460. return true
  461. }
  462. return false
  463. }
  464. // Look for ep1 on network1
  465. epName = "ep11"
  466. net1.WalkEndpoints(wlk)
  467. if epWanted == nil {
  468. t.Fatal(err)
  469. }
  470. if ep11 != epWanted {
  471. t.Fatal(err)
  472. }
  473. // Test Network Walk method
  474. var netName string
  475. var netWanted libnetwork.Network
  476. nwWlk := func(nw libnetwork.Network) bool {
  477. if nw.Name() == netName {
  478. netWanted = nw
  479. return true
  480. }
  481. return false
  482. }
  483. // Look for network named "network1"
  484. netName = "network1"
  485. controller.WalkNetworks(nwWlk)
  486. if netWanted == nil {
  487. t.Fatal(err)
  488. }
  489. if net1 != netWanted {
  490. t.Fatal(err)
  491. }
  492. }
  493. func TestControllerQuery(t *testing.T) {
  494. if !netutils.IsRunningInContainer() {
  495. defer netutils.SetupTestNetNS(t)()
  496. }
  497. controller, err := libnetwork.New()
  498. if err != nil {
  499. t.Fatal(err)
  500. }
  501. err = controller.ConfigureNetworkDriver(bridgeNetType, getEmptyGenericOption())
  502. if err != nil {
  503. t.Fatal(err)
  504. }
  505. // Create network 1
  506. net1, err := controller.NewNetwork(bridgeNetType, "network1")
  507. if err != nil {
  508. t.Fatal(err)
  509. }
  510. _, err = controller.NetworkByName("")
  511. if err == nil {
  512. t.Fatalf("NetworkByName() succeeded with invalid target name")
  513. }
  514. if _, ok := err.(libnetwork.ErrInvalidName); !ok {
  515. t.Fatalf("Expected NetworkByName() to fail with ErrInvalidName error. Got: %v", err)
  516. }
  517. _, err = controller.NetworkByID("")
  518. if err == nil {
  519. t.Fatalf("NetworkByID() succeeded with invalid target id")
  520. }
  521. if _, ok := err.(libnetwork.ErrInvalidID); !ok {
  522. t.Fatalf("NetworkByID() failed with unexpected error: %v", err)
  523. }
  524. g, err := controller.NetworkByID("network1")
  525. if err == nil {
  526. t.Fatalf("Unexpected success for NetworkByID(): %v", g)
  527. }
  528. if _, ok := err.(libnetwork.ErrNoSuchNetwork); !ok {
  529. t.Fatalf("NetworkByID() failed with unexpected error: %v", err)
  530. }
  531. g, err = controller.NetworkByName("network1")
  532. if err != nil {
  533. t.Fatalf("Unexpected failure for NetworkByName(): %v", err)
  534. }
  535. if g == nil {
  536. t.Fatalf("NetworkByName() did not find the network")
  537. }
  538. if g != net1 {
  539. t.Fatalf("NetworkByName() returned the wrong network")
  540. }
  541. g, err = controller.NetworkByID(net1.ID())
  542. if err != nil {
  543. t.Fatalf("Unexpected failure for NetworkByID(): %v", err)
  544. }
  545. if net1 != g {
  546. t.Fatalf("NetworkByID() returned unexpected element: %v", g)
  547. }
  548. }
  549. func TestNetworkQuery(t *testing.T) {
  550. if !netutils.IsRunningInContainer() {
  551. defer netutils.SetupTestNetNS(t)()
  552. }
  553. controller, err := libnetwork.New()
  554. if err != nil {
  555. t.Fatal(err)
  556. }
  557. err = controller.ConfigureNetworkDriver(bridgeNetType, getEmptyGenericOption())
  558. if err != nil {
  559. t.Fatal(err)
  560. }
  561. // Create network 1 and add 2 endpoint: ep11, ep12
  562. net1, err := controller.NewNetwork(bridgeNetType, "network1")
  563. if err != nil {
  564. t.Fatal(err)
  565. }
  566. ep11, err := net1.CreateEndpoint("ep11")
  567. if err != nil {
  568. t.Fatal(err)
  569. }
  570. ep12, err := net1.CreateEndpoint("ep12")
  571. if err != nil {
  572. t.Fatal(err)
  573. }
  574. e, err := net1.EndpointByName("ep11")
  575. if err != nil {
  576. t.Fatal(err)
  577. }
  578. if ep11 != e {
  579. t.Fatalf("EndpointByName() returned %v instead of %v", e, ep11)
  580. }
  581. e, err = net1.EndpointByName("")
  582. if err == nil {
  583. t.Fatalf("EndpointByName() succeeded with invalid target name")
  584. }
  585. if _, ok := err.(libnetwork.ErrInvalidName); !ok {
  586. t.Fatalf("Expected EndpointByName() to fail with ErrInvalidName error. Got: %v", err)
  587. }
  588. e, err = net1.EndpointByName("IamNotAnEndpoint")
  589. if err == nil {
  590. t.Fatalf("EndpointByName() succeeded with unknown target name")
  591. }
  592. if _, ok := err.(libnetwork.ErrNoSuchEndpoint); !ok {
  593. t.Fatal(err)
  594. }
  595. if e != nil {
  596. t.Fatalf("EndpointByName(): expected nil, got %v", e)
  597. }
  598. e, err = net1.EndpointByID(ep12.ID())
  599. if err != nil {
  600. t.Fatal(err)
  601. }
  602. if ep12 != e {
  603. t.Fatalf("EndpointByID() returned %v instead of %v", e, ep12)
  604. }
  605. e, err = net1.EndpointByID("")
  606. if err == nil {
  607. t.Fatalf("EndpointByID() succeeded with invalid target id")
  608. }
  609. if _, ok := err.(libnetwork.ErrInvalidID); !ok {
  610. t.Fatalf("EndpointByID() failed with unexpected error: %v", err)
  611. }
  612. }
  613. const containerID = "valid_container"
  614. func checkSandbox(t *testing.T, info libnetwork.EndpointInfo) {
  615. origns, err := netns.Get()
  616. if err != nil {
  617. t.Fatalf("Could not get the current netns: %v", err)
  618. }
  619. defer origns.Close()
  620. key := info.SandboxKey()
  621. f, err := os.OpenFile(key, os.O_RDONLY, 0)
  622. if err != nil {
  623. t.Fatalf("Failed to open network namespace path %q: %v", key, err)
  624. }
  625. defer f.Close()
  626. runtime.LockOSThread()
  627. defer runtime.UnlockOSThread()
  628. nsFD := f.Fd()
  629. if err = netns.Set(netns.NsHandle(nsFD)); err != nil {
  630. t.Fatalf("Setting to the namespace pointed to by the sandbox %s failed: %v", key, err)
  631. }
  632. defer netns.Set(origns)
  633. _, err = netlink.LinkByName("eth0")
  634. if err != nil {
  635. t.Fatalf("Could not find the interface eth0 inside the sandbox: %v", err)
  636. }
  637. }
  638. func TestEndpointJoin(t *testing.T) {
  639. if !netutils.IsRunningInContainer() {
  640. defer netutils.SetupTestNetNS(t)()
  641. }
  642. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  643. if err != nil {
  644. t.Fatal(err)
  645. }
  646. ep, err := n.CreateEndpoint("ep1")
  647. if err != nil {
  648. t.Fatal(err)
  649. }
  650. // Validate if ep.Info() only gives me IP address info and not names and gateway during CreateEndpoint()
  651. info := ep.Info()
  652. for _, iface := range info.InterfaceList() {
  653. if iface.Address().IP.To4() == nil {
  654. t.Fatalf("Invalid IP address returned: %v", iface.Address())
  655. }
  656. }
  657. if info.Gateway().To4() != nil {
  658. t.Fatalf("Expected empty gateway for an empty endpoint. Instead found a gateway: %v", info.Gateway())
  659. }
  660. if info.SandboxKey() != "" {
  661. t.Fatalf("Expected an empty sandbox key for an empty endpoint. Instead found a non-empty sandbox key: %s", info.SandboxKey())
  662. }
  663. _, err = ep.Join(containerID,
  664. libnetwork.JoinOptionHostname("test"),
  665. libnetwork.JoinOptionDomainname("docker.io"),
  666. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
  667. if err != nil {
  668. t.Fatal(err)
  669. }
  670. defer func() {
  671. err = ep.Leave(containerID)
  672. if err != nil {
  673. t.Fatal(err)
  674. }
  675. }()
  676. // Validate if ep.Info() only gives valid gateway and sandbox key after has container has joined.
  677. info = ep.Info()
  678. if info.Gateway().To4() == nil {
  679. t.Fatalf("Expected a valid gateway for a joined endpoint. Instead found an invalid gateway: %v", info.Gateway())
  680. }
  681. if info.SandboxKey() == "" {
  682. t.Fatalf("Expected an non-empty sandbox key for a joined endpoint. Instead found a empty sandbox key")
  683. }
  684. checkSandbox(t, info)
  685. }
  686. func TestEndpointJoinInvalidContainerId(t *testing.T) {
  687. if !netutils.IsRunningInContainer() {
  688. defer netutils.SetupTestNetNS(t)()
  689. }
  690. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  691. if err != nil {
  692. t.Fatal(err)
  693. }
  694. ep, err := n.CreateEndpoint("ep1")
  695. if err != nil {
  696. t.Fatal(err)
  697. }
  698. _, err = ep.Join("")
  699. if err == nil {
  700. t.Fatal("Expected to fail join with empty container id string")
  701. }
  702. if _, ok := err.(libnetwork.InvalidContainerIDError); !ok {
  703. t.Fatalf("Failed for unexpected reason: %v", err)
  704. }
  705. }
  706. func TestEndpointDeleteWithActiveContainer(t *testing.T) {
  707. if !netutils.IsRunningInContainer() {
  708. defer netutils.SetupTestNetNS(t)()
  709. }
  710. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  711. if err != nil {
  712. t.Fatal(err)
  713. }
  714. ep, err := n.CreateEndpoint("ep1")
  715. if err != nil {
  716. t.Fatal(err)
  717. }
  718. _, err = ep.Join(containerID,
  719. libnetwork.JoinOptionHostname("test"),
  720. libnetwork.JoinOptionDomainname("docker.io"),
  721. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
  722. if err != nil {
  723. t.Fatal(err)
  724. }
  725. defer func() {
  726. err = ep.Leave(containerID)
  727. if err != nil {
  728. t.Fatal(err)
  729. }
  730. err = ep.Delete()
  731. if err != nil {
  732. t.Fatal(err)
  733. }
  734. }()
  735. err = ep.Delete()
  736. if err == nil {
  737. t.Fatal("Expected to fail. But instead succeeded")
  738. }
  739. if _, ok := err.(*libnetwork.ActiveContainerError); !ok {
  740. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  741. }
  742. }
  743. func TestEndpointMultipleJoins(t *testing.T) {
  744. if !netutils.IsRunningInContainer() {
  745. defer netutils.SetupTestNetNS(t)()
  746. }
  747. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  748. if err != nil {
  749. t.Fatal(err)
  750. }
  751. ep, err := n.CreateEndpoint("ep1")
  752. if err != nil {
  753. t.Fatal(err)
  754. }
  755. _, err = ep.Join(containerID,
  756. libnetwork.JoinOptionHostname("test"),
  757. libnetwork.JoinOptionDomainname("docker.io"),
  758. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
  759. if err != nil {
  760. t.Fatal(err)
  761. }
  762. defer func() {
  763. err = ep.Leave(containerID)
  764. if err != nil {
  765. t.Fatal(err)
  766. }
  767. }()
  768. _, err = ep.Join("container2")
  769. if err == nil {
  770. t.Fatal("Expected to fail multiple joins for the same endpoint")
  771. }
  772. if _, ok := err.(libnetwork.ErrInvalidJoin); !ok {
  773. t.Fatalf("Failed for unexpected reason: %v", err)
  774. }
  775. }
  776. func TestEndpointInvalidLeave(t *testing.T) {
  777. if !netutils.IsRunningInContainer() {
  778. defer netutils.SetupTestNetNS(t)()
  779. }
  780. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  781. if err != nil {
  782. t.Fatal(err)
  783. }
  784. ep, err := n.CreateEndpoint("ep1")
  785. if err != nil {
  786. t.Fatal(err)
  787. }
  788. err = ep.Leave(containerID)
  789. if err == nil {
  790. t.Fatal("Expected to fail leave from an endpoint which has no active join")
  791. }
  792. if _, ok := err.(libnetwork.InvalidContainerIDError); !ok {
  793. if _, ok := err.(libnetwork.ErrNoContainer); !ok {
  794. t.Fatalf("Failed for unexpected reason: %v", err)
  795. }
  796. }
  797. _, err = ep.Join(containerID,
  798. libnetwork.JoinOptionHostname("test"),
  799. libnetwork.JoinOptionDomainname("docker.io"),
  800. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
  801. if err != nil {
  802. t.Fatal(err)
  803. }
  804. defer func() {
  805. err = ep.Leave(containerID)
  806. if err != nil {
  807. t.Fatal(err)
  808. }
  809. }()
  810. err = ep.Leave("")
  811. if err == nil {
  812. t.Fatal("Expected to fail leave with empty container id")
  813. }
  814. if _, ok := err.(libnetwork.InvalidContainerIDError); !ok {
  815. t.Fatalf("Failed for unexpected reason: %v", err)
  816. }
  817. err = ep.Leave("container2")
  818. if err == nil {
  819. t.Fatal("Expected to fail leave with wrong container id")
  820. }
  821. if _, ok := err.(libnetwork.InvalidContainerIDError); !ok {
  822. t.Fatalf("Failed for unexpected reason: %v", err)
  823. }
  824. }
  825. func TestEndpointUpdateParent(t *testing.T) {
  826. if !netutils.IsRunningInContainer() {
  827. defer netutils.SetupTestNetNS(t)()
  828. }
  829. n, err := createTestNetwork("bridge", "testnetwork", options.Generic{}, options.Generic{})
  830. if err != nil {
  831. t.Fatal(err)
  832. }
  833. ep1, err := n.CreateEndpoint("ep1", nil)
  834. if err != nil {
  835. t.Fatal(err)
  836. }
  837. _, err = ep1.Join(containerID,
  838. libnetwork.JoinOptionHostname("test1"),
  839. libnetwork.JoinOptionDomainname("docker.io"),
  840. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
  841. if err != nil {
  842. t.Fatal(err)
  843. }
  844. defer func() {
  845. err = ep1.Leave(containerID)
  846. if err != nil {
  847. t.Fatal(err)
  848. }
  849. }()
  850. ep2, err := n.CreateEndpoint("ep2", nil)
  851. if err != nil {
  852. t.Fatal(err)
  853. }
  854. _, err = ep2.Join("container2",
  855. libnetwork.JoinOptionHostname("test2"),
  856. libnetwork.JoinOptionDomainname("docker.io"),
  857. libnetwork.JoinOptionHostsPath("/var/lib/docker/test_network/container2/hosts"),
  858. libnetwork.JoinOptionParentUpdate(ep1.ID(), "web", "192.168.0.2"))
  859. if err != nil {
  860. t.Fatal(err)
  861. }
  862. defer func() {
  863. err = ep2.Leave("container2")
  864. if err != nil {
  865. t.Fatal(err)
  866. }
  867. }()
  868. }
  869. func TestEnableIPv6(t *testing.T) {
  870. if !netutils.IsRunningInContainer() {
  871. defer netutils.SetupTestNetNS(t)()
  872. }
  873. tmpResolvConf := []byte("search pommesfrites.fr\nnameserver 12.34.56.78\nnameserver 2001:4860:4860::8888")
  874. //take a copy of resolv.conf for restoring after test completes
  875. resolvConfSystem, err := ioutil.ReadFile("/etc/resolv.conf")
  876. if err != nil {
  877. t.Fatal(err)
  878. }
  879. //cleanup
  880. defer func() {
  881. if err := ioutil.WriteFile("/etc/resolv.conf", resolvConfSystem, 0644); err != nil {
  882. t.Fatal(err)
  883. }
  884. }()
  885. ip, cidrv6, err := net.ParseCIDR("fe80::1/64")
  886. if err != nil {
  887. t.Fatal(err)
  888. }
  889. cidrv6.IP = ip
  890. netOption := options.Generic{
  891. netlabel.EnableIPv6: true,
  892. netlabel.GenericData: options.Generic{
  893. "FixedCIDRv6": cidrv6,
  894. },
  895. }
  896. n, err := createTestNetwork("bridge", "testnetwork", options.Generic{}, netOption)
  897. if err != nil {
  898. t.Fatal(err)
  899. }
  900. ep1, err := n.CreateEndpoint("ep1", nil)
  901. if err != nil {
  902. t.Fatal(err)
  903. }
  904. if err := ioutil.WriteFile("/etc/resolv.conf", tmpResolvConf, 0644); err != nil {
  905. t.Fatal(err)
  906. }
  907. resolvConfPath := "/tmp/libnetwork_test/resolv.conf"
  908. defer os.Remove(resolvConfPath)
  909. _, err = ep1.Join(containerID,
  910. libnetwork.JoinOptionResolvConfPath(resolvConfPath))
  911. if err != nil {
  912. t.Fatal(err)
  913. }
  914. defer func() {
  915. err = ep1.Leave(containerID)
  916. if err != nil {
  917. t.Fatal(err)
  918. }
  919. }()
  920. content, err := ioutil.ReadFile(resolvConfPath)
  921. if err != nil {
  922. t.Fatal(err)
  923. }
  924. if !bytes.Equal(content, tmpResolvConf) {
  925. t.Fatalf("Expected %s, Got %s", string(tmpResolvConf), string(content))
  926. }
  927. if err != nil {
  928. t.Fatal(err)
  929. }
  930. }
  931. func TestResolvConf(t *testing.T) {
  932. if !netutils.IsRunningInContainer() {
  933. defer netutils.SetupTestNetNS(t)()
  934. }
  935. tmpResolvConf1 := []byte("search pommesfrites.fr\nnameserver 12.34.56.78\nnameserver 2001:4860:4860::8888")
  936. expectedResolvConf1 := []byte("search pommesfrites.fr\nnameserver 12.34.56.78\n")
  937. tmpResolvConf2 := []byte("search pommesfrites.fr\nnameserver 112.34.56.78\nnameserver 2001:4860:4860::8888")
  938. expectedResolvConf2 := []byte("search pommesfrites.fr\nnameserver 112.34.56.78\n")
  939. tmpResolvConf3 := []byte("search pommesfrites.fr\nnameserver 113.34.56.78\n")
  940. //take a copy of resolv.conf for restoring after test completes
  941. resolvConfSystem, err := ioutil.ReadFile("/etc/resolv.conf")
  942. if err != nil {
  943. t.Fatal(err)
  944. }
  945. //cleanup
  946. defer func() {
  947. if err := ioutil.WriteFile("/etc/resolv.conf", resolvConfSystem, 0644); err != nil {
  948. t.Fatal(err)
  949. }
  950. }()
  951. n, err := createTestNetwork("bridge", "testnetwork", options.Generic{}, options.Generic{})
  952. if err != nil {
  953. t.Fatal(err)
  954. }
  955. ep1, err := n.CreateEndpoint("ep1", nil)
  956. if err != nil {
  957. t.Fatal(err)
  958. }
  959. if err := ioutil.WriteFile("/etc/resolv.conf", tmpResolvConf1, 0644); err != nil {
  960. t.Fatal(err)
  961. }
  962. resolvConfPath := "/tmp/libnetwork_test/resolv.conf"
  963. defer os.Remove(resolvConfPath)
  964. _, err = ep1.Join(containerID,
  965. libnetwork.JoinOptionResolvConfPath(resolvConfPath))
  966. if err != nil {
  967. t.Fatal(err)
  968. }
  969. defer func() {
  970. err = ep1.Leave(containerID)
  971. if err != nil {
  972. t.Fatal(err)
  973. }
  974. }()
  975. finfo, err := os.Stat(resolvConfPath)
  976. if err != nil {
  977. t.Fatal(err)
  978. }
  979. fmode := (os.FileMode)(0644)
  980. if finfo.Mode() != fmode {
  981. t.Fatalf("Expected file mode %s, got %s", fmode.String(), finfo.Mode().String())
  982. }
  983. content, err := ioutil.ReadFile(resolvConfPath)
  984. if err != nil {
  985. t.Fatal(err)
  986. }
  987. if !bytes.Equal(content, expectedResolvConf1) {
  988. t.Fatalf("Expected %s, Got %s", string(expectedResolvConf1), string(content))
  989. }
  990. err = ep1.Leave(containerID)
  991. if err != nil {
  992. t.Fatal(err)
  993. }
  994. if err := ioutil.WriteFile("/etc/resolv.conf", tmpResolvConf2, 0644); err != nil {
  995. t.Fatal(err)
  996. }
  997. _, err = ep1.Join(containerID,
  998. libnetwork.JoinOptionResolvConfPath(resolvConfPath))
  999. if err != nil {
  1000. t.Fatal(err)
  1001. }
  1002. content, err = ioutil.ReadFile(resolvConfPath)
  1003. if err != nil {
  1004. t.Fatal(err)
  1005. }
  1006. if !bytes.Equal(content, expectedResolvConf2) {
  1007. t.Fatalf("Expected %s, Got %s", string(expectedResolvConf2), string(content))
  1008. }
  1009. if err := ioutil.WriteFile(resolvConfPath, tmpResolvConf3, 0644); err != nil {
  1010. t.Fatal(err)
  1011. }
  1012. err = ep1.Leave(containerID)
  1013. if err != nil {
  1014. t.Fatal(err)
  1015. }
  1016. _, err = ep1.Join(containerID,
  1017. libnetwork.JoinOptionResolvConfPath(resolvConfPath))
  1018. if err != nil {
  1019. t.Fatal(err)
  1020. }
  1021. content, err = ioutil.ReadFile(resolvConfPath)
  1022. if err != nil {
  1023. t.Fatal(err)
  1024. }
  1025. if !bytes.Equal(content, tmpResolvConf3) {
  1026. t.Fatalf("Expected %s, Got %s", string(tmpResolvConf3), string(content))
  1027. }
  1028. }
  1029. func TestInvalidRemoteDriver(t *testing.T) {
  1030. if !netutils.IsRunningInContainer() {
  1031. t.Skip("Skipping test when not running inside a Container")
  1032. }
  1033. mux := http.NewServeMux()
  1034. server := httptest.NewServer(mux)
  1035. if server == nil {
  1036. t.Fatal("Failed to start a HTTP Server")
  1037. }
  1038. defer server.Close()
  1039. type pluginRequest struct {
  1040. name string
  1041. }
  1042. mux.HandleFunc("/Plugin.Activate", func(w http.ResponseWriter, r *http.Request) {
  1043. w.Header().Set("Content-Type", "application/vnd.docker.plugins.v1+json")
  1044. fmt.Fprintln(w, `{"Implements": ["InvalidDriver"]}`)
  1045. })
  1046. if err := os.MkdirAll("/usr/share/docker/plugins", 0755); err != nil {
  1047. t.Fatal(err)
  1048. }
  1049. defer func() {
  1050. if err := os.RemoveAll("/usr/share/docker/plugins"); err != nil {
  1051. t.Fatal(err)
  1052. }
  1053. }()
  1054. if err := ioutil.WriteFile("/usr/share/docker/plugins/invalid-network-driver.spec", []byte(server.URL), 0644); err != nil {
  1055. t.Fatal(err)
  1056. }
  1057. controller, err := libnetwork.New()
  1058. if err != nil {
  1059. t.Fatal(err)
  1060. }
  1061. _, err = controller.NewNetwork("invalid-network-driver", "dummy",
  1062. libnetwork.NetworkOptionGeneric(getEmptyGenericOption()))
  1063. if err == nil {
  1064. t.Fatal("Expected to fail. But instead succeeded")
  1065. }
  1066. if err != plugins.ErrNotImplements {
  1067. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  1068. }
  1069. }
  1070. func TestValidRemoteDriver(t *testing.T) {
  1071. if !netutils.IsRunningInContainer() {
  1072. t.Skip("Skipping test when not running inside a Container")
  1073. }
  1074. mux := http.NewServeMux()
  1075. server := httptest.NewServer(mux)
  1076. if server == nil {
  1077. t.Fatal("Failed to start a HTTP Server")
  1078. }
  1079. defer server.Close()
  1080. type pluginRequest struct {
  1081. name string
  1082. }
  1083. mux.HandleFunc("/Plugin.Activate", func(w http.ResponseWriter, r *http.Request) {
  1084. w.Header().Set("Content-Type", "application/vnd.docker.plugins.v1+json")
  1085. fmt.Fprintf(w, `{"Implements": ["%s"]}`, driverapi.NetworkPluginEndpointType)
  1086. })
  1087. mux.HandleFunc(fmt.Sprintf("/%s.CreateNetwork", driverapi.NetworkPluginEndpointType), func(w http.ResponseWriter, r *http.Request) {
  1088. w.Header().Set("Content-Type", "application/vnd.docker.plugins.v1+json")
  1089. fmt.Fprintf(w, "null")
  1090. })
  1091. if err := os.MkdirAll("/usr/share/docker/plugins", 0755); err != nil {
  1092. t.Fatal(err)
  1093. }
  1094. defer func() {
  1095. if err := os.RemoveAll("/usr/share/docker/plugins"); err != nil {
  1096. t.Fatal(err)
  1097. }
  1098. }()
  1099. if err := ioutil.WriteFile("/usr/share/docker/plugins/valid-network-driver.spec", []byte(server.URL), 0644); err != nil {
  1100. t.Fatal(err)
  1101. }
  1102. controller, err := libnetwork.New()
  1103. if err != nil {
  1104. t.Fatal(err)
  1105. }
  1106. _, err = controller.NewNetwork("valid-network-driver", "dummy",
  1107. libnetwork.NetworkOptionGeneric(getEmptyGenericOption()))
  1108. if err != nil {
  1109. t.Fatal(err)
  1110. }
  1111. }
  1112. var (
  1113. once sync.Once
  1114. ctrlr libnetwork.NetworkController
  1115. start = make(chan struct{})
  1116. done = make(chan chan struct{}, numThreads-1)
  1117. origns = netns.None()
  1118. testns = netns.None()
  1119. )
  1120. const (
  1121. iterCnt = 25
  1122. numThreads = 3
  1123. first = 1
  1124. last = numThreads
  1125. debug = false
  1126. )
  1127. func createGlobalInstance(t *testing.T) {
  1128. var err error
  1129. defer close(start)
  1130. origns, err = netns.Get()
  1131. if err != nil {
  1132. t.Fatal(err)
  1133. }
  1134. if netutils.IsRunningInContainer() {
  1135. testns = origns
  1136. } else {
  1137. testns, err = netns.New()
  1138. if err != nil {
  1139. t.Fatal(err)
  1140. }
  1141. }
  1142. ctrlr, err = libnetwork.New()
  1143. if err != nil {
  1144. t.Fatal(err)
  1145. }
  1146. err = ctrlr.ConfigureNetworkDriver(bridgeNetType, getEmptyGenericOption())
  1147. if err != nil {
  1148. t.Fatal("configure driver")
  1149. }
  1150. net, err := ctrlr.NewNetwork(bridgeNetType, "network1")
  1151. if err != nil {
  1152. t.Fatal("new network")
  1153. }
  1154. _, err = net.CreateEndpoint("ep1")
  1155. if err != nil {
  1156. t.Fatal("createendpoint")
  1157. }
  1158. }
  1159. func debugf(format string, a ...interface{}) (int, error) {
  1160. if debug {
  1161. return fmt.Printf(format, a...)
  1162. }
  1163. return 0, nil
  1164. }
  1165. func parallelJoin(t *testing.T, ep libnetwork.Endpoint, thrNumber int) {
  1166. debugf("J%d.", thrNumber)
  1167. _, err := ep.Join("racing_container")
  1168. runtime.LockOSThread()
  1169. if err != nil {
  1170. if _, ok := err.(libnetwork.ErrNoContainer); !ok {
  1171. if _, ok := err.(libnetwork.ErrInvalidJoin); !ok {
  1172. t.Fatal(err)
  1173. }
  1174. }
  1175. debugf("JE%d(%v).", thrNumber, err)
  1176. }
  1177. debugf("JD%d.", thrNumber)
  1178. }
  1179. func parallelLeave(t *testing.T, ep libnetwork.Endpoint, thrNumber int) {
  1180. debugf("L%d.", thrNumber)
  1181. err := ep.Leave("racing_container")
  1182. runtime.LockOSThread()
  1183. if err != nil {
  1184. if _, ok := err.(libnetwork.ErrNoContainer); !ok {
  1185. if _, ok := err.(libnetwork.ErrInvalidJoin); !ok {
  1186. t.Fatal(err)
  1187. }
  1188. }
  1189. debugf("LE%d(%v).", thrNumber, err)
  1190. }
  1191. debugf("LD%d.", thrNumber)
  1192. }
  1193. func runParallelTests(t *testing.T, thrNumber int) {
  1194. var err error
  1195. t.Parallel()
  1196. pTest := flag.Lookup("test.parallel")
  1197. if pTest == nil {
  1198. t.Skip("Skipped because test.parallel flag not set;")
  1199. }
  1200. numParallel, err := strconv.Atoi(pTest.Value.String())
  1201. if err != nil {
  1202. t.Fatal(err)
  1203. }
  1204. if numParallel < numThreads {
  1205. t.Skip("Skipped because t.parallel was less than ", numThreads)
  1206. }
  1207. runtime.LockOSThread()
  1208. defer runtime.UnlockOSThread()
  1209. if thrNumber == first {
  1210. createGlobalInstance(t)
  1211. }
  1212. if thrNumber != first {
  1213. select {
  1214. case <-start:
  1215. }
  1216. thrdone := make(chan struct{})
  1217. done <- thrdone
  1218. defer close(thrdone)
  1219. if thrNumber == last {
  1220. defer close(done)
  1221. }
  1222. err = netns.Set(testns)
  1223. if err != nil {
  1224. t.Fatal(err)
  1225. }
  1226. }
  1227. defer netns.Set(origns)
  1228. net, err := ctrlr.NetworkByName("network1")
  1229. if err != nil {
  1230. t.Fatal(err)
  1231. }
  1232. if net == nil {
  1233. t.Fatal("Could not find network1")
  1234. }
  1235. ep, err := net.EndpointByName("ep1")
  1236. if err != nil {
  1237. t.Fatal(err)
  1238. }
  1239. if ep == nil {
  1240. t.Fatal("Got nil ep with no error")
  1241. }
  1242. for i := 0; i < iterCnt; i++ {
  1243. parallelJoin(t, ep, thrNumber)
  1244. parallelLeave(t, ep, thrNumber)
  1245. }
  1246. debugf("\n")
  1247. if thrNumber == first {
  1248. for thrdone := range done {
  1249. select {
  1250. case <-thrdone:
  1251. }
  1252. }
  1253. testns.Close()
  1254. err = ep.Delete()
  1255. if err != nil {
  1256. t.Fatal(err)
  1257. }
  1258. }
  1259. }
  1260. func TestParallel1(t *testing.T) {
  1261. runParallelTests(t, 1)
  1262. }
  1263. func TestParallel2(t *testing.T) {
  1264. runParallelTests(t, 2)
  1265. }
  1266. func TestParallel3(t *testing.T) {
  1267. runParallelTests(t, 3)
  1268. }