api_test.go 32 KB

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