runtime_test.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897
  1. package docker
  2. import (
  3. "bytes"
  4. "fmt"
  5. "io"
  6. std_log "log"
  7. "net"
  8. "net/url"
  9. "os"
  10. "path/filepath"
  11. "runtime"
  12. "strconv"
  13. "strings"
  14. "syscall"
  15. "testing"
  16. "time"
  17. "github.com/Sirupsen/logrus"
  18. "github.com/docker/docker/daemon"
  19. "github.com/docker/docker/daemon/execdriver"
  20. "github.com/docker/docker/engine"
  21. "github.com/docker/docker/graph"
  22. "github.com/docker/docker/image"
  23. "github.com/docker/docker/nat"
  24. "github.com/docker/docker/pkg/fileutils"
  25. "github.com/docker/docker/pkg/ioutils"
  26. "github.com/docker/docker/pkg/reexec"
  27. "github.com/docker/docker/pkg/stringid"
  28. "github.com/docker/docker/runconfig"
  29. "github.com/docker/docker/utils"
  30. )
  31. const (
  32. unitTestImageName = "docker-test-image"
  33. unitTestImageID = "83599e29c455eb719f77d799bc7c51521b9551972f5a850d7ad265bc1b5292f6" // 1.0
  34. unitTestImageIDShort = "83599e29c455"
  35. unitTestNetworkBridge = "testdockbr0"
  36. unitTestStoreBase = "/var/lib/docker/unit-tests"
  37. unitTestDockerTmpdir = "/var/lib/docker/tmp"
  38. testDaemonAddr = "127.0.0.1:4270"
  39. testDaemonProto = "tcp"
  40. testDaemonHttpsProto = "tcp"
  41. testDaemonHttpsAddr = "localhost:4271"
  42. testDaemonRogueHttpsAddr = "localhost:4272"
  43. )
  44. var (
  45. // FIXME: globalDaemon is deprecated by globalEngine. All tests should be converted.
  46. globalDaemon *daemon.Daemon
  47. globalEngine *engine.Engine
  48. globalHttpsEngine *engine.Engine
  49. globalRogueHttpsEngine *engine.Engine
  50. startFds int
  51. startGoroutines int
  52. )
  53. // FIXME: nuke() is deprecated by Daemon.Nuke()
  54. func nuke(daemon *daemon.Daemon) error {
  55. return daemon.Nuke()
  56. }
  57. // FIXME: cleanup and nuke are redundant.
  58. func cleanup(eng *engine.Engine, t *testing.T) error {
  59. daemon := mkDaemonFromEngine(eng, t)
  60. for _, container := range daemon.List() {
  61. container.Kill()
  62. daemon.Rm(container)
  63. }
  64. images, err := daemon.Repositories().Images(&graph.ImagesConfig{})
  65. if err != nil {
  66. t.Fatal(err)
  67. }
  68. for _, image := range images {
  69. if image.ID != unitTestImageID {
  70. eng.Job("image_delete", image.ID).Run()
  71. }
  72. }
  73. return nil
  74. }
  75. func init() {
  76. // Always use the same driver (vfs) for all integration tests.
  77. // To test other drivers, we need a dedicated driver validation suite.
  78. os.Setenv("DOCKER_DRIVER", "vfs")
  79. os.Setenv("TEST", "1")
  80. os.Setenv("DOCKER_TMPDIR", unitTestDockerTmpdir)
  81. // Hack to run sys init during unit testing
  82. if reexec.Init() {
  83. return
  84. }
  85. if uid := syscall.Geteuid(); uid != 0 {
  86. logrus.Fatalf("docker tests need to be run as root")
  87. }
  88. // Copy dockerinit into our current testing directory, if provided (so we can test a separate dockerinit binary)
  89. if dockerinit := os.Getenv("TEST_DOCKERINIT_PATH"); dockerinit != "" {
  90. src, err := os.Open(dockerinit)
  91. if err != nil {
  92. logrus.Fatalf("Unable to open TEST_DOCKERINIT_PATH: %s", err)
  93. }
  94. defer src.Close()
  95. dst, err := os.OpenFile(filepath.Join(filepath.Dir(utils.SelfPath()), "dockerinit"), os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0555)
  96. if err != nil {
  97. logrus.Fatalf("Unable to create dockerinit in test directory: %s", err)
  98. }
  99. defer dst.Close()
  100. if _, err := io.Copy(dst, src); err != nil {
  101. logrus.Fatalf("Unable to copy dockerinit to TEST_DOCKERINIT_PATH: %s", err)
  102. }
  103. dst.Close()
  104. src.Close()
  105. }
  106. // Setup the base daemon, which will be duplicated for each test.
  107. // (no tests are run directly in the base)
  108. setupBaseImage()
  109. // Create the "global daemon" with a long-running daemons for integration tests
  110. spawnGlobalDaemon()
  111. spawnLegitHttpsDaemon()
  112. spawnRogueHttpsDaemon()
  113. startFds, startGoroutines = fileutils.GetTotalUsedFds(), runtime.NumGoroutine()
  114. }
  115. func setupBaseImage() {
  116. eng := newTestEngine(std_log.New(os.Stderr, "", 0), false, unitTestStoreBase)
  117. job := eng.Job("image_inspect", unitTestImageName)
  118. img, _ := job.Stdout.AddEnv()
  119. // If the unit test is not found, try to download it.
  120. if err := job.Run(); err != nil || img.Get("Id") != unitTestImageID {
  121. // Retrieve the Image
  122. job = eng.Job("pull", unitTestImageName)
  123. job.Stdout.Add(ioutils.NopWriteCloser(os.Stdout))
  124. if err := job.Run(); err != nil {
  125. logrus.Fatalf("Unable to pull the test image: %s", err)
  126. }
  127. }
  128. }
  129. func spawnGlobalDaemon() {
  130. if globalDaemon != nil {
  131. logrus.Debugf("Global daemon already exists. Skipping.")
  132. return
  133. }
  134. t := std_log.New(os.Stderr, "", 0)
  135. eng := NewTestEngine(t)
  136. globalEngine = eng
  137. globalDaemon = mkDaemonFromEngine(eng, t)
  138. // Spawn a Daemon
  139. go func() {
  140. logrus.Debugf("Spawning global daemon for integration tests")
  141. listenURL := &url.URL{
  142. Scheme: testDaemonProto,
  143. Host: testDaemonAddr,
  144. }
  145. job := eng.Job("serveapi", listenURL.String())
  146. job.SetenvBool("Logging", true)
  147. if err := job.Run(); err != nil {
  148. logrus.Fatalf("Unable to spawn the test daemon: %s", err)
  149. }
  150. }()
  151. // Give some time to ListenAndServer to actually start
  152. // FIXME: use inmem transports instead of tcp
  153. time.Sleep(time.Second)
  154. if err := eng.Job("acceptconnections").Run(); err != nil {
  155. logrus.Fatalf("Unable to accept connections for test api: %s", err)
  156. }
  157. }
  158. func spawnLegitHttpsDaemon() {
  159. if globalHttpsEngine != nil {
  160. return
  161. }
  162. globalHttpsEngine = spawnHttpsDaemon(testDaemonHttpsAddr, "fixtures/https/ca.pem",
  163. "fixtures/https/server-cert.pem", "fixtures/https/server-key.pem")
  164. }
  165. func spawnRogueHttpsDaemon() {
  166. if globalRogueHttpsEngine != nil {
  167. return
  168. }
  169. globalRogueHttpsEngine = spawnHttpsDaemon(testDaemonRogueHttpsAddr, "fixtures/https/ca.pem",
  170. "fixtures/https/server-rogue-cert.pem", "fixtures/https/server-rogue-key.pem")
  171. }
  172. func spawnHttpsDaemon(addr, cacert, cert, key string) *engine.Engine {
  173. t := std_log.New(os.Stderr, "", 0)
  174. root, err := newTestDirectory(unitTestStoreBase)
  175. if err != nil {
  176. t.Fatal(err)
  177. }
  178. // FIXME: here we don't use NewTestEngine because it configures the daemon with Autorestart=false,
  179. // and we want to set it to true.
  180. eng := newTestEngine(t, true, root)
  181. // Spawn a Daemon
  182. go func() {
  183. logrus.Debugf("Spawning https daemon for integration tests")
  184. listenURL := &url.URL{
  185. Scheme: testDaemonHttpsProto,
  186. Host: addr,
  187. }
  188. job := eng.Job("serveapi", listenURL.String())
  189. job.SetenvBool("Logging", true)
  190. job.SetenvBool("Tls", true)
  191. job.SetenvBool("TlsVerify", true)
  192. job.Setenv("TlsCa", cacert)
  193. job.Setenv("TlsCert", cert)
  194. job.Setenv("TlsKey", key)
  195. if err := job.Run(); err != nil {
  196. logrus.Fatalf("Unable to spawn the test daemon: %s", err)
  197. }
  198. }()
  199. // Give some time to ListenAndServer to actually start
  200. time.Sleep(time.Second)
  201. if err := eng.Job("acceptconnections").Run(); err != nil {
  202. logrus.Fatalf("Unable to accept connections for test api: %s", err)
  203. }
  204. return eng
  205. }
  206. // FIXME: test that ImagePull(json=true) send correct json output
  207. func GetTestImage(daemon *daemon.Daemon) *image.Image {
  208. imgs, err := daemon.Graph().Map()
  209. if err != nil {
  210. logrus.Fatalf("Unable to get the test image: %s", err)
  211. }
  212. for _, image := range imgs {
  213. if image.ID == unitTestImageID {
  214. return image
  215. }
  216. }
  217. logrus.Fatalf("Test image %v not found in %s: %s", unitTestImageID, daemon.Graph().Root, imgs)
  218. return nil
  219. }
  220. func TestDaemonCreate(t *testing.T) {
  221. daemon := mkDaemon(t)
  222. defer nuke(daemon)
  223. // Make sure we start we 0 containers
  224. if len(daemon.List()) != 0 {
  225. t.Errorf("Expected 0 containers, %v found", len(daemon.List()))
  226. }
  227. container, _, err := daemon.Create(&runconfig.Config{
  228. Image: GetTestImage(daemon).ID,
  229. Cmd: runconfig.NewCommand("ls", "-al"),
  230. },
  231. &runconfig.HostConfig{},
  232. "",
  233. )
  234. if err != nil {
  235. t.Fatal(err)
  236. }
  237. defer func() {
  238. if err := daemon.Rm(container); err != nil {
  239. t.Error(err)
  240. }
  241. }()
  242. // Make sure we can find the newly created container with List()
  243. if len(daemon.List()) != 1 {
  244. t.Errorf("Expected 1 container, %v found", len(daemon.List()))
  245. }
  246. // Make sure the container List() returns is the right one
  247. if daemon.List()[0].ID != container.ID {
  248. t.Errorf("Unexpected container %v returned by List", daemon.List()[0])
  249. }
  250. // Make sure we can get the container with Get()
  251. if _, err := daemon.Get(container.ID); err != nil {
  252. t.Errorf("Unable to get newly created container")
  253. }
  254. // Make sure it is the right container
  255. if c, _ := daemon.Get(container.ID); c != container {
  256. t.Errorf("Get() returned the wrong container")
  257. }
  258. // Make sure Exists returns it as existing
  259. if !daemon.Exists(container.ID) {
  260. t.Errorf("Exists() returned false for a newly created container")
  261. }
  262. // Test that conflict error displays correct details
  263. cmd := runconfig.NewCommand("ls", "-al")
  264. testContainer, _, _ := daemon.Create(
  265. &runconfig.Config{
  266. Image: GetTestImage(daemon).ID,
  267. Cmd: cmd,
  268. },
  269. &runconfig.HostConfig{},
  270. "conflictname",
  271. )
  272. if _, _, err := daemon.Create(&runconfig.Config{Image: GetTestImage(daemon).ID, Cmd: cmd}, &runconfig.HostConfig{}, testContainer.Name); err == nil || !strings.Contains(err.Error(), stringid.TruncateID(testContainer.ID)) {
  273. t.Fatalf("Name conflict error doesn't include the correct short id. Message was: %v", err)
  274. }
  275. // Make sure create with bad parameters returns an error
  276. if _, _, err = daemon.Create(&runconfig.Config{Image: GetTestImage(daemon).ID}, &runconfig.HostConfig{}, ""); err == nil {
  277. t.Fatal("Builder.Create should throw an error when Cmd is missing")
  278. }
  279. if _, _, err := daemon.Create(
  280. &runconfig.Config{
  281. Image: GetTestImage(daemon).ID,
  282. Cmd: runconfig.NewCommand(),
  283. },
  284. &runconfig.HostConfig{},
  285. "",
  286. ); err == nil {
  287. t.Fatal("Builder.Create should throw an error when Cmd is empty")
  288. }
  289. config := &runconfig.Config{
  290. Image: GetTestImage(daemon).ID,
  291. Cmd: runconfig.NewCommand("/bin/ls"),
  292. PortSpecs: []string{"80"},
  293. }
  294. container, _, err = daemon.Create(config, &runconfig.HostConfig{}, "")
  295. _, err = daemon.Commit(container, "testrepo", "testtag", "", "", true, config)
  296. if err != nil {
  297. t.Error(err)
  298. }
  299. // test expose 80:8000
  300. container, warnings, err := daemon.Create(&runconfig.Config{
  301. Image: GetTestImage(daemon).ID,
  302. Cmd: runconfig.NewCommand("ls", "-al"),
  303. PortSpecs: []string{"80:8000"},
  304. },
  305. &runconfig.HostConfig{},
  306. "",
  307. )
  308. if err != nil {
  309. t.Fatal(err)
  310. }
  311. if warnings == nil || len(warnings) != 1 {
  312. t.Error("Expected a warning, got none")
  313. }
  314. }
  315. func TestDestroy(t *testing.T) {
  316. daemon := mkDaemon(t)
  317. defer nuke(daemon)
  318. container, _, err := daemon.Create(&runconfig.Config{
  319. Image: GetTestImage(daemon).ID,
  320. Cmd: runconfig.NewCommand("ls", "-al"),
  321. },
  322. &runconfig.HostConfig{},
  323. "")
  324. if err != nil {
  325. t.Fatal(err)
  326. }
  327. // Destroy
  328. if err := daemon.Rm(container); err != nil {
  329. t.Error(err)
  330. }
  331. // Make sure daemon.Exists() behaves correctly
  332. if daemon.Exists("test_destroy") {
  333. t.Errorf("Exists() returned true")
  334. }
  335. // Make sure daemon.List() doesn't list the destroyed container
  336. if len(daemon.List()) != 0 {
  337. t.Errorf("Expected 0 container, %v found", len(daemon.List()))
  338. }
  339. // Make sure daemon.Get() refuses to return the unexisting container
  340. if c, _ := daemon.Get(container.ID); c != nil {
  341. t.Errorf("Got a container that should not exist")
  342. }
  343. // Test double destroy
  344. if err := daemon.Rm(container); err == nil {
  345. // It should have failed
  346. t.Errorf("Double destroy did not fail")
  347. }
  348. }
  349. func TestGet(t *testing.T) {
  350. daemon := mkDaemon(t)
  351. defer nuke(daemon)
  352. container1, _, _ := mkContainer(daemon, []string{"_", "ls", "-al"}, t)
  353. defer daemon.Rm(container1)
  354. container2, _, _ := mkContainer(daemon, []string{"_", "ls", "-al"}, t)
  355. defer daemon.Rm(container2)
  356. container3, _, _ := mkContainer(daemon, []string{"_", "ls", "-al"}, t)
  357. defer daemon.Rm(container3)
  358. if c, _ := daemon.Get(container1.ID); c != container1 {
  359. t.Errorf("Get(test1) returned %v while expecting %v", c, container1)
  360. }
  361. if c, _ := daemon.Get(container2.ID); c != container2 {
  362. t.Errorf("Get(test2) returned %v while expecting %v", c, container2)
  363. }
  364. if c, _ := daemon.Get(container3.ID); c != container3 {
  365. t.Errorf("Get(test3) returned %v while expecting %v", c, container3)
  366. }
  367. }
  368. func startEchoServerContainer(t *testing.T, proto string) (*daemon.Daemon, *daemon.Container, string) {
  369. var (
  370. err error
  371. id string
  372. strPort string
  373. eng = NewTestEngine(t)
  374. daemon = mkDaemonFromEngine(eng, t)
  375. port = 5554
  376. p nat.Port
  377. )
  378. defer func() {
  379. if err != nil {
  380. daemon.Nuke()
  381. }
  382. }()
  383. for {
  384. port += 1
  385. strPort = strconv.Itoa(port)
  386. var cmd string
  387. if proto == "tcp" {
  388. cmd = "socat TCP-LISTEN:" + strPort + ",reuseaddr,fork EXEC:/bin/cat"
  389. } else if proto == "udp" {
  390. cmd = "socat UDP-RECVFROM:" + strPort + ",fork EXEC:/bin/cat"
  391. } else {
  392. t.Fatal(fmt.Errorf("Unknown protocol %v", proto))
  393. }
  394. ep := make(map[nat.Port]struct{}, 1)
  395. p = nat.Port(fmt.Sprintf("%s/%s", strPort, proto))
  396. ep[p] = struct{}{}
  397. c := &runconfig.Config{
  398. Image: unitTestImageID,
  399. Cmd: runconfig.NewCommand("sh", "-c", cmd),
  400. PortSpecs: []string{fmt.Sprintf("%s/%s", strPort, proto)},
  401. ExposedPorts: ep,
  402. }
  403. id, _, err = daemon.ContainerCreate(unitTestImageID, c, &runconfig.HostConfig{})
  404. // FIXME: this relies on the undocumented behavior of daemon.Create
  405. // which will return a nil error AND container if the exposed ports
  406. // are invalid. That behavior should be fixed!
  407. if id != "" {
  408. break
  409. }
  410. t.Logf("Port %v already in use, trying another one", strPort)
  411. }
  412. if err := daemon.ContainerStart(id, &runconfig.HostConfig{}); err != nil {
  413. t.Fatal(err)
  414. }
  415. container, err := daemon.Get(id)
  416. if err != nil {
  417. t.Fatal(err)
  418. }
  419. setTimeout(t, "Waiting for the container to be started timed out", 2*time.Second, func() {
  420. for !container.IsRunning() {
  421. time.Sleep(10 * time.Millisecond)
  422. }
  423. })
  424. // Even if the state is running, lets give some time to lxc to spawn the process
  425. container.WaitStop(500 * time.Millisecond)
  426. strPort = container.NetworkSettings.Ports[p][0].HostPort
  427. return daemon, container, strPort
  428. }
  429. // Run a container with a TCP port allocated, and test that it can receive connections on localhost
  430. func TestAllocateTCPPortLocalhost(t *testing.T) {
  431. daemon, container, port := startEchoServerContainer(t, "tcp")
  432. defer nuke(daemon)
  433. defer container.Kill()
  434. for i := 0; i != 10; i++ {
  435. conn, err := net.Dial("tcp", fmt.Sprintf("localhost:%v", port))
  436. if err != nil {
  437. t.Fatal(err)
  438. }
  439. defer conn.Close()
  440. input := bytes.NewBufferString("well hello there\n")
  441. _, err = conn.Write(input.Bytes())
  442. if err != nil {
  443. t.Fatal(err)
  444. }
  445. buf := make([]byte, 16)
  446. read := 0
  447. conn.SetReadDeadline(time.Now().Add(3 * time.Second))
  448. read, err = conn.Read(buf)
  449. if err != nil {
  450. if err, ok := err.(*net.OpError); ok {
  451. if err.Err == syscall.ECONNRESET {
  452. t.Logf("Connection reset by the proxy, socat is probably not listening yet, trying again in a sec")
  453. conn.Close()
  454. time.Sleep(time.Second)
  455. continue
  456. }
  457. if err.Timeout() {
  458. t.Log("Timeout, trying again")
  459. conn.Close()
  460. continue
  461. }
  462. }
  463. t.Fatal(err)
  464. }
  465. output := string(buf[:read])
  466. if !strings.Contains(output, "well hello there") {
  467. t.Fatal(fmt.Errorf("[%v] doesn't contain [well hello there]", output))
  468. } else {
  469. return
  470. }
  471. }
  472. t.Fatal("No reply from the container")
  473. }
  474. // Run a container with an UDP port allocated, and test that it can receive connections on localhost
  475. func TestAllocateUDPPortLocalhost(t *testing.T) {
  476. daemon, container, port := startEchoServerContainer(t, "udp")
  477. defer nuke(daemon)
  478. defer container.Kill()
  479. conn, err := net.Dial("udp", fmt.Sprintf("localhost:%v", port))
  480. if err != nil {
  481. t.Fatal(err)
  482. }
  483. defer conn.Close()
  484. input := bytes.NewBufferString("well hello there\n")
  485. buf := make([]byte, 16)
  486. // Try for a minute, for some reason the select in socat may take ages
  487. // to return even though everything on the path seems fine (i.e: the
  488. // UDPProxy forwards the traffic correctly and you can see the packets
  489. // on the interface from within the container).
  490. for i := 0; i != 120; i++ {
  491. _, err := conn.Write(input.Bytes())
  492. if err != nil {
  493. t.Fatal(err)
  494. }
  495. conn.SetReadDeadline(time.Now().Add(500 * time.Millisecond))
  496. read, err := conn.Read(buf)
  497. if err == nil {
  498. output := string(buf[:read])
  499. if strings.Contains(output, "well hello there") {
  500. return
  501. }
  502. }
  503. }
  504. t.Fatal("No reply from the container")
  505. }
  506. func TestRestore(t *testing.T) {
  507. eng := NewTestEngine(t)
  508. daemon1 := mkDaemonFromEngine(eng, t)
  509. defer daemon1.Nuke()
  510. // Create a container with one instance of docker
  511. container1, _, _ := mkContainer(daemon1, []string{"_", "ls", "-al"}, t)
  512. defer daemon1.Rm(container1)
  513. // Create a second container meant to be killed
  514. container2, _, _ := mkContainer(daemon1, []string{"-i", "_", "/bin/cat"}, t)
  515. defer daemon1.Rm(container2)
  516. // Start the container non blocking
  517. if err := container2.Start(); err != nil {
  518. t.Fatal(err)
  519. }
  520. if !container2.IsRunning() {
  521. t.Fatalf("Container %v should appear as running but isn't", container2.ID)
  522. }
  523. // Simulate a crash/manual quit of dockerd: process dies, states stays 'Running'
  524. cStdin := container2.StdinPipe()
  525. cStdin.Close()
  526. if _, err := container2.WaitStop(2 * time.Second); err != nil {
  527. t.Fatal(err)
  528. }
  529. container2.SetRunning(42)
  530. container2.ToDisk()
  531. if len(daemon1.List()) != 2 {
  532. t.Errorf("Expected 2 container, %v found", len(daemon1.List()))
  533. }
  534. if err := container1.Run(); err != nil {
  535. t.Fatal(err)
  536. }
  537. if !container2.IsRunning() {
  538. t.Fatalf("Container %v should appear as running but isn't", container2.ID)
  539. }
  540. // Here are are simulating a docker restart - that is, reloading all containers
  541. // from scratch
  542. eng = newTestEngine(t, false, daemon1.Config().Root)
  543. daemon2 := mkDaemonFromEngine(eng, t)
  544. if len(daemon2.List()) != 2 {
  545. t.Errorf("Expected 2 container, %v found", len(daemon2.List()))
  546. }
  547. runningCount := 0
  548. for _, c := range daemon2.List() {
  549. if c.IsRunning() {
  550. t.Errorf("Running container found: %v (%v)", c.ID, c.Path)
  551. runningCount++
  552. }
  553. }
  554. if runningCount != 0 {
  555. t.Fatalf("Expected 0 container alive, %d found", runningCount)
  556. }
  557. container3, err := daemon2.Get(container1.ID)
  558. if err != nil {
  559. t.Fatal("Unable to Get container")
  560. }
  561. if err := container3.Run(); err != nil {
  562. t.Fatal(err)
  563. }
  564. container2.SetStopped(&execdriver.ExitStatus{ExitCode: 0})
  565. }
  566. func TestDefaultContainerName(t *testing.T) {
  567. eng := NewTestEngine(t)
  568. daemon := mkDaemonFromEngine(eng, t)
  569. defer nuke(daemon)
  570. config, _, _, err := parseRun([]string{unitTestImageID, "echo test"})
  571. if err != nil {
  572. t.Fatal(err)
  573. }
  574. container, err := daemon.Get(createNamedTestContainer(eng, config, t, "some_name"))
  575. if err != nil {
  576. t.Fatal(err)
  577. }
  578. containerID := container.ID
  579. if container.Name != "/some_name" {
  580. t.Fatalf("Expect /some_name got %s", container.Name)
  581. }
  582. c, err := daemon.Get("/some_name")
  583. if err != nil {
  584. t.Fatalf("Couldn't retrieve test container as /some_name")
  585. }
  586. if c.ID != containerID {
  587. t.Fatalf("Container /some_name has ID %s instead of %s", c.ID, containerID)
  588. }
  589. }
  590. func TestRandomContainerName(t *testing.T) {
  591. eng := NewTestEngine(t)
  592. daemon := mkDaemonFromEngine(eng, t)
  593. defer nuke(daemon)
  594. config, _, _, err := parseRun([]string{GetTestImage(daemon).ID, "echo test"})
  595. if err != nil {
  596. t.Fatal(err)
  597. }
  598. container, err := daemon.Get(createTestContainer(eng, config, t))
  599. if err != nil {
  600. t.Fatal(err)
  601. }
  602. containerID := container.ID
  603. if container.Name == "" {
  604. t.Fatalf("Expected not empty container name")
  605. }
  606. if c, err := daemon.Get(container.Name); err != nil {
  607. logrus.Fatalf("Could not lookup container %s by its name", container.Name)
  608. } else if c.ID != containerID {
  609. logrus.Fatalf("Looking up container name %s returned id %s instead of %s", container.Name, c.ID, containerID)
  610. }
  611. }
  612. func TestContainerNameValidation(t *testing.T) {
  613. eng := NewTestEngine(t)
  614. daemon := mkDaemonFromEngine(eng, t)
  615. defer nuke(daemon)
  616. for _, test := range []struct {
  617. Name string
  618. Valid bool
  619. }{
  620. {"abc-123_AAA.1", true},
  621. {"\000asdf", false},
  622. } {
  623. config, _, _, err := parseRun([]string{unitTestImageID, "echo test"})
  624. if err != nil {
  625. if !test.Valid {
  626. continue
  627. }
  628. t.Fatal(err)
  629. }
  630. containerId, _, err := daemon.ContainerCreate(test.Name, config, &runconfig.HostConfig{})
  631. if err != nil {
  632. if !test.Valid {
  633. continue
  634. }
  635. t.Fatal(err)
  636. }
  637. container, err := daemon.Get(containerId)
  638. if err != nil {
  639. t.Fatal(err)
  640. }
  641. if container.Name != "/"+test.Name {
  642. t.Fatalf("Expect /%s got %s", test.Name, container.Name)
  643. }
  644. if c, err := daemon.Get("/" + test.Name); err != nil {
  645. t.Fatalf("Couldn't retrieve test container as /%s", test.Name)
  646. } else if c.ID != container.ID {
  647. t.Fatalf("Container /%s has ID %s instead of %s", test.Name, c.ID, container.ID)
  648. }
  649. }
  650. }
  651. func TestLinkChildContainer(t *testing.T) {
  652. eng := NewTestEngine(t)
  653. daemon := mkDaemonFromEngine(eng, t)
  654. defer nuke(daemon)
  655. config, _, _, err := parseRun([]string{unitTestImageID, "echo test"})
  656. if err != nil {
  657. t.Fatal(err)
  658. }
  659. container, err := daemon.Get(createNamedTestContainer(eng, config, t, "/webapp"))
  660. if err != nil {
  661. t.Fatal(err)
  662. }
  663. webapp, err := daemon.GetByName("/webapp")
  664. if err != nil {
  665. t.Fatal(err)
  666. }
  667. if webapp.ID != container.ID {
  668. t.Fatalf("Expect webapp id to match container id: %s != %s", webapp.ID, container.ID)
  669. }
  670. config, _, _, err = parseRun([]string{GetTestImage(daemon).ID, "echo test"})
  671. if err != nil {
  672. t.Fatal(err)
  673. }
  674. childContainer, err := daemon.Get(createTestContainer(eng, config, t))
  675. if err != nil {
  676. t.Fatal(err)
  677. }
  678. if err := daemon.RegisterLink(webapp, childContainer, "db"); err != nil {
  679. t.Fatal(err)
  680. }
  681. // Get the child by it's new name
  682. db, err := daemon.GetByName("/webapp/db")
  683. if err != nil {
  684. t.Fatal(err)
  685. }
  686. if db.ID != childContainer.ID {
  687. t.Fatalf("Expect db id to match container id: %s != %s", db.ID, childContainer.ID)
  688. }
  689. }
  690. func TestGetAllChildren(t *testing.T) {
  691. eng := NewTestEngine(t)
  692. daemon := mkDaemonFromEngine(eng, t)
  693. defer nuke(daemon)
  694. config, _, _, err := parseRun([]string{unitTestImageID, "echo test"})
  695. if err != nil {
  696. t.Fatal(err)
  697. }
  698. container, err := daemon.Get(createNamedTestContainer(eng, config, t, "/webapp"))
  699. if err != nil {
  700. t.Fatal(err)
  701. }
  702. webapp, err := daemon.GetByName("/webapp")
  703. if err != nil {
  704. t.Fatal(err)
  705. }
  706. if webapp.ID != container.ID {
  707. t.Fatalf("Expect webapp id to match container id: %s != %s", webapp.ID, container.ID)
  708. }
  709. config, _, _, err = parseRun([]string{unitTestImageID, "echo test"})
  710. if err != nil {
  711. t.Fatal(err)
  712. }
  713. childContainer, err := daemon.Get(createTestContainer(eng, config, t))
  714. if err != nil {
  715. t.Fatal(err)
  716. }
  717. if err := daemon.RegisterLink(webapp, childContainer, "db"); err != nil {
  718. t.Fatal(err)
  719. }
  720. children, err := daemon.Children("/webapp")
  721. if err != nil {
  722. t.Fatal(err)
  723. }
  724. if children == nil {
  725. t.Fatal("Children should not be nil")
  726. }
  727. if len(children) == 0 {
  728. t.Fatal("Children should not be empty")
  729. }
  730. for key, value := range children {
  731. if key != "/webapp/db" {
  732. t.Fatalf("Expected /webapp/db got %s", key)
  733. }
  734. if value.ID != childContainer.ID {
  735. t.Fatalf("Expected id %s got %s", childContainer.ID, value.ID)
  736. }
  737. }
  738. }
  739. func TestDestroyWithInitLayer(t *testing.T) {
  740. daemon := mkDaemon(t)
  741. defer nuke(daemon)
  742. container, _, err := daemon.Create(&runconfig.Config{
  743. Image: GetTestImage(daemon).ID,
  744. Cmd: runconfig.NewCommand("ls", "-al"),
  745. },
  746. &runconfig.HostConfig{},
  747. "")
  748. if err != nil {
  749. t.Fatal(err)
  750. }
  751. // Destroy
  752. if err := daemon.Rm(container); err != nil {
  753. t.Fatal(err)
  754. }
  755. // Make sure daemon.Exists() behaves correctly
  756. if daemon.Exists("test_destroy") {
  757. t.Fatalf("Exists() returned true")
  758. }
  759. // Make sure daemon.List() doesn't list the destroyed container
  760. if len(daemon.List()) != 0 {
  761. t.Fatalf("Expected 0 container, %v found", len(daemon.List()))
  762. }
  763. driver := daemon.Graph().Driver()
  764. // Make sure that the container does not exist in the driver
  765. if _, err := driver.Get(container.ID, ""); err == nil {
  766. t.Fatal("Conttainer should not exist in the driver")
  767. }
  768. // Make sure that the init layer is removed from the driver
  769. if _, err := driver.Get(fmt.Sprintf("%s-init", container.ID), ""); err == nil {
  770. t.Fatal("Container's init layer should not exist in the driver")
  771. }
  772. }