github.com/robertojrojas/docker@v1.9.1/integration-cli/docker_cli_start_volume_driver_unix_test.go (about) 1 // +build !windows 2 3 package main 4 5 import ( 6 "encoding/json" 7 "fmt" 8 "io/ioutil" 9 "net/http" 10 "net/http/httptest" 11 "os" 12 "os/exec" 13 "path/filepath" 14 "strings" 15 "time" 16 17 "github.com/go-check/check" 18 ) 19 20 func init() { 21 check.Suite(&DockerExternalVolumeSuite{ 22 ds: &DockerSuite{}, 23 }) 24 } 25 26 type eventCounter struct { 27 activations int 28 creations int 29 removals int 30 mounts int 31 unmounts int 32 paths int 33 } 34 35 type DockerExternalVolumeSuite struct { 36 server *httptest.Server 37 ds *DockerSuite 38 d *Daemon 39 ec *eventCounter 40 } 41 42 func (s *DockerExternalVolumeSuite) SetUpTest(c *check.C) { 43 s.d = NewDaemon(c) 44 s.ec = &eventCounter{} 45 } 46 47 func (s *DockerExternalVolumeSuite) TearDownTest(c *check.C) { 48 s.d.Stop() 49 s.ds.TearDownTest(c) 50 } 51 52 func (s *DockerExternalVolumeSuite) SetUpSuite(c *check.C) { 53 mux := http.NewServeMux() 54 s.server = httptest.NewServer(mux) 55 56 type pluginRequest struct { 57 name string 58 } 59 60 mux.HandleFunc("/Plugin.Activate", func(w http.ResponseWriter, r *http.Request) { 61 s.ec.activations++ 62 63 w.Header().Set("Content-Type", "application/vnd.docker.plugins.v1+json") 64 fmt.Fprintln(w, `{"Implements": ["VolumeDriver"]}`) 65 }) 66 67 mux.HandleFunc("/VolumeDriver.Create", func(w http.ResponseWriter, r *http.Request) { 68 s.ec.creations++ 69 70 w.Header().Set("Content-Type", "application/vnd.docker.plugins.v1+json") 71 fmt.Fprintln(w, `{}`) 72 }) 73 74 mux.HandleFunc("/VolumeDriver.Remove", func(w http.ResponseWriter, r *http.Request) { 75 s.ec.removals++ 76 77 w.Header().Set("Content-Type", "application/vnd.docker.plugins.v1+json") 78 fmt.Fprintln(w, `{}`) 79 }) 80 81 mux.HandleFunc("/VolumeDriver.Path", func(w http.ResponseWriter, r *http.Request) { 82 s.ec.paths++ 83 84 var pr pluginRequest 85 if err := json.NewDecoder(r.Body).Decode(&pr); err != nil { 86 http.Error(w, err.Error(), 500) 87 } 88 89 p := hostVolumePath(pr.name) 90 91 w.Header().Set("Content-Type", "application/vnd.docker.plugins.v1+json") 92 fmt.Fprintln(w, fmt.Sprintf("{\"Mountpoint\": \"%s\"}", p)) 93 }) 94 95 mux.HandleFunc("/VolumeDriver.Mount", func(w http.ResponseWriter, r *http.Request) { 96 s.ec.mounts++ 97 98 var pr pluginRequest 99 if err := json.NewDecoder(r.Body).Decode(&pr); err != nil { 100 http.Error(w, err.Error(), 500) 101 } 102 103 p := hostVolumePath(pr.name) 104 if err := os.MkdirAll(p, 0755); err != nil { 105 http.Error(w, err.Error(), 500) 106 } 107 108 if err := ioutil.WriteFile(filepath.Join(p, "test"), []byte(s.server.URL), 0644); err != nil { 109 http.Error(w, err.Error(), 500) 110 } 111 112 w.Header().Set("Content-Type", "application/vnd.docker.plugins.v1+json") 113 fmt.Fprintln(w, fmt.Sprintf("{\"Mountpoint\": \"%s\"}", p)) 114 }) 115 116 mux.HandleFunc("/VolumeDriver.Unmount", func(w http.ResponseWriter, r *http.Request) { 117 s.ec.unmounts++ 118 119 var pr pluginRequest 120 if err := json.NewDecoder(r.Body).Decode(&pr); err != nil { 121 http.Error(w, err.Error(), 500) 122 } 123 124 w.Header().Set("Content-Type", "application/vnd.docker.plugins.v1+json") 125 fmt.Fprintln(w, `{}`) 126 }) 127 128 if err := os.MkdirAll("/etc/docker/plugins", 0755); err != nil { 129 c.Fatal(err) 130 } 131 132 if err := ioutil.WriteFile("/etc/docker/plugins/test-external-volume-driver.spec", []byte(s.server.URL), 0644); err != nil { 133 c.Fatal(err) 134 } 135 } 136 137 func (s *DockerExternalVolumeSuite) TearDownSuite(c *check.C) { 138 s.server.Close() 139 140 if err := os.RemoveAll("/etc/docker/plugins"); err != nil { 141 c.Fatal(err) 142 } 143 } 144 145 func (s *DockerExternalVolumeSuite) TestStartExternalNamedVolumeDriver(c *check.C) { 146 if err := s.d.StartWithBusybox(); err != nil { 147 c.Fatal(err) 148 } 149 150 out, err := s.d.Cmd("run", "--rm", "--name", "test-data", "-v", "external-volume-test:/tmp/external-volume-test", "--volume-driver", "test-external-volume-driver", "busybox:latest", "cat", "/tmp/external-volume-test/test") 151 if err != nil { 152 c.Fatal(out, err) 153 } 154 155 if !strings.Contains(out, s.server.URL) { 156 c.Fatalf("External volume mount failed. Output: %s\n", out) 157 } 158 159 p := hostVolumePath("external-volume-test") 160 _, err = os.Lstat(p) 161 if err == nil { 162 c.Fatalf("Expected error checking volume path in host: %s\n", p) 163 } 164 165 if !os.IsNotExist(err) { 166 c.Fatalf("Expected volume path in host to not exist: %s, %v\n", p, err) 167 } 168 169 c.Assert(s.ec.activations, check.Equals, 1) 170 c.Assert(s.ec.creations, check.Equals, 1) 171 c.Assert(s.ec.removals, check.Equals, 1) 172 c.Assert(s.ec.mounts, check.Equals, 1) 173 c.Assert(s.ec.unmounts, check.Equals, 1) 174 } 175 176 func (s *DockerExternalVolumeSuite) TestStartExternalVolumeUnnamedDriver(c *check.C) { 177 if err := s.d.StartWithBusybox(); err != nil { 178 c.Fatal(err) 179 } 180 181 out, err := s.d.Cmd("run", "--rm", "--name", "test-data", "-v", "/tmp/external-volume-test", "--volume-driver", "test-external-volume-driver", "busybox:latest", "cat", "/tmp/external-volume-test/test") 182 if err != nil { 183 c.Fatal(err) 184 } 185 186 if !strings.Contains(out, s.server.URL) { 187 c.Fatalf("External volume mount failed. Output: %s\n", out) 188 } 189 190 c.Assert(s.ec.activations, check.Equals, 1) 191 c.Assert(s.ec.creations, check.Equals, 1) 192 c.Assert(s.ec.removals, check.Equals, 1) 193 c.Assert(s.ec.mounts, check.Equals, 1) 194 c.Assert(s.ec.unmounts, check.Equals, 1) 195 } 196 197 func (s DockerExternalVolumeSuite) TestStartExternalVolumeDriverVolumesFrom(c *check.C) { 198 if err := s.d.StartWithBusybox(); err != nil { 199 c.Fatal(err) 200 } 201 202 out, err := s.d.Cmd("run", "-d", "--name", "vol-test1", "-v", "/foo", "--volume-driver", "test-external-volume-driver", "busybox:latest") 203 c.Assert(err, check.IsNil, check.Commentf(out)) 204 205 out, err = s.d.Cmd("run", "--rm", "--volumes-from", "vol-test1", "--name", "vol-test2", "busybox", "ls", "/tmp") 206 c.Assert(err, check.IsNil, check.Commentf(out)) 207 208 out, err = s.d.Cmd("rm", "-fv", "vol-test1") 209 c.Assert(err, check.IsNil, check.Commentf(out)) 210 211 c.Assert(s.ec.activations, check.Equals, 1) 212 c.Assert(s.ec.creations, check.Equals, 1) 213 c.Assert(s.ec.removals, check.Equals, 1) 214 c.Assert(s.ec.mounts, check.Equals, 2) 215 c.Assert(s.ec.unmounts, check.Equals, 2) 216 } 217 218 func (s DockerExternalVolumeSuite) TestStartExternalVolumeDriverDeleteContainer(c *check.C) { 219 if err := s.d.StartWithBusybox(); err != nil { 220 c.Fatal(err) 221 } 222 223 if out, err := s.d.Cmd("run", "-d", "--name", "vol-test1", "-v", "/foo", "--volume-driver", "test-external-volume-driver", "busybox:latest"); err != nil { 224 c.Fatal(out, err) 225 } 226 227 if out, err := s.d.Cmd("rm", "-fv", "vol-test1"); err != nil { 228 c.Fatal(out, err) 229 } 230 231 c.Assert(s.ec.activations, check.Equals, 1) 232 c.Assert(s.ec.creations, check.Equals, 1) 233 c.Assert(s.ec.removals, check.Equals, 1) 234 c.Assert(s.ec.mounts, check.Equals, 1) 235 c.Assert(s.ec.unmounts, check.Equals, 1) 236 } 237 238 func hostVolumePath(name string) string { 239 return fmt.Sprintf("/var/lib/docker/volumes/%s", name) 240 } 241 242 func (s *DockerExternalVolumeSuite) TestStartExternalNamedVolumeDriverCheckBindLocalVolume(c *check.C) { 243 if err := s.d.StartWithBusybox(); err != nil { 244 c.Fatal(err) 245 } 246 247 expected := s.server.URL 248 249 dockerfile := fmt.Sprintf(`FROM busybox:latest 250 RUN mkdir /nobindthenlocalvol 251 RUN echo %s > /nobindthenlocalvol/test 252 VOLUME ["/nobindthenlocalvol"]`, expected) 253 254 img := "test-checkbindlocalvolume" 255 256 _, err := buildImageWithOutInDamon(s.d.sock(), img, dockerfile, true) 257 c.Assert(err, check.IsNil) 258 259 out, err := s.d.Cmd("run", "--rm", "--name", "test-data-nobind", "-v", "external-volume-test:/tmp/external-volume-test", "--volume-driver", "test-external-volume-driver", img, "cat", "/nobindthenlocalvol/test") 260 c.Assert(err, check.IsNil) 261 262 if !strings.Contains(out, expected) { 263 c.Fatalf("External volume mount failed. Output: %s\n", out) 264 } 265 266 c.Assert(s.ec.activations, check.Equals, 1) 267 c.Assert(s.ec.creations, check.Equals, 1) 268 c.Assert(s.ec.removals, check.Equals, 1) 269 c.Assert(s.ec.mounts, check.Equals, 1) 270 c.Assert(s.ec.unmounts, check.Equals, 1) 271 } 272 273 // Make sure a request to use a down driver doesn't block other requests 274 func (s *DockerExternalVolumeSuite) TestStartExternalVolumeDriverLookupNotBlocked(c *check.C) { 275 specPath := "/etc/docker/plugins/down-driver.spec" 276 err := ioutil.WriteFile("/etc/docker/plugins/down-driver.spec", []byte("tcp://127.0.0.7:9999"), 0644) 277 c.Assert(err, check.IsNil) 278 defer os.RemoveAll(specPath) 279 280 chCmd1 := make(chan struct{}) 281 chCmd2 := make(chan error) 282 cmd1 := exec.Command(dockerBinary, "volume", "create", "-d", "down-driver") 283 cmd2 := exec.Command(dockerBinary, "volume", "create") 284 285 c.Assert(cmd1.Start(), check.IsNil) 286 defer cmd1.Process.Kill() 287 time.Sleep(100 * time.Millisecond) // ensure API has been called 288 c.Assert(cmd2.Start(), check.IsNil) 289 290 go func() { 291 cmd1.Wait() 292 close(chCmd1) 293 }() 294 go func() { 295 chCmd2 <- cmd2.Wait() 296 }() 297 298 select { 299 case <-chCmd1: 300 cmd2.Process.Kill() 301 c.Fatalf("volume create with down driver finished unexpectedly") 302 case err := <-chCmd2: 303 c.Assert(err, check.IsNil, check.Commentf("error creating volume")) 304 case <-time.After(5 * time.Second): 305 c.Fatal("volume creates are blocked by previous create requests when previous driver is down") 306 cmd2.Process.Kill() 307 } 308 } 309 310 func (s *DockerExternalVolumeSuite) TestStartExternalVolumeDriverRetryNotImmediatelyExists(c *check.C) { 311 if err := s.d.StartWithBusybox(); err != nil { 312 c.Fatal(err) 313 } 314 315 specPath := "/etc/docker/plugins/test-external-volume-driver-retry.spec" 316 os.RemoveAll(specPath) 317 defer os.RemoveAll(specPath) 318 319 errchan := make(chan error) 320 go func() { 321 if out, err := s.d.Cmd("run", "--rm", "--name", "test-data-retry", "-v", "external-volume-test:/tmp/external-volume-test", "--volume-driver", "test-external-volume-driver-retry", "busybox:latest"); err != nil { 322 errchan <- fmt.Errorf("%v:\n%s", err, out) 323 } 324 close(errchan) 325 }() 326 go func() { 327 // wait for a retry to occur, then create spec to allow plugin to register 328 time.Sleep(2000 * time.Millisecond) 329 if err := ioutil.WriteFile(specPath, []byte(s.server.URL), 0644); err != nil { 330 c.Fatal(err) 331 } 332 }() 333 334 select { 335 case err := <-errchan: 336 if err != nil { 337 c.Fatal(err) 338 } 339 case <-time.After(8 * time.Second): 340 c.Fatal("volume creates fail when plugin not immediately available") 341 } 342 343 c.Assert(s.ec.activations, check.Equals, 1) 344 c.Assert(s.ec.creations, check.Equals, 1) 345 c.Assert(s.ec.removals, check.Equals, 1) 346 c.Assert(s.ec.mounts, check.Equals, 1) 347 c.Assert(s.ec.unmounts, check.Equals, 1) 348 } 349 350 func (s *DockerExternalVolumeSuite) TestStartExternalVolumeDriverBindExternalVolume(c *check.C) { 351 dockerCmd(c, "volume", "create", "-d", "test-external-volume-driver", "--name", "foo") 352 dockerCmd(c, "run", "-d", "--name", "testing", "-v", "foo:/bar", "busybox", "top") 353 354 var mounts []struct { 355 Name string 356 Driver string 357 } 358 out, err := inspectFieldJSON("testing", "Mounts") 359 c.Assert(err, check.IsNil) 360 c.Assert(json.NewDecoder(strings.NewReader(out)).Decode(&mounts), check.IsNil) 361 c.Assert(len(mounts), check.Equals, 1, check.Commentf(out)) 362 c.Assert(mounts[0].Name, check.Equals, "foo") 363 c.Assert(mounts[0].Driver, check.Equals, "test-external-volume-driver") 364 }