ridv/gorealis
1
0
Fork 0
Code Issues Pull requests Projects Releases 23 Packages Wiki Activity Actions

Addressing feedback comments. Running code through goimports.

Browse source
This commit is contained in:
Renan DelValle 2019-01-08 11:06:08 -08:00
parent 25f5ff9f07
commit f39551af14
No known key found for this signature in database
GPG key ID: C240AD6D6F443EC9
2 changed files with 8 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

11
job.go
View file

@ -96,15 +96,8 @@ func NewJob() Job {
ramMb := aurora.NewResource() ramMb := aurora.NewResource()
diskMb := aurora.NewResource() diskMb := aurora.NewResource()
resources := make(map[ResourceType]*aurora.Resource) resources := map[ResourceType]*aurora.Resource{CPU: numCpus, RAM: ramMb, DISK: diskMb}
resources[CPU] = numCpus taskConfig.Resources = map[*aurora.Resource]bool{numCpus: true, ramMb: true, diskMb: true}
resources[RAM] = ramMb
resources[DISK] = diskMb
taskConfig.Resources = make(map[*aurora.Resource]bool)
taskConfig.Resources[numCpus] = true
taskConfig.Resources[ramMb] = true
taskConfig.Resources[diskMb] = true
numCpus.NumCpus = new(float64) numCpus.NumCpus = new(float64)
ramMb.RamMb = new(int64) ramMb.RamMb = new(int64)

7
realis_e2e_test.go
View file

@ -28,6 +28,7 @@ import (
"github.com/paypal/gorealis/response" "github.com/paypal/gorealis/response"
"github.com/pkg/errors" "github.com/pkg/errors"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
) )
var r realis.Realis var r realis.Realis
@ -338,10 +339,9 @@ pulseLoop:
select { select {
case <-ticker.C: case <-ticker.C:
fmt.Println("sending PulseJobUpdate....") fmt.Println("sending PulseJobUpdate....")
resp, err = r.PulseJobUpdate(result.GetKey()) resp, err = r.PulseJobUpdate(result.GetKey())
assert.NotNil(t, resp) require.NotNil(t, resp, "received a nil response from Aurora")
assert.Nil(t, err) assert.Nil(t, err)
respDetail, err := r.JobUpdateDetails(updateQ) respDetail, err := r.JobUpdateDetails(updateQ)
@ -372,8 +372,7 @@ pulseLoop:
_, err := r.AbortJobUpdate(*updateDetails[0].GetUpdate().GetSummary().GetKey(), "") _, err := r.AbortJobUpdate(*updateDetails[0].GetUpdate().GetSummary().GetKey(), "")
assert.NoError(t, err) assert.NoError(t, err)
_, err = r.KillJob(job.JobKey()) _, err = r.KillJob(job.JobKey())
assert.NoError(t, err) require.NoError(t, err, "timed out during pulse update test")
assert.FailNow(t, "timed out during pulse update test")
} }
} }