api_test.go 32 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264
  1. package docker
  2. import (
  3. "archive/tar"
  4. "bufio"
  5. "bytes"
  6. "encoding/json"
  7. "fmt"
  8. "github.com/dotcloud/docker"
  9. "github.com/dotcloud/docker/engine"
  10. "github.com/dotcloud/docker/utils"
  11. "io"
  12. "net"
  13. "net/http"
  14. "net/http/httptest"
  15. "strings"
  16. "testing"
  17. "time"
  18. )
  19. func TestGetVersion(t *testing.T) {
  20. eng := NewTestEngine(t)
  21. defer mkRuntimeFromEngine(eng, t).Nuke()
  22. srv := mkServerFromEngine(eng, t)
  23. var err error
  24. r := httptest.NewRecorder()
  25. req, err := http.NewRequest("GET", "/version", nil)
  26. if err != nil {
  27. t.Fatal(err)
  28. }
  29. // FIXME getting the version should require an actual running Server
  30. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  31. t.Fatal(err)
  32. }
  33. assertHttpNotError(r, t)
  34. out := engine.NewOutput()
  35. v, err := out.AddEnv()
  36. if err != nil {
  37. t.Fatal(err)
  38. }
  39. if _, err := io.Copy(out, r.Body); err != nil {
  40. t.Fatal(err)
  41. }
  42. out.Close()
  43. expected := docker.VERSION
  44. if result := v.Get("Version"); result != expected {
  45. t.Errorf("Expected version %s, %s found", expected, result)
  46. }
  47. }
  48. func TestGetInfo(t *testing.T) {
  49. eng := NewTestEngine(t)
  50. defer mkRuntimeFromEngine(eng, t).Nuke()
  51. srv := mkServerFromEngine(eng, t)
  52. initialImages, err := srv.Images(false, "")
  53. if err != nil {
  54. t.Fatal(err)
  55. }
  56. req, err := http.NewRequest("GET", "/info", nil)
  57. if err != nil {
  58. t.Fatal(err)
  59. }
  60. r := httptest.NewRecorder()
  61. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  62. t.Fatal(err)
  63. }
  64. assertHttpNotError(r, t)
  65. out := engine.NewOutput()
  66. i, err := out.AddEnv()
  67. if err != nil {
  68. t.Fatal(err)
  69. }
  70. if _, err := io.Copy(out, r.Body); err != nil {
  71. t.Fatal(err)
  72. }
  73. out.Close()
  74. if images := i.GetInt("Images"); images != int64(len(initialImages)) {
  75. t.Errorf("Expected images: %d, %d found", len(initialImages), images)
  76. }
  77. }
  78. func TestGetEvents(t *testing.T) {
  79. eng := NewTestEngine(t)
  80. srv := mkServerFromEngine(eng, t)
  81. // FIXME: we might not need runtime, why not simply nuke
  82. // the engine?
  83. runtime := mkRuntimeFromEngine(eng, t)
  84. defer nuke(runtime)
  85. var events []*utils.JSONMessage
  86. for _, parts := range [][3]string{
  87. {"fakeaction", "fakeid", "fakeimage"},
  88. {"fakeaction2", "fakeid", "fakeimage"},
  89. } {
  90. action, id, from := parts[0], parts[1], parts[2]
  91. ev := srv.LogEvent(action, id, from)
  92. events = append(events, ev)
  93. }
  94. req, err := http.NewRequest("GET", "/events?since=1", nil)
  95. if err != nil {
  96. t.Fatal(err)
  97. }
  98. r := httptest.NewRecorder()
  99. setTimeout(t, "", 500*time.Millisecond, func() {
  100. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  101. t.Fatal(err)
  102. }
  103. assertHttpNotError(r, t)
  104. })
  105. dec := json.NewDecoder(r.Body)
  106. for i := 0; i < 2; i++ {
  107. var jm utils.JSONMessage
  108. if err := dec.Decode(&jm); err == io.EOF {
  109. break
  110. } else if err != nil {
  111. t.Fatal(err)
  112. }
  113. if jm != *events[i] {
  114. t.Fatalf("Event received it different than expected")
  115. }
  116. }
  117. }
  118. func TestGetImagesJSON(t *testing.T) {
  119. eng := NewTestEngine(t)
  120. defer mkRuntimeFromEngine(eng, t).Nuke()
  121. srv := mkServerFromEngine(eng, t)
  122. // all=0
  123. initialImages, err := srv.Images(false, "")
  124. if err != nil {
  125. t.Fatal(err)
  126. }
  127. req, err := http.NewRequest("GET", "/images/json?all=0", nil)
  128. if err != nil {
  129. t.Fatal(err)
  130. }
  131. r := httptest.NewRecorder()
  132. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  133. t.Fatal(err)
  134. }
  135. assertHttpNotError(r, t)
  136. images := []docker.APIImages{}
  137. if err := json.Unmarshal(r.Body.Bytes(), &images); err != nil {
  138. t.Fatal(err)
  139. }
  140. if len(images) != len(initialImages) {
  141. t.Errorf("Expected %d image, %d found", len(initialImages), len(images))
  142. }
  143. found := false
  144. for _, img := range images {
  145. if strings.Contains(img.RepoTags[0], unitTestImageName) {
  146. found = true
  147. break
  148. }
  149. }
  150. if !found {
  151. t.Errorf("Expected image %s, %+v found", unitTestImageName, images)
  152. }
  153. r2 := httptest.NewRecorder()
  154. // all=1
  155. initialImages, err = srv.Images(true, "")
  156. if err != nil {
  157. t.Fatal(err)
  158. }
  159. req2, err := http.NewRequest("GET", "/images/json?all=true", nil)
  160. if err != nil {
  161. t.Fatal(err)
  162. }
  163. if err := docker.ServeRequest(srv, docker.APIVERSION, r2, req2); err != nil {
  164. t.Fatal(err)
  165. }
  166. assertHttpNotError(r2, t)
  167. images2 := []docker.APIImages{}
  168. if err := json.Unmarshal(r2.Body.Bytes(), &images2); err != nil {
  169. t.Fatal(err)
  170. }
  171. if len(images2) != len(initialImages) {
  172. t.Errorf("Expected %d image, %d found", len(initialImages), len(images2))
  173. }
  174. found = false
  175. for _, img := range images2 {
  176. if img.ID == unitTestImageID {
  177. found = true
  178. break
  179. }
  180. }
  181. if !found {
  182. t.Errorf("Retrieved image Id differs, expected %s, received %+v", unitTestImageID, images2)
  183. }
  184. r3 := httptest.NewRecorder()
  185. // filter=a
  186. req3, err := http.NewRequest("GET", "/images/json?filter=aaaaaaaaaa", nil)
  187. if err != nil {
  188. t.Fatal(err)
  189. }
  190. if err := docker.ServeRequest(srv, docker.APIVERSION, r3, req3); err != nil {
  191. t.Fatal(err)
  192. }
  193. assertHttpNotError(r3, t)
  194. images3 := []docker.APIImages{}
  195. if err := json.Unmarshal(r3.Body.Bytes(), &images3); err != nil {
  196. t.Fatal(err)
  197. }
  198. if len(images3) != 0 {
  199. t.Errorf("Expected 0 image, %d found", len(images3))
  200. }
  201. r4 := httptest.NewRecorder()
  202. // all=foobar
  203. req4, err := http.NewRequest("GET", "/images/json?all=foobar", nil)
  204. if err != nil {
  205. t.Fatal(err)
  206. }
  207. if err := docker.ServeRequest(srv, docker.APIVERSION, r4, req4); err != nil {
  208. t.Fatal(err)
  209. }
  210. // Don't assert against HTTP error since we expect an error
  211. if r4.Code != http.StatusBadRequest {
  212. t.Fatalf("%d Bad Request expected, received %d\n", http.StatusBadRequest, r4.Code)
  213. }
  214. }
  215. func TestGetImagesHistory(t *testing.T) {
  216. eng := NewTestEngine(t)
  217. defer mkRuntimeFromEngine(eng, t).Nuke()
  218. srv := mkServerFromEngine(eng, t)
  219. r := httptest.NewRecorder()
  220. req, err := http.NewRequest("GET", fmt.Sprintf("/images/%s/history", unitTestImageName), nil)
  221. if err != nil {
  222. t.Fatal(err)
  223. }
  224. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  225. t.Fatal(err)
  226. }
  227. assertHttpNotError(r, t)
  228. history := []docker.APIHistory{}
  229. if err := json.Unmarshal(r.Body.Bytes(), &history); err != nil {
  230. t.Fatal(err)
  231. }
  232. if len(history) != 1 {
  233. t.Errorf("Expected 1 line, %d found", len(history))
  234. }
  235. }
  236. func TestGetImagesByName(t *testing.T) {
  237. eng := NewTestEngine(t)
  238. defer mkRuntimeFromEngine(eng, t).Nuke()
  239. srv := mkServerFromEngine(eng, t)
  240. req, err := http.NewRequest("GET", "/images/"+unitTestImageName+"/json", nil)
  241. if err != nil {
  242. t.Fatal(err)
  243. }
  244. r := httptest.NewRecorder()
  245. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  246. t.Fatal(err)
  247. }
  248. assertHttpNotError(r, t)
  249. img := &docker.Image{}
  250. if err := json.Unmarshal(r.Body.Bytes(), img); err != nil {
  251. t.Fatal(err)
  252. }
  253. if img.ID != unitTestImageID {
  254. t.Errorf("Error inspecting image")
  255. }
  256. }
  257. func TestGetContainersJSON(t *testing.T) {
  258. eng := NewTestEngine(t)
  259. defer mkRuntimeFromEngine(eng, t).Nuke()
  260. srv := mkServerFromEngine(eng, t)
  261. beginLen := len(srv.Containers(true, false, -1, "", ""))
  262. containerID := createTestContainer(eng, &docker.Config{
  263. Image: unitTestImageID,
  264. Cmd: []string{"echo", "test"},
  265. }, t)
  266. if containerID == "" {
  267. t.Fatalf("Received empty container ID")
  268. }
  269. req, err := http.NewRequest("GET", "/containers/json?all=1", nil)
  270. if err != nil {
  271. t.Fatal(err)
  272. }
  273. r := httptest.NewRecorder()
  274. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  275. t.Fatal(err)
  276. }
  277. assertHttpNotError(r, t)
  278. containers := []docker.APIContainers{}
  279. if err := json.Unmarshal(r.Body.Bytes(), &containers); err != nil {
  280. t.Fatal(err)
  281. }
  282. if len(containers) != beginLen+1 {
  283. t.Fatalf("Expected %d container, %d found (started with: %d)", beginLen+1, len(containers), beginLen)
  284. }
  285. if containers[0].ID != containerID {
  286. t.Fatalf("Container ID mismatch. Expected: %s, received: %s\n", containerID, containers[0].ID)
  287. }
  288. }
  289. func TestGetContainersExport(t *testing.T) {
  290. eng := NewTestEngine(t)
  291. defer mkRuntimeFromEngine(eng, t).Nuke()
  292. srv := mkServerFromEngine(eng, t)
  293. // Create a container and remove a file
  294. containerID := createTestContainer(eng,
  295. &docker.Config{
  296. Image: unitTestImageID,
  297. Cmd: []string{"touch", "/test"},
  298. },
  299. t,
  300. )
  301. containerRun(eng, containerID, t)
  302. r := httptest.NewRecorder()
  303. req, err := http.NewRequest("GET", "/containers/"+containerID+"/export", nil)
  304. if err != nil {
  305. t.Fatal(err)
  306. }
  307. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  308. t.Fatal(err)
  309. }
  310. assertHttpNotError(r, t)
  311. if r.Code != http.StatusOK {
  312. t.Fatalf("%d OK expected, received %d\n", http.StatusOK, r.Code)
  313. }
  314. found := false
  315. for tarReader := tar.NewReader(r.Body); ; {
  316. h, err := tarReader.Next()
  317. if err != nil {
  318. if err == io.EOF {
  319. break
  320. }
  321. t.Fatal(err)
  322. }
  323. if h.Name == "./test" {
  324. found = true
  325. break
  326. }
  327. }
  328. if !found {
  329. t.Fatalf("The created test file has not been found in the exported image")
  330. }
  331. }
  332. func TestGetContainersChanges(t *testing.T) {
  333. eng := NewTestEngine(t)
  334. defer mkRuntimeFromEngine(eng, t).Nuke()
  335. srv := mkServerFromEngine(eng, t)
  336. // Create a container and remove a file
  337. containerID := createTestContainer(eng,
  338. &docker.Config{
  339. Image: unitTestImageID,
  340. Cmd: []string{"/bin/rm", "/etc/passwd"},
  341. },
  342. t,
  343. )
  344. containerRun(eng, containerID, t)
  345. r := httptest.NewRecorder()
  346. req, err := http.NewRequest("GET", "/containers/"+containerID+"/changes", nil)
  347. if err != nil {
  348. t.Fatal(err)
  349. }
  350. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  351. t.Fatal(err)
  352. }
  353. assertHttpNotError(r, t)
  354. changes := []docker.Change{}
  355. if err := json.Unmarshal(r.Body.Bytes(), &changes); err != nil {
  356. t.Fatal(err)
  357. }
  358. // Check the changelog
  359. success := false
  360. for _, elem := range changes {
  361. if elem.Path == "/etc/passwd" && elem.Kind == 2 {
  362. success = true
  363. }
  364. }
  365. if !success {
  366. t.Fatalf("/etc/passwd as been removed but is not present in the diff")
  367. }
  368. }
  369. func TestGetContainersTop(t *testing.T) {
  370. t.Skip("Fixme. Skipping test for now. Reported error when testing using dind: 'api_test.go:527: Expected 2 processes, found 0.'")
  371. eng := NewTestEngine(t)
  372. defer mkRuntimeFromEngine(eng, t).Nuke()
  373. srv := mkServerFromEngine(eng, t)
  374. containerID := createTestContainer(eng,
  375. &docker.Config{
  376. Image: unitTestImageID,
  377. Cmd: []string{"/bin/sh", "-c", "cat"},
  378. OpenStdin: true,
  379. },
  380. t,
  381. )
  382. defer func() {
  383. // Make sure the process dies before destroying runtime
  384. containerKill(eng, containerID, t)
  385. containerWait(eng, containerID, t)
  386. }()
  387. startContainer(eng, containerID, t)
  388. setTimeout(t, "Waiting for the container to be started timed out", 10*time.Second, func() {
  389. for {
  390. if containerRunning(eng, containerID, t) {
  391. break
  392. }
  393. time.Sleep(10 * time.Millisecond)
  394. }
  395. })
  396. if !containerRunning(eng, containerID, t) {
  397. t.Fatalf("Container should be running")
  398. }
  399. // Make sure sh spawn up cat
  400. setTimeout(t, "read/write assertion timed out", 2*time.Second, func() {
  401. in, out := containerAttach(eng, containerID, t)
  402. if err := assertPipe("hello\n", "hello", out, in, 150); err != nil {
  403. t.Fatal(err)
  404. }
  405. })
  406. r := httptest.NewRecorder()
  407. req, err := http.NewRequest("GET", "/"+containerID+"/top?ps_args=u", bytes.NewReader([]byte{}))
  408. if err != nil {
  409. t.Fatal(err)
  410. }
  411. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  412. t.Fatal(err)
  413. }
  414. assertHttpNotError(r, t)
  415. procs := docker.APITop{}
  416. if err := json.Unmarshal(r.Body.Bytes(), &procs); err != nil {
  417. t.Fatal(err)
  418. }
  419. if len(procs.Titles) != 11 {
  420. t.Fatalf("Expected 11 titles, found %d.", len(procs.Titles))
  421. }
  422. if procs.Titles[0] != "USER" || procs.Titles[10] != "COMMAND" {
  423. t.Fatalf("Expected Titles[0] to be USER and Titles[10] to be COMMAND, found %s and %s.", procs.Titles[0], procs.Titles[10])
  424. }
  425. if len(procs.Processes) != 2 {
  426. t.Fatalf("Expected 2 processes, found %d.", len(procs.Processes))
  427. }
  428. if procs.Processes[0][10] != "/bin/sh" && procs.Processes[0][10] != "cat" {
  429. t.Fatalf("Expected `cat` or `/bin/sh`, found %s.", procs.Processes[0][10])
  430. }
  431. if procs.Processes[1][10] != "/bin/sh" && procs.Processes[1][10] != "cat" {
  432. t.Fatalf("Expected `cat` or `/bin/sh`, found %s.", procs.Processes[1][10])
  433. }
  434. }
  435. func TestGetContainersByName(t *testing.T) {
  436. eng := NewTestEngine(t)
  437. defer mkRuntimeFromEngine(eng, t).Nuke()
  438. srv := mkServerFromEngine(eng, t)
  439. // Create a container and remove a file
  440. containerID := createTestContainer(eng,
  441. &docker.Config{
  442. Image: unitTestImageID,
  443. Cmd: []string{"echo", "test"},
  444. },
  445. t,
  446. )
  447. r := httptest.NewRecorder()
  448. req, err := http.NewRequest("GET", "/containers/"+containerID+"/json", nil)
  449. if err != nil {
  450. t.Fatal(err)
  451. }
  452. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  453. t.Fatal(err)
  454. }
  455. assertHttpNotError(r, t)
  456. outContainer := &docker.Container{}
  457. if err := json.Unmarshal(r.Body.Bytes(), outContainer); err != nil {
  458. t.Fatal(err)
  459. }
  460. if outContainer.ID != containerID {
  461. t.Fatalf("Wrong containers retrieved. Expected %s, received %s", containerID, outContainer.ID)
  462. }
  463. }
  464. func TestPostCommit(t *testing.T) {
  465. eng := NewTestEngine(t)
  466. defer mkRuntimeFromEngine(eng, t).Nuke()
  467. srv := mkServerFromEngine(eng, t)
  468. // Create a container and remove a file
  469. containerID := createTestContainer(eng,
  470. &docker.Config{
  471. Image: unitTestImageID,
  472. Cmd: []string{"touch", "/test"},
  473. },
  474. t,
  475. )
  476. containerRun(eng, containerID, t)
  477. req, err := http.NewRequest("POST", "/commit?repo=testrepo&testtag=tag&container="+containerID, bytes.NewReader([]byte{}))
  478. if err != nil {
  479. t.Fatal(err)
  480. }
  481. r := httptest.NewRecorder()
  482. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  483. t.Fatal(err)
  484. }
  485. assertHttpNotError(r, t)
  486. if r.Code != http.StatusCreated {
  487. t.Fatalf("%d Created expected, received %d\n", http.StatusCreated, r.Code)
  488. }
  489. apiID := &docker.APIID{}
  490. if err := json.Unmarshal(r.Body.Bytes(), apiID); err != nil {
  491. t.Fatal(err)
  492. }
  493. if _, err := srv.ImageInspect(apiID.ID); err != nil {
  494. t.Fatalf("The image has not been committed")
  495. }
  496. }
  497. func TestPostContainersCreate(t *testing.T) {
  498. eng := NewTestEngine(t)
  499. defer mkRuntimeFromEngine(eng, t).Nuke()
  500. srv := mkServerFromEngine(eng, t)
  501. configJSON, err := json.Marshal(&docker.Config{
  502. Image: unitTestImageID,
  503. Memory: 33554432,
  504. Cmd: []string{"touch", "/test"},
  505. })
  506. if err != nil {
  507. t.Fatal(err)
  508. }
  509. req, err := http.NewRequest("POST", "/containers/create", bytes.NewReader(configJSON))
  510. if err != nil {
  511. t.Fatal(err)
  512. }
  513. r := httptest.NewRecorder()
  514. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  515. t.Fatal(err)
  516. }
  517. assertHttpNotError(r, t)
  518. if r.Code != http.StatusCreated {
  519. t.Fatalf("%d Created expected, received %d\n", http.StatusCreated, r.Code)
  520. }
  521. apiRun := &docker.APIRun{}
  522. if err := json.Unmarshal(r.Body.Bytes(), apiRun); err != nil {
  523. t.Fatal(err)
  524. }
  525. containerID := apiRun.ID
  526. containerAssertExists(eng, containerID, t)
  527. containerRun(eng, containerID, t)
  528. if !containerFileExists(eng, containerID, "test", t) {
  529. t.Fatal("Test file was not created")
  530. }
  531. }
  532. func TestPostContainersKill(t *testing.T) {
  533. eng := NewTestEngine(t)
  534. defer mkRuntimeFromEngine(eng, t).Nuke()
  535. srv := mkServerFromEngine(eng, t)
  536. containerID := createTestContainer(eng,
  537. &docker.Config{
  538. Image: unitTestImageID,
  539. Cmd: []string{"/bin/cat"},
  540. OpenStdin: true,
  541. },
  542. t,
  543. )
  544. startContainer(eng, containerID, t)
  545. // Give some time to the process to start
  546. containerWaitTimeout(eng, containerID, t)
  547. if !containerRunning(eng, containerID, t) {
  548. t.Errorf("Container should be running")
  549. }
  550. r := httptest.NewRecorder()
  551. req, err := http.NewRequest("POST", "/containers/"+containerID+"/kill", bytes.NewReader([]byte{}))
  552. if err != nil {
  553. t.Fatal(err)
  554. }
  555. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  556. t.Fatal(err)
  557. }
  558. assertHttpNotError(r, t)
  559. if r.Code != http.StatusNoContent {
  560. t.Fatalf("%d NO CONTENT expected, received %d\n", http.StatusNoContent, r.Code)
  561. }
  562. if containerRunning(eng, containerID, t) {
  563. t.Fatalf("The container hasn't been killed")
  564. }
  565. }
  566. func TestPostContainersRestart(t *testing.T) {
  567. eng := NewTestEngine(t)
  568. defer mkRuntimeFromEngine(eng, t).Nuke()
  569. srv := mkServerFromEngine(eng, t)
  570. containerID := createTestContainer(eng,
  571. &docker.Config{
  572. Image: unitTestImageID,
  573. Cmd: []string{"/bin/top"},
  574. OpenStdin: true,
  575. },
  576. t,
  577. )
  578. startContainer(eng, containerID, t)
  579. // Give some time to the process to start
  580. containerWaitTimeout(eng, containerID, t)
  581. if !containerRunning(eng, containerID, t) {
  582. t.Errorf("Container should be running")
  583. }
  584. req, err := http.NewRequest("POST", "/containers/"+containerID+"/restart?t=1", bytes.NewReader([]byte{}))
  585. if err != nil {
  586. t.Fatal(err)
  587. }
  588. r := httptest.NewRecorder()
  589. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  590. t.Fatal(err)
  591. }
  592. assertHttpNotError(r, t)
  593. if r.Code != http.StatusNoContent {
  594. t.Fatalf("%d NO CONTENT expected, received %d\n", http.StatusNoContent, r.Code)
  595. }
  596. // Give some time to the process to restart
  597. containerWaitTimeout(eng, containerID, t)
  598. if !containerRunning(eng, containerID, t) {
  599. t.Fatalf("Container should be running")
  600. }
  601. containerKill(eng, containerID, t)
  602. }
  603. func TestPostContainersStart(t *testing.T) {
  604. eng := NewTestEngine(t)
  605. defer mkRuntimeFromEngine(eng, t).Nuke()
  606. srv := mkServerFromEngine(eng, t)
  607. containerID := createTestContainer(
  608. eng,
  609. &docker.Config{
  610. Image: unitTestImageID,
  611. Cmd: []string{"/bin/cat"},
  612. OpenStdin: true,
  613. },
  614. t,
  615. )
  616. hostConfigJSON, err := json.Marshal(&docker.HostConfig{})
  617. req, err := http.NewRequest("POST", "/containers/"+containerID+"/start", bytes.NewReader(hostConfigJSON))
  618. if err != nil {
  619. t.Fatal(err)
  620. }
  621. req.Header.Set("Content-Type", "application/json")
  622. r := httptest.NewRecorder()
  623. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  624. t.Fatal(err)
  625. }
  626. assertHttpNotError(r, t)
  627. if r.Code != http.StatusNoContent {
  628. t.Fatalf("%d NO CONTENT expected, received %d\n", http.StatusNoContent, r.Code)
  629. }
  630. containerAssertExists(eng, containerID, t)
  631. // Give some time to the process to start
  632. // FIXME: use Wait once it's available as a job
  633. containerWaitTimeout(eng, containerID, t)
  634. if !containerRunning(eng, containerID, t) {
  635. t.Errorf("Container should be running")
  636. }
  637. r = httptest.NewRecorder()
  638. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  639. t.Fatal(err)
  640. }
  641. // Starting an already started container should return an error
  642. // FIXME: verify a precise error code. There is a possible bug here
  643. // which causes this to return 404 even though the container exists.
  644. assertHttpError(r, t)
  645. containerAssertExists(eng, containerID, t)
  646. containerKill(eng, containerID, t)
  647. }
  648. // Expected behaviour: using / as a bind mount source should throw an error
  649. func TestRunErrorBindMountRootSource(t *testing.T) {
  650. eng := NewTestEngine(t)
  651. defer mkRuntimeFromEngine(eng, t).Nuke()
  652. srv := mkServerFromEngine(eng, t)
  653. containerID := createTestContainer(
  654. eng,
  655. &docker.Config{
  656. Image: unitTestImageID,
  657. Cmd: []string{"/bin/cat"},
  658. OpenStdin: true,
  659. },
  660. t,
  661. )
  662. hostConfigJSON, err := json.Marshal(&docker.HostConfig{
  663. Binds: []string{"/:/tmp"},
  664. })
  665. req, err := http.NewRequest("POST", "/containers/"+containerID+"/start", bytes.NewReader(hostConfigJSON))
  666. if err != nil {
  667. t.Fatal(err)
  668. }
  669. req.Header.Set("Content-Type", "application/json")
  670. r := httptest.NewRecorder()
  671. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  672. t.Fatal(err)
  673. }
  674. if r.Code != http.StatusInternalServerError {
  675. containerKill(eng, containerID, t)
  676. t.Fatal("should have failed to run when using / as a source for the bind mount")
  677. }
  678. }
  679. func TestPostContainersStop(t *testing.T) {
  680. eng := NewTestEngine(t)
  681. defer mkRuntimeFromEngine(eng, t).Nuke()
  682. srv := mkServerFromEngine(eng, t)
  683. containerID := createTestContainer(eng,
  684. &docker.Config{
  685. Image: unitTestImageID,
  686. Cmd: []string{"/bin/top"},
  687. OpenStdin: true,
  688. },
  689. t,
  690. )
  691. startContainer(eng, containerID, t)
  692. // Give some time to the process to start
  693. containerWaitTimeout(eng, containerID, t)
  694. if !containerRunning(eng, containerID, t) {
  695. t.Errorf("Container should be running")
  696. }
  697. // Note: as it is a POST request, it requires a body.
  698. req, err := http.NewRequest("POST", "/containers/"+containerID+"/stop?t=1", bytes.NewReader([]byte{}))
  699. if err != nil {
  700. t.Fatal(err)
  701. }
  702. r := httptest.NewRecorder()
  703. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  704. t.Fatal(err)
  705. }
  706. assertHttpNotError(r, t)
  707. if r.Code != http.StatusNoContent {
  708. t.Fatalf("%d NO CONTENT expected, received %d\n", http.StatusNoContent, r.Code)
  709. }
  710. if containerRunning(eng, containerID, t) {
  711. t.Fatalf("The container hasn't been stopped")
  712. }
  713. }
  714. func TestPostContainersWait(t *testing.T) {
  715. eng := NewTestEngine(t)
  716. defer mkRuntimeFromEngine(eng, t).Nuke()
  717. srv := mkServerFromEngine(eng, t)
  718. containerID := createTestContainer(eng,
  719. &docker.Config{
  720. Image: unitTestImageID,
  721. Cmd: []string{"/bin/sleep", "1"},
  722. OpenStdin: true,
  723. },
  724. t,
  725. )
  726. startContainer(eng, containerID, t)
  727. setTimeout(t, "Wait timed out", 3*time.Second, func() {
  728. r := httptest.NewRecorder()
  729. req, err := http.NewRequest("POST", "/containers/"+containerID+"/wait", bytes.NewReader([]byte{}))
  730. if err != nil {
  731. t.Fatal(err)
  732. }
  733. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  734. t.Fatal(err)
  735. }
  736. assertHttpNotError(r, t)
  737. apiWait := &docker.APIWait{}
  738. if err := json.Unmarshal(r.Body.Bytes(), apiWait); err != nil {
  739. t.Fatal(err)
  740. }
  741. if apiWait.StatusCode != 0 {
  742. t.Fatalf("Non zero exit code for sleep: %d\n", apiWait.StatusCode)
  743. }
  744. })
  745. if containerRunning(eng, containerID, t) {
  746. t.Fatalf("The container should be stopped after wait")
  747. }
  748. }
  749. func TestPostContainersAttach(t *testing.T) {
  750. eng := NewTestEngine(t)
  751. defer mkRuntimeFromEngine(eng, t).Nuke()
  752. srv := mkServerFromEngine(eng, t)
  753. containerID := createTestContainer(eng,
  754. &docker.Config{
  755. Image: unitTestImageID,
  756. Cmd: []string{"/bin/cat"},
  757. OpenStdin: true,
  758. },
  759. t,
  760. )
  761. // Start the process
  762. startContainer(eng, containerID, t)
  763. stdin, stdinPipe := io.Pipe()
  764. stdout, stdoutPipe := io.Pipe()
  765. // Try to avoid the timeout in destroy. Best effort, don't check error
  766. defer func() {
  767. closeWrap(stdin, stdinPipe, stdout, stdoutPipe)
  768. containerKill(eng, containerID, t)
  769. }()
  770. // Attach to it
  771. c1 := make(chan struct{})
  772. go func() {
  773. defer close(c1)
  774. r := &hijackTester{
  775. ResponseRecorder: httptest.NewRecorder(),
  776. in: stdin,
  777. out: stdoutPipe,
  778. }
  779. req, err := http.NewRequest("POST", "/containers/"+containerID+"/attach?stream=1&stdin=1&stdout=1&stderr=1", bytes.NewReader([]byte{}))
  780. if err != nil {
  781. t.Fatal(err)
  782. }
  783. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  784. t.Fatal(err)
  785. }
  786. assertHttpNotError(r.ResponseRecorder, t)
  787. }()
  788. // Acknowledge hijack
  789. setTimeout(t, "hijack acknowledge timed out", 2*time.Second, func() {
  790. stdout.Read([]byte{})
  791. stdout.Read(make([]byte, 4096))
  792. })
  793. setTimeout(t, "read/write assertion timed out", 2*time.Second, func() {
  794. if err := assertPipe("hello\n", string([]byte{1, 0, 0, 0, 0, 0, 0, 6})+"hello", stdout, stdinPipe, 150); err != nil {
  795. t.Fatal(err)
  796. }
  797. })
  798. // Close pipes (client disconnects)
  799. if err := closeWrap(stdin, stdinPipe, stdout, stdoutPipe); err != nil {
  800. t.Fatal(err)
  801. }
  802. // Wait for attach to finish, the client disconnected, therefore, Attach finished his job
  803. setTimeout(t, "Waiting for CmdAttach timed out", 10*time.Second, func() {
  804. <-c1
  805. })
  806. // We closed stdin, expect /bin/cat to still be running
  807. // Wait a little bit to make sure container.monitor() did his thing
  808. containerWaitTimeout(eng, containerID, t)
  809. // Try to avoid the timeout in destroy. Best effort, don't check error
  810. cStdin, _ := containerAttach(eng, containerID, t)
  811. cStdin.Close()
  812. containerWait(eng, containerID, t)
  813. }
  814. func TestPostContainersAttachStderr(t *testing.T) {
  815. eng := NewTestEngine(t)
  816. defer mkRuntimeFromEngine(eng, t).Nuke()
  817. srv := mkServerFromEngine(eng, t)
  818. containerID := createTestContainer(eng,
  819. &docker.Config{
  820. Image: unitTestImageID,
  821. Cmd: []string{"/bin/sh", "-c", "/bin/cat >&2"},
  822. OpenStdin: true,
  823. },
  824. t,
  825. )
  826. // Start the process
  827. startContainer(eng, containerID, t)
  828. stdin, stdinPipe := io.Pipe()
  829. stdout, stdoutPipe := io.Pipe()
  830. // Try to avoid the timeout in destroy. Best effort, don't check error
  831. defer func() {
  832. closeWrap(stdin, stdinPipe, stdout, stdoutPipe)
  833. containerKill(eng, containerID, t)
  834. }()
  835. // Attach to it
  836. c1 := make(chan struct{})
  837. go func() {
  838. defer close(c1)
  839. r := &hijackTester{
  840. ResponseRecorder: httptest.NewRecorder(),
  841. in: stdin,
  842. out: stdoutPipe,
  843. }
  844. req, err := http.NewRequest("POST", "/containers/"+containerID+"/attach?stream=1&stdin=1&stdout=1&stderr=1", bytes.NewReader([]byte{}))
  845. if err != nil {
  846. t.Fatal(err)
  847. }
  848. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  849. t.Fatal(err)
  850. }
  851. assertHttpNotError(r.ResponseRecorder, t)
  852. }()
  853. // Acknowledge hijack
  854. setTimeout(t, "hijack acknowledge timed out", 2*time.Second, func() {
  855. stdout.Read([]byte{})
  856. stdout.Read(make([]byte, 4096))
  857. })
  858. setTimeout(t, "read/write assertion timed out", 2*time.Second, func() {
  859. if err := assertPipe("hello\n", string([]byte{2, 0, 0, 0, 0, 0, 0, 6})+"hello", stdout, stdinPipe, 150); err != nil {
  860. t.Fatal(err)
  861. }
  862. })
  863. // Close pipes (client disconnects)
  864. if err := closeWrap(stdin, stdinPipe, stdout, stdoutPipe); err != nil {
  865. t.Fatal(err)
  866. }
  867. // Wait for attach to finish, the client disconnected, therefore, Attach finished his job
  868. setTimeout(t, "Waiting for CmdAttach timed out", 10*time.Second, func() {
  869. <-c1
  870. })
  871. // We closed stdin, expect /bin/cat to still be running
  872. // Wait a little bit to make sure container.monitor() did his thing
  873. containerWaitTimeout(eng, containerID, t)
  874. // Try to avoid the timeout in destroy. Best effort, don't check error
  875. cStdin, _ := containerAttach(eng, containerID, t)
  876. cStdin.Close()
  877. containerWait(eng, containerID, t)
  878. }
  879. // FIXME: Test deleting running container
  880. // FIXME: Test deleting container with volume
  881. // FIXME: Test deleting volume in use by other container
  882. func TestDeleteContainers(t *testing.T) {
  883. eng := NewTestEngine(t)
  884. defer mkRuntimeFromEngine(eng, t).Nuke()
  885. srv := mkServerFromEngine(eng, t)
  886. containerID := createTestContainer(eng,
  887. &docker.Config{
  888. Image: unitTestImageID,
  889. Cmd: []string{"touch", "/test"},
  890. },
  891. t,
  892. )
  893. req, err := http.NewRequest("DELETE", "/containers/"+containerID, nil)
  894. if err != nil {
  895. t.Fatal(err)
  896. }
  897. r := httptest.NewRecorder()
  898. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  899. t.Fatal(err)
  900. }
  901. assertHttpNotError(r, t)
  902. if r.Code != http.StatusNoContent {
  903. t.Fatalf("%d NO CONTENT expected, received %d\n", http.StatusNoContent, r.Code)
  904. }
  905. containerAssertNotExists(eng, containerID, t)
  906. }
  907. func TestOptionsRoute(t *testing.T) {
  908. eng := NewTestEngine(t)
  909. defer mkRuntimeFromEngine(eng, t).Nuke()
  910. srv := mkServerFromEngine(eng, t)
  911. r := httptest.NewRecorder()
  912. req, err := http.NewRequest("OPTIONS", "/", nil)
  913. if err != nil {
  914. t.Fatal(err)
  915. }
  916. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  917. t.Fatal(err)
  918. }
  919. assertHttpNotError(r, t)
  920. if r.Code != http.StatusOK {
  921. t.Errorf("Expected response for OPTIONS request to be \"200\", %v found.", r.Code)
  922. }
  923. }
  924. func TestGetEnabledCors(t *testing.T) {
  925. eng := NewTestEngine(t)
  926. defer mkRuntimeFromEngine(eng, t).Nuke()
  927. srv := mkServerFromEngine(eng, t)
  928. r := httptest.NewRecorder()
  929. req, err := http.NewRequest("GET", "/version", nil)
  930. if err != nil {
  931. t.Fatal(err)
  932. }
  933. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  934. t.Fatal(err)
  935. }
  936. assertHttpNotError(r, t)
  937. if r.Code != http.StatusOK {
  938. t.Errorf("Expected response for OPTIONS request to be \"200\", %v found.", r.Code)
  939. }
  940. allowOrigin := r.Header().Get("Access-Control-Allow-Origin")
  941. allowHeaders := r.Header().Get("Access-Control-Allow-Headers")
  942. allowMethods := r.Header().Get("Access-Control-Allow-Methods")
  943. if allowOrigin != "*" {
  944. t.Errorf("Expected header Access-Control-Allow-Origin to be \"*\", %s found.", allowOrigin)
  945. }
  946. if allowHeaders != "Origin, X-Requested-With, Content-Type, Accept" {
  947. t.Errorf("Expected header Access-Control-Allow-Headers to be \"Origin, X-Requested-With, Content-Type, Accept\", %s found.", allowHeaders)
  948. }
  949. if allowMethods != "GET, POST, DELETE, PUT, OPTIONS" {
  950. t.Errorf("Expected hearder Access-Control-Allow-Methods to be \"GET, POST, DELETE, PUT, OPTIONS\", %s found.", allowMethods)
  951. }
  952. }
  953. func TestDeleteImages(t *testing.T) {
  954. eng := NewTestEngine(t)
  955. defer mkRuntimeFromEngine(eng, t).Nuke()
  956. srv := mkServerFromEngine(eng, t)
  957. initialImages, err := srv.Images(false, "")
  958. if err != nil {
  959. t.Fatal(err)
  960. }
  961. if err := eng.Job("tag", unitTestImageName, "test", "test").Run(); err != nil {
  962. t.Fatal(err)
  963. }
  964. images, err := srv.Images(false, "")
  965. if err != nil {
  966. t.Fatal(err)
  967. }
  968. if len(images[0].RepoTags) != len(initialImages[0].RepoTags)+1 {
  969. t.Errorf("Expected %d images, %d found", len(initialImages)+1, len(images))
  970. }
  971. req, err := http.NewRequest("DELETE", "/images/"+unitTestImageID, nil)
  972. if err != nil {
  973. t.Fatal(err)
  974. }
  975. r := httptest.NewRecorder()
  976. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  977. t.Fatal(err)
  978. }
  979. if r.Code != http.StatusConflict {
  980. t.Fatalf("Expected http status 409-conflict, got %v", r.Code)
  981. }
  982. req2, err := http.NewRequest("DELETE", "/images/test:test", nil)
  983. if err != nil {
  984. t.Fatal(err)
  985. }
  986. r2 := httptest.NewRecorder()
  987. if err := docker.ServeRequest(srv, docker.APIVERSION, r2, req2); err != nil {
  988. t.Fatal(err)
  989. }
  990. assertHttpNotError(r2, t)
  991. if r2.Code != http.StatusOK {
  992. t.Fatalf("%d OK expected, received %d\n", http.StatusOK, r.Code)
  993. }
  994. var outs []docker.APIRmi
  995. if err := json.Unmarshal(r2.Body.Bytes(), &outs); err != nil {
  996. t.Fatal(err)
  997. }
  998. if len(outs) != 1 {
  999. t.Fatalf("Expected %d event (untagged), got %d", 1, len(outs))
  1000. }
  1001. images, err = srv.Images(false, "")
  1002. if err != nil {
  1003. t.Fatal(err)
  1004. }
  1005. if len(images[0].RepoTags) != len(initialImages[0].RepoTags) {
  1006. t.Errorf("Expected %d image, %d found", len(initialImages), len(images))
  1007. }
  1008. }
  1009. func TestPostContainersCopy(t *testing.T) {
  1010. eng := NewTestEngine(t)
  1011. defer mkRuntimeFromEngine(eng, t).Nuke()
  1012. srv := mkServerFromEngine(eng, t)
  1013. // Create a container and remove a file
  1014. containerID := createTestContainer(eng,
  1015. &docker.Config{
  1016. Image: unitTestImageID,
  1017. Cmd: []string{"touch", "/test.txt"},
  1018. },
  1019. t,
  1020. )
  1021. containerRun(eng, containerID, t)
  1022. r := httptest.NewRecorder()
  1023. copyData := docker.APICopy{HostPath: ".", Resource: "/test.txt"}
  1024. jsonData, err := json.Marshal(copyData)
  1025. if err != nil {
  1026. t.Fatal(err)
  1027. }
  1028. req, err := http.NewRequest("POST", "/containers/"+containerID+"/copy", bytes.NewReader(jsonData))
  1029. if err != nil {
  1030. t.Fatal(err)
  1031. }
  1032. req.Header.Add("Content-Type", "application/json")
  1033. if err := docker.ServeRequest(srv, docker.APIVERSION, r, req); err != nil {
  1034. t.Fatal(err)
  1035. }
  1036. assertHttpNotError(r, t)
  1037. if r.Code != http.StatusOK {
  1038. t.Fatalf("%d OK expected, received %d\n", http.StatusOK, r.Code)
  1039. }
  1040. found := false
  1041. for tarReader := tar.NewReader(r.Body); ; {
  1042. h, err := tarReader.Next()
  1043. if err != nil {
  1044. if err == io.EOF {
  1045. break
  1046. }
  1047. t.Fatal(err)
  1048. }
  1049. if h.Name == "test.txt" {
  1050. found = true
  1051. break
  1052. }
  1053. }
  1054. if !found {
  1055. t.Fatalf("The created test file has not been found in the copied output")
  1056. }
  1057. }
  1058. // Mocked types for tests
  1059. type NopConn struct {
  1060. io.ReadCloser
  1061. io.Writer
  1062. }
  1063. func (c *NopConn) LocalAddr() net.Addr { return nil }
  1064. func (c *NopConn) RemoteAddr() net.Addr { return nil }
  1065. func (c *NopConn) SetDeadline(t time.Time) error { return nil }
  1066. func (c *NopConn) SetReadDeadline(t time.Time) error { return nil }
  1067. func (c *NopConn) SetWriteDeadline(t time.Time) error { return nil }
  1068. type hijackTester struct {
  1069. *httptest.ResponseRecorder
  1070. in io.ReadCloser
  1071. out io.Writer
  1072. }
  1073. func (t *hijackTester) Hijack() (net.Conn, *bufio.ReadWriter, error) {
  1074. bufrw := bufio.NewReadWriter(bufio.NewReader(t.in), bufio.NewWriter(t.out))
  1075. conn := &NopConn{
  1076. ReadCloser: t.in,
  1077. Writer: t.out,
  1078. }
  1079. return conn, bufrw, nil
  1080. }