test: generate mode task id for testings (#3526)

Signed-off-by: Gaius <gaius.qi@gmail.com>
This commit is contained in:
Gaius 2024-09-24 10:38:18 +08:00 committed by GitHub
parent 80717c79df
commit 7579a9ba2d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 9 additions and 33 deletions

View File

@ -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",
} }
) )

View File

@ -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)

View File

@ -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)