api_test.go 32 KB

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