test: generate mode task id for testings (#3526)
Signed-off-by: Gaius <gaius.qi@gmail.com>
This commit is contained in:
parent
80717c79df
commit
7579a9ba2d
|
|
@ -81,21 +81,21 @@ var (
|
||||||
UserID: 4,
|
UserID: 4,
|
||||||
Type: "preheat",
|
Type: "preheat",
|
||||||
BIO: "bio",
|
BIO: "bio",
|
||||||
TaskID: "2",
|
TaskID: "dec6fe878785cea844dcecdf2ea25e19156822201016455733e47e9f0bfab563",
|
||||||
}
|
}
|
||||||
mockGetTaskJobModel = &models.Job{
|
mockGetTaskJobModel = &models.Job{
|
||||||
BaseModel: mockBaseModel,
|
BaseModel: mockBaseModel,
|
||||||
UserID: 4,
|
UserID: 4,
|
||||||
Type: "get_task",
|
Type: "get_task",
|
||||||
BIO: "bio",
|
BIO: "bio",
|
||||||
TaskID: "2",
|
TaskID: "7575d21d69495905a4709bf4e10d0e5cffcf7fd1e76e93171e0ef6e0abcf07a8",
|
||||||
}
|
}
|
||||||
mockDeleteTaskJobModel = &models.Job{
|
mockDeleteTaskJobModel = &models.Job{
|
||||||
BaseModel: mockBaseModel,
|
BaseModel: mockBaseModel,
|
||||||
UserID: 4,
|
UserID: 4,
|
||||||
Type: "delete_task",
|
Type: "delete_task",
|
||||||
BIO: "bio",
|
BIO: "bio",
|
||||||
TaskID: "2",
|
TaskID: "04a29122b0c4d0affde2d577fb36bb956caa3da10e9130375623c24a5f865a49",
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -47,7 +47,7 @@ func TestTask_CreateGetTask(t *testing.T) {
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
args: types.GetTaskArgs{
|
args: types.GetTaskArgs{
|
||||||
TaskID: "valid-task-id",
|
TaskID: "e7a7132d133ab769af4b64c54c0ea4572d2089142d9f1c04203572697439f467",
|
||||||
},
|
},
|
||||||
expect: func(t *testing.T, g *job.GroupJobState, e error) {
|
expect: func(t *testing.T, g *job.GroupJobState, e error) {
|
||||||
assert := assert.New(t)
|
assert := assert.New(t)
|
||||||
|
|
@ -63,7 +63,7 @@ func TestTask_CreateGetTask(t *testing.T) {
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
args: types.GetTaskArgs{
|
args: types.GetTaskArgs{
|
||||||
TaskID: "valid-task-id",
|
TaskID: "55697b6298a6ee5118443f23c4f4532666421361d31d4b1e822d2480c42eda0c",
|
||||||
},
|
},
|
||||||
expect: func(t *testing.T, g *job.GroupJobState, e error) {
|
expect: func(t *testing.T, g *job.GroupJobState, e error) {
|
||||||
assert := assert.New(t)
|
assert := assert.New(t)
|
||||||
|
|
@ -98,7 +98,7 @@ func TestTask_CreateDeleteTask(t *testing.T) {
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
args: types.DeleteTaskArgs{
|
args: types.DeleteTaskArgs{
|
||||||
TaskID: "valid-task-id",
|
TaskID: "e7a7132d133ab769af4b64c54c0ea4572d2089142d9f1c04203572697439f467",
|
||||||
},
|
},
|
||||||
expect: func(t *testing.T, g *job.GroupJobState, e error) {
|
expect: func(t *testing.T, g *job.GroupJobState, e error) {
|
||||||
assert := assert.New(t)
|
assert := assert.New(t)
|
||||||
|
|
@ -114,7 +114,7 @@ func TestTask_CreateDeleteTask(t *testing.T) {
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
args: types.DeleteTaskArgs{
|
args: types.DeleteTaskArgs{
|
||||||
TaskID: "valid-task-id",
|
TaskID: "55697b6298a6ee5118443f23c4f4532666421361d31d4b1e822d2480c42eda0c",
|
||||||
},
|
},
|
||||||
expect: func(t *testing.T, g *job.GroupJobState, e error) {
|
expect: func(t *testing.T, g *job.GroupJobState, e error) {
|
||||||
assert := assert.New(t)
|
assert := assert.New(t)
|
||||||
|
|
|
||||||
|
|
@ -33,7 +33,6 @@ import (
|
||||||
var _ = Describe("GetTask and DeleteTask with Manager", func() {
|
var _ = Describe("GetTask and DeleteTask with Manager", func() {
|
||||||
Context("/bin/cat file", Label("getTask", "deleteTask", "file"), func() {
|
Context("/bin/cat file", Label("getTask", "deleteTask", "file"), func() {
|
||||||
It("getTask and deleteTask should be ok", func() {
|
It("getTask and deleteTask should be ok", func() {
|
||||||
// Create preheat job.
|
|
||||||
managerPod, err := util.ManagerExec(0)
|
managerPod, err := util.ManagerExec(0)
|
||||||
fmt.Println(err)
|
fmt.Println(err)
|
||||||
Expect(err).NotTo(HaveOccurred())
|
Expect(err).NotTo(HaveOccurred())
|
||||||
|
|
@ -74,12 +73,10 @@ var _ = Describe("GetTask and DeleteTask with Manager", func() {
|
||||||
Expect(err).NotTo(HaveOccurred())
|
Expect(err).NotTo(HaveOccurred())
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check the file is downloaded successfully.
|
|
||||||
sha256sum, err := util.CalculateSha256ByTaskID(seedClientPods, fileMetadata.ID)
|
sha256sum, err := util.CalculateSha256ByTaskID(seedClientPods, fileMetadata.ID)
|
||||||
Expect(err).NotTo(HaveOccurred())
|
Expect(err).NotTo(HaveOccurred())
|
||||||
Expect(fileMetadata.Sha256).To(Equal(sha256sum))
|
Expect(fileMetadata.Sha256).To(Equal(sha256sum))
|
||||||
|
|
||||||
// Get task.
|
|
||||||
req, err = structure.StructToMap(types.CreateGetTaskJobRequest{
|
req, err = structure.StructToMap(types.CreateGetTaskJobRequest{
|
||||||
Type: internaljob.GetTaskJob,
|
Type: internaljob.GetTaskJob,
|
||||||
Args: types.GetTaskArgs{
|
Args: types.GetTaskArgs{
|
||||||
|
|
@ -100,11 +97,8 @@ var _ = Describe("GetTask and DeleteTask with Manager", func() {
|
||||||
|
|
||||||
done = waitForDone(job, managerPod)
|
done = waitForDone(job, managerPod)
|
||||||
Expect(done).Should(BeTrue())
|
Expect(done).Should(BeTrue())
|
||||||
|
|
||||||
// Check get task response is valid.
|
|
||||||
Expect(validateTaskResponse(job)).To(BeTrue())
|
Expect(validateTaskResponse(job)).To(BeTrue())
|
||||||
|
|
||||||
// Delete task.
|
|
||||||
req, err = structure.StructToMap(types.CreateDeleteTaskJobRequest{
|
req, err = structure.StructToMap(types.CreateDeleteTaskJobRequest{
|
||||||
Type: internaljob.DeleteTaskJob,
|
Type: internaljob.DeleteTaskJob,
|
||||||
Args: types.DeleteTaskArgs{
|
Args: types.DeleteTaskArgs{
|
||||||
|
|
@ -125,11 +119,8 @@ var _ = Describe("GetTask and DeleteTask with Manager", func() {
|
||||||
|
|
||||||
done = waitForDone(job, managerPod)
|
done = waitForDone(job, managerPod)
|
||||||
Expect(done).Should(BeTrue())
|
Expect(done).Should(BeTrue())
|
||||||
|
|
||||||
// Check delete task response is valid.
|
|
||||||
Expect(validateTaskResponse(job)).To(BeTrue())
|
Expect(validateTaskResponse(job)).To(BeTrue())
|
||||||
|
|
||||||
// Check file is deleted successfully.
|
|
||||||
exist := util.CheckFilesExist(seedClientPods, fileMetadata.ID)
|
exist := util.CheckFilesExist(seedClientPods, fileMetadata.ID)
|
||||||
Expect(exist).Should(BeFalse())
|
Expect(exist).Should(BeFalse())
|
||||||
})
|
})
|
||||||
|
|
@ -137,7 +128,6 @@ var _ = Describe("GetTask and DeleteTask with Manager", func() {
|
||||||
|
|
||||||
Context("/bin/pwd file", Label("getTask", "deleteTask", "file"), func() {
|
Context("/bin/pwd file", Label("getTask", "deleteTask", "file"), func() {
|
||||||
It("getTask and deleteTask should be ok", func() {
|
It("getTask and deleteTask should be ok", func() {
|
||||||
// Create preheat job.
|
|
||||||
managerPod, err := util.ManagerExec(0)
|
managerPod, err := util.ManagerExec(0)
|
||||||
fmt.Println(err)
|
fmt.Println(err)
|
||||||
Expect(err).NotTo(HaveOccurred())
|
Expect(err).NotTo(HaveOccurred())
|
||||||
|
|
@ -178,12 +168,10 @@ var _ = Describe("GetTask and DeleteTask with Manager", func() {
|
||||||
Expect(err).NotTo(HaveOccurred())
|
Expect(err).NotTo(HaveOccurred())
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check the file is downloaded successfully.
|
|
||||||
sha256sum, err := util.CalculateSha256ByTaskID(seedClientPods, fileMetadata.ID)
|
sha256sum, err := util.CalculateSha256ByTaskID(seedClientPods, fileMetadata.ID)
|
||||||
Expect(err).NotTo(HaveOccurred())
|
Expect(err).NotTo(HaveOccurred())
|
||||||
Expect(fileMetadata.Sha256).To(Equal(sha256sum))
|
Expect(fileMetadata.Sha256).To(Equal(sha256sum))
|
||||||
|
|
||||||
// Get task.
|
|
||||||
req, err = structure.StructToMap(types.CreateGetTaskJobRequest{
|
req, err = structure.StructToMap(types.CreateGetTaskJobRequest{
|
||||||
Type: internaljob.GetTaskJob,
|
Type: internaljob.GetTaskJob,
|
||||||
Args: types.GetTaskArgs{
|
Args: types.GetTaskArgs{
|
||||||
|
|
@ -204,11 +192,8 @@ var _ = Describe("GetTask and DeleteTask with Manager", func() {
|
||||||
|
|
||||||
done = waitForDone(job, managerPod)
|
done = waitForDone(job, managerPod)
|
||||||
Expect(done).Should(BeTrue())
|
Expect(done).Should(BeTrue())
|
||||||
|
|
||||||
// Check get task response is valid.
|
|
||||||
Expect(validateTaskResponse(job)).To(BeTrue())
|
Expect(validateTaskResponse(job)).To(BeTrue())
|
||||||
|
|
||||||
// Delete task.
|
|
||||||
req, err = structure.StructToMap(types.CreateDeleteTaskJobRequest{
|
req, err = structure.StructToMap(types.CreateDeleteTaskJobRequest{
|
||||||
Type: internaljob.DeleteTaskJob,
|
Type: internaljob.DeleteTaskJob,
|
||||||
Args: types.DeleteTaskArgs{
|
Args: types.DeleteTaskArgs{
|
||||||
|
|
@ -229,11 +214,8 @@ var _ = Describe("GetTask and DeleteTask with Manager", func() {
|
||||||
|
|
||||||
done = waitForDone(job, managerPod)
|
done = waitForDone(job, managerPod)
|
||||||
Expect(done).Should(BeTrue())
|
Expect(done).Should(BeTrue())
|
||||||
|
|
||||||
// Check delete task response is valid.
|
|
||||||
Expect(validateTaskResponse(job)).To(BeTrue())
|
Expect(validateTaskResponse(job)).To(BeTrue())
|
||||||
|
|
||||||
// Check file is deleted successfully.
|
|
||||||
exist := util.CheckFilesExist(seedClientPods, fileMetadata.ID)
|
exist := util.CheckFilesExist(seedClientPods, fileMetadata.ID)
|
||||||
Expect(exist).Should(BeFalse())
|
Expect(exist).Should(BeFalse())
|
||||||
})
|
})
|
||||||
|
|
@ -257,11 +239,9 @@ var _ = Describe("GetTask and DeleteTask with Manager", func() {
|
||||||
Expect(err).NotTo(HaveOccurred())
|
Expect(err).NotTo(HaveOccurred())
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check the file is not exist.
|
|
||||||
exist := util.CheckFilesExist(seedClientPods, fileMetadata.ID)
|
exist := util.CheckFilesExist(seedClientPods, fileMetadata.ID)
|
||||||
Expect(exist).Should(BeFalse())
|
Expect(exist).Should(BeFalse())
|
||||||
|
|
||||||
// Get task.
|
|
||||||
req, err := structure.StructToMap(types.CreateGetTaskJobRequest{
|
req, err := structure.StructToMap(types.CreateGetTaskJobRequest{
|
||||||
Type: internaljob.GetTaskJob,
|
Type: internaljob.GetTaskJob,
|
||||||
Args: types.GetTaskArgs{
|
Args: types.GetTaskArgs{
|
||||||
|
|
@ -282,11 +262,8 @@ var _ = Describe("GetTask and DeleteTask with Manager", func() {
|
||||||
|
|
||||||
done := waitForDone(job, managerPod)
|
done := waitForDone(job, managerPod)
|
||||||
Expect(done).Should(BeTrue())
|
Expect(done).Should(BeTrue())
|
||||||
|
|
||||||
// Check get task response is invalid.
|
|
||||||
Expect(validateTaskResponse(job)).To(BeFalse())
|
Expect(validateTaskResponse(job)).To(BeFalse())
|
||||||
|
|
||||||
// Delete task.
|
|
||||||
req, err = structure.StructToMap(types.CreateDeleteTaskJobRequest{
|
req, err = structure.StructToMap(types.CreateDeleteTaskJobRequest{
|
||||||
Type: internaljob.DeleteTaskJob,
|
Type: internaljob.DeleteTaskJob,
|
||||||
Args: types.DeleteTaskArgs{
|
Args: types.DeleteTaskArgs{
|
||||||
|
|
@ -308,23 +285,22 @@ var _ = Describe("GetTask and DeleteTask with Manager", func() {
|
||||||
done = waitForDone(job, managerPod)
|
done = waitForDone(job, managerPod)
|
||||||
Expect(done).Should(BeTrue())
|
Expect(done).Should(BeTrue())
|
||||||
|
|
||||||
// Check delete task response is invalid.
|
|
||||||
Expect(validateTaskResponse(job)).To(BeFalse())
|
Expect(validateTaskResponse(job)).To(BeFalse())
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
func validateTaskResponse(job *models.Job) bool {
|
func validateTaskResponse(job *models.Job) bool {
|
||||||
// Check task response is not null.
|
|
||||||
Expect(job.Result).NotTo(BeNil())
|
Expect(job.Result).NotTo(BeNil())
|
||||||
|
|
||||||
groupJobStateData, err := json.Marshal(job.Result)
|
groupJobStateData, err := json.Marshal(job.Result)
|
||||||
Expect(err).NotTo(HaveOccurred())
|
Expect(err).NotTo(HaveOccurred())
|
||||||
|
|
||||||
groupJobState := internaljob.GroupJobState{}
|
groupJobState := internaljob.GroupJobState{}
|
||||||
err = json.Unmarshal(groupJobStateData, &groupJobState)
|
err = json.Unmarshal(groupJobStateData, &groupJobState)
|
||||||
Expect(err).NotTo(HaveOccurred())
|
Expect(err).NotTo(HaveOccurred())
|
||||||
Expect(len(groupJobState.JobStates)).Should(BeNumerically("==", 3))
|
Expect(len(groupJobState.JobStates)).Should(BeNumerically("==", 3))
|
||||||
|
|
||||||
// Check task response is valid.
|
|
||||||
for _, state := range groupJobState.JobStates {
|
for _, state := range groupJobState.JobStates {
|
||||||
for _, result := range state.Results {
|
for _, result := range state.Results {
|
||||||
resultData, err := json.Marshal(result)
|
resultData, err := json.Marshal(result)
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue