api_test.go 31 KB

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