libnetwork_test.go 34 KB

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