File: task_test.go

package info (click to toggle)
aptly 1.5.0%2Bds1-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 48,840 kB
  • sloc: python: 7,966; sh: 798; makefile: 81
file content (75 lines) | stat: -rw-r--r-- 2,513 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
package api

import (
	"encoding/json"
	"fmt"
	"time"

	"github.com/aptly-dev/aptly/task"

	. "gopkg.in/check.v1"
)

type TaskSuite struct {
	ApiSuite
}

var _ = Suite(&TaskSuite{})

func (s *TaskSuite) TestTasksDummy(c *C) {
	response, _ := s.HTTPRequest("POST", "/api/tasks-dummy", nil)
	c.Check(response.Code, Equals, 418)
	c.Check(response.Body.String(), Equals, "[1,2,3]")
}

func (s *TaskSuite) TestTasksDummyAsync(c *C) {
	response, _ := s.HTTPRequest("POST", "/api/tasks-dummy?_async=true", nil)
	c.Check(response.Code, Equals, 202)
	var t task.Task
	err := json.Unmarshal(response.Body.Bytes(), &t)
	c.Assert(err, IsNil)
	c.Check(t.Name, Equals, "Dummy task")
	response, _ = s.HTTPRequest("GET", fmt.Sprintf("/api/tasks/%d/wait", t.ID), nil)
	err = json.Unmarshal(response.Body.Bytes(), &t)
	c.Assert(err, IsNil)
	c.Check(t.State, Equals, task.SUCCEEDED)
	response, _ = s.HTTPRequest("GET", fmt.Sprintf("/api/tasks/%d/detail", t.ID), nil)
	c.Check(response.Code, Equals, 200)
	c.Check(response.Body.String(), Equals, "[1,2,3]")
	response, _ = s.HTTPRequest("GET", fmt.Sprintf("/api/tasks/%d/output", t.ID), nil)
	c.Check(response.Code, Equals, 200)
	c.Check(response.Body.String(), Matches, "\"Dummy task started.*")
}

func (s *TaskSuite) TestTaskDelete(c *C) {
	response, _ := s.HTTPRequest("POST", "/api/tasks-dummy?_async=true", nil)
	c.Check(response.Code, Equals, 202)
	c.Check(response.Body.String(), Equals, "{\"Name\":\"Dummy task\",\"ID\":1,\"State\":0}")
	// Give the task time to start
	time.Sleep(time.Second)
	response, _ = s.HTTPRequest("DELETE", "/api/tasks/1", nil)
	c.Check(response.Code, Equals, 200)
}

func (s *TaskSuite) TestTasksClear(c *C) {
	response, _ := s.HTTPRequest("POST", "/api/tasks-dummy?_async=true", nil)
	c.Check(response.Code, Equals, 202)
	var t task.Task
	err := json.Unmarshal(response.Body.Bytes(), &t)
	c.Assert(err, IsNil)
	c.Check(t.Name, Equals, "Dummy task")
	response, _ = s.HTTPRequest("GET", "/api/tasks-wait", nil)
	c.Check(response.Code, Equals, 200)
	response, _ = s.HTTPRequest("GET", "/api/tasks", nil)
	c.Check(response.Code, Equals, 200)
	var ts []task.Task
	err = json.Unmarshal(response.Body.Bytes(), &ts)
	c.Assert(err, IsNil)
	c.Check(len(ts), Equals, 1)
	c.Check(ts[0].State, Equals, task.SUCCEEDED)
	response, _ = s.HTTPRequest("POST", "/api/tasks-clear", nil)
	c.Check(response.Code, Equals, 200)
	response, _ = s.HTTPRequest("GET", "/api/tasks", nil)
	c.Check(response.Code, Equals, 200)
	c.Check(response.Body.String(), Equals, "null")
}