kemko/nomad
1
0
Fork 0
mirror of https://github.com/kemko/nomad.git synced 2026-01-06 10:25:42 +03:00
Code Issues Packages Projects Releases Wiki Activity

adding migration token validation for gc endpoint

Browse Source
This commit is contained in:
Chelsea Holland Komlo
2017-10-06 21:45:55 -04:00
committed by Alex Dadgar
parent 2368068355
commit 5e0da10bcd
1 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
command/agent/alloc_endpoint_test.go
View File

@@ -1,6 +1,7 @@
package agent
import (
"fmt"
"net/http"
"net/http/httptest"
"reflect"
@@ -317,6 +318,13 @@ func TestHTTP_AllocSnapshot(t *testing.T) {
})
}
func createMigrateTokenForClientAndAlloc(allocID, clientSecret string) (string, error) {
h, err := blake2b.New512([]byte(clientSecret))
h.Write([]byte(allocID))
validMigrateToken, err := string(h.Sum(nil)), nil
return validMigrateToken, err
}
func TestHTTP_AllocSnapshot_WithMigrateToken(t *testing.T) {
t.Parallel()
assert := assert.New(t)
@@ -336,22 +344,20 @@ func TestHTTP_AllocSnapshot_WithMigrateToken(t *testing.T) {
alloc := mock.Alloc()
state.UpsertJobSummary(998, mock.JobSummary(alloc.JobID))
// Set up data to create an authenticated request
h, err := blake2b.New512([]byte(s.Agent.Client().Node().SecretID))
h.Write([]byte(alloc.ID))
validMigrateToken, err := string(h.Sum(nil)), nil
validMigrateToken, err := createMigrateTokenForClientAndAlloc(alloc.ID, s.Agent.Client().Node().SecretID)
assert.Nil(err)
// Request with a token succeeds
req.Header.Set("X-Nomad-Token", validMigrateToken)
req, err = http.NewRequest("GET", "/v1/client/allocation/123/snapshot", nil)
url := fmt.Sprintf("/v1/client/allocation/%s/snapshot", alloc.ID)
req, err = http.NewRequest("GET", url, nil)
assert.Nil(err)
req.Header.Set("X-Nomad-Token", validMigrateToken)
// Make the unauthorized request
respW = httptest.NewRecorder()
_, err = s.Server.ClientAllocRequest(respW, req)
assert.NotNil(err)
assert.Contains(err.Error(), "invalid migrate token")
assert.NotContains(err.Error(), "invalid migrate token")
})
}