api_test.go 32 KB

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