docker_api_network_test.go 8.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287
  1. package main
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "net"
  6. "net/http"
  7. "net/url"
  8. "strings"
  9. "github.com/docker/docker/api/types"
  10. "github.com/docker/docker/daemon/network"
  11. "github.com/docker/docker/pkg/parsers/filters"
  12. "github.com/go-check/check"
  13. )
  14. func (s *DockerSuite) TestApiNetworkGetDefaults(c *check.C) {
  15. // By default docker daemon creates 3 networks. check if they are present
  16. defaults := []string{"bridge", "host", "none"}
  17. for _, nn := range defaults {
  18. c.Assert(isNetworkAvailable(c, nn), check.Equals, true)
  19. }
  20. }
  21. func (s *DockerSuite) TestApiNetworkCreateDelete(c *check.C) {
  22. // Create a network
  23. name := "testnetwork"
  24. config := types.NetworkCreate{
  25. Name: name,
  26. CheckDuplicate: true,
  27. }
  28. id := createNetwork(c, config, true)
  29. c.Assert(isNetworkAvailable(c, name), check.Equals, true)
  30. // POST another network with same name and CheckDuplicate must fail
  31. createNetwork(c, config, false)
  32. // delete the network and make sure it is deleted
  33. deleteNetwork(c, id, true)
  34. c.Assert(isNetworkAvailable(c, name), check.Equals, false)
  35. }
  36. func (s *DockerSuite) TestApiNetworkFilter(c *check.C) {
  37. nr := getNetworkResource(c, getNetworkIDByName(c, "bridge"))
  38. c.Assert(nr.Name, check.Equals, "bridge")
  39. }
  40. func (s *DockerSuite) TestApiNetworkInspect(c *check.C) {
  41. // Inspect default bridge network
  42. nr := getNetworkResource(c, "bridge")
  43. c.Assert(nr.Name, check.Equals, "bridge")
  44. // run a container and attach it to the default bridge network
  45. out, _ := dockerCmd(c, "run", "-d", "--name", "test", "busybox", "top")
  46. containerID := strings.TrimSpace(out)
  47. containerIP := findContainerIP(c, "test")
  48. // inspect default bridge network again and make sure the container is connected
  49. nr = getNetworkResource(c, nr.ID)
  50. c.Assert(nr.Driver, check.Equals, "bridge")
  51. c.Assert(nr.Scope, check.Equals, "local")
  52. c.Assert(nr.IPAM.Driver, check.Equals, "default")
  53. c.Assert(len(nr.Containers), check.Equals, 1)
  54. c.Assert(nr.Containers[containerID], check.NotNil)
  55. ip, _, err := net.ParseCIDR(nr.Containers[containerID].IPv4Address)
  56. c.Assert(err, check.IsNil)
  57. c.Assert(ip.String(), check.Equals, containerIP)
  58. // IPAM configuration inspect
  59. ipam := network.IPAM{
  60. Driver: "default",
  61. Config: []network.IPAMConfig{{Subnet: "172.28.0.0/16", IPRange: "172.28.5.0/24", Gateway: "172.28.5.254"}},
  62. }
  63. config := types.NetworkCreate{
  64. Name: "br0",
  65. Driver: "bridge",
  66. IPAM: ipam,
  67. }
  68. id0 := createNetwork(c, config, true)
  69. c.Assert(isNetworkAvailable(c, "br0"), check.Equals, true)
  70. nr = getNetworkResource(c, id0)
  71. c.Assert(len(nr.IPAM.Config), check.Equals, 1)
  72. c.Assert(nr.IPAM.Config[0].Subnet, check.Equals, "172.28.0.0/16")
  73. c.Assert(nr.IPAM.Config[0].IPRange, check.Equals, "172.28.5.0/24")
  74. c.Assert(nr.IPAM.Config[0].Gateway, check.Equals, "172.28.5.254")
  75. // delete the network and make sure it is deleted
  76. deleteNetwork(c, id0, true)
  77. c.Assert(isNetworkAvailable(c, "br0"), check.Equals, false)
  78. }
  79. func (s *DockerSuite) TestApiNetworkConnectDisconnect(c *check.C) {
  80. // Create test network
  81. name := "testnetwork"
  82. config := types.NetworkCreate{
  83. Name: name,
  84. }
  85. id := createNetwork(c, config, true)
  86. nr := getNetworkResource(c, id)
  87. c.Assert(nr.Name, check.Equals, name)
  88. c.Assert(nr.ID, check.Equals, id)
  89. c.Assert(len(nr.Containers), check.Equals, 0)
  90. // run a container
  91. out, _ := dockerCmd(c, "run", "-d", "--name", "test", "busybox", "top")
  92. containerID := strings.TrimSpace(out)
  93. // connect the container to the test network
  94. connectNetwork(c, nr.ID, containerID)
  95. // inspect the network to make sure container is connected
  96. nr = getNetworkResource(c, nr.ID)
  97. c.Assert(len(nr.Containers), check.Equals, 1)
  98. c.Assert(nr.Containers[containerID], check.NotNil)
  99. // check if container IP matches network inspect
  100. ip, _, err := net.ParseCIDR(nr.Containers[containerID].IPv4Address)
  101. c.Assert(err, check.IsNil)
  102. containerIP := findContainerIP(c, "test")
  103. c.Assert(ip.String(), check.Equals, containerIP)
  104. // disconnect container from the network
  105. disconnectNetwork(c, nr.ID, containerID)
  106. nr = getNetworkResource(c, nr.ID)
  107. c.Assert(nr.Name, check.Equals, name)
  108. c.Assert(len(nr.Containers), check.Equals, 0)
  109. // delete the network
  110. deleteNetwork(c, nr.ID, true)
  111. }
  112. func (s *DockerSuite) TestApiNetworkIpamMultipleBridgeNetworks(c *check.C) {
  113. // test0 bridge network
  114. ipam0 := network.IPAM{
  115. Driver: "default",
  116. Config: []network.IPAMConfig{{Subnet: "192.178.0.0/16", IPRange: "192.178.128.0/17", Gateway: "192.178.138.100"}},
  117. }
  118. config0 := types.NetworkCreate{
  119. Name: "test0",
  120. Driver: "bridge",
  121. IPAM: ipam0,
  122. }
  123. id0 := createNetwork(c, config0, true)
  124. c.Assert(isNetworkAvailable(c, "test0"), check.Equals, true)
  125. ipam1 := network.IPAM{
  126. Driver: "default",
  127. Config: []network.IPAMConfig{{Subnet: "192.178.128.0/17", Gateway: "192.178.128.1"}},
  128. }
  129. // test1 bridge network overlaps with test0
  130. config1 := types.NetworkCreate{
  131. Name: "test1",
  132. Driver: "bridge",
  133. IPAM: ipam1,
  134. }
  135. createNetwork(c, config1, false)
  136. c.Assert(isNetworkAvailable(c, "test1"), check.Equals, false)
  137. ipam2 := network.IPAM{
  138. Driver: "default",
  139. Config: []network.IPAMConfig{{Subnet: "192.169.0.0/16", Gateway: "192.169.100.100"}},
  140. }
  141. // test2 bridge network does not overlap
  142. config2 := types.NetworkCreate{
  143. Name: "test2",
  144. Driver: "bridge",
  145. IPAM: ipam2,
  146. }
  147. createNetwork(c, config2, true)
  148. c.Assert(isNetworkAvailable(c, "test2"), check.Equals, true)
  149. // remove test0 and retry to create test1
  150. deleteNetwork(c, id0, true)
  151. createNetwork(c, config1, true)
  152. c.Assert(isNetworkAvailable(c, "test1"), check.Equals, true)
  153. // for networks w/o ipam specified, docker will choose proper non-overlapping subnets
  154. createNetwork(c, types.NetworkCreate{Name: "test3"}, true)
  155. c.Assert(isNetworkAvailable(c, "test3"), check.Equals, true)
  156. createNetwork(c, types.NetworkCreate{Name: "test4"}, true)
  157. c.Assert(isNetworkAvailable(c, "test4"), check.Equals, true)
  158. createNetwork(c, types.NetworkCreate{Name: "test5"}, true)
  159. c.Assert(isNetworkAvailable(c, "test5"), check.Equals, true)
  160. for i := 1; i < 6; i++ {
  161. deleteNetwork(c, fmt.Sprintf("test%d", i), true)
  162. }
  163. }
  164. func isNetworkAvailable(c *check.C, name string) bool {
  165. status, body, err := sockRequest("GET", "/networks", nil)
  166. c.Assert(status, check.Equals, http.StatusOK)
  167. c.Assert(err, check.IsNil)
  168. nJSON := []types.NetworkResource{}
  169. err = json.Unmarshal(body, &nJSON)
  170. c.Assert(err, check.IsNil)
  171. for _, n := range nJSON {
  172. if n.Name == name {
  173. return true
  174. }
  175. }
  176. return false
  177. }
  178. func getNetworkIDByName(c *check.C, name string) string {
  179. var (
  180. v = url.Values{}
  181. filterArgs = filters.Args{}
  182. )
  183. filterArgs["name"] = []string{name}
  184. filterJSON, err := filters.ToParam(filterArgs)
  185. c.Assert(err, check.IsNil)
  186. v.Set("filters", filterJSON)
  187. status, body, err := sockRequest("GET", "/networks?"+v.Encode(), nil)
  188. c.Assert(status, check.Equals, http.StatusOK)
  189. c.Assert(err, check.IsNil)
  190. nJSON := []types.NetworkResource{}
  191. err = json.Unmarshal(body, &nJSON)
  192. c.Assert(err, check.IsNil)
  193. c.Assert(len(nJSON), check.Equals, 1)
  194. return nJSON[0].ID
  195. }
  196. func getNetworkResource(c *check.C, id string) *types.NetworkResource {
  197. _, obj, err := sockRequest("GET", "/networks/"+id, nil)
  198. c.Assert(err, check.IsNil)
  199. nr := types.NetworkResource{}
  200. err = json.Unmarshal(obj, &nr)
  201. c.Assert(err, check.IsNil)
  202. return &nr
  203. }
  204. func createNetwork(c *check.C, config types.NetworkCreate, shouldSucceed bool) string {
  205. status, resp, err := sockRequest("POST", "/networks/create", config)
  206. if !shouldSucceed {
  207. c.Assert(status, check.Not(check.Equals), http.StatusCreated)
  208. return ""
  209. }
  210. c.Assert(status, check.Equals, http.StatusCreated)
  211. c.Assert(err, check.IsNil)
  212. var nr types.NetworkCreateResponse
  213. err = json.Unmarshal(resp, &nr)
  214. c.Assert(err, check.IsNil)
  215. return nr.ID
  216. }
  217. func connectNetwork(c *check.C, nid, cid string) {
  218. config := types.NetworkConnect{
  219. Container: cid,
  220. }
  221. status, _, err := sockRequest("POST", "/networks/"+nid+"/connect", config)
  222. c.Assert(status, check.Equals, http.StatusOK)
  223. c.Assert(err, check.IsNil)
  224. }
  225. func disconnectNetwork(c *check.C, nid, cid string) {
  226. config := types.NetworkConnect{
  227. Container: cid,
  228. }
  229. status, _, err := sockRequest("POST", "/networks/"+nid+"/disconnect", config)
  230. c.Assert(status, check.Equals, http.StatusOK)
  231. c.Assert(err, check.IsNil)
  232. }
  233. func deleteNetwork(c *check.C, id string, shouldSucceed bool) {
  234. status, _, err := sockRequest("DELETE", "/networks/"+id, nil)
  235. if !shouldSucceed {
  236. c.Assert(status, check.Not(check.Equals), http.StatusOK)
  237. c.Assert(err, check.NotNil)
  238. return
  239. }
  240. c.Assert(status, check.Equals, http.StatusOK)
  241. c.Assert(err, check.IsNil)
  242. }