mocks mocks mocks

mock/server/server.go

@@ -129,6 +129,7 @@ func FakeHandler() http.Handler {
 	e.GET("/api/v1/workers/:worker", getWorker)
 	e.POST("/api/v1/workers", addWorker)
 	e.PUT("/api/v1/workers/:worker", updateWorker)
+	e.POST("/api/v1/workers/:worker/refresh", refreshWorkerAuth)
 	e.DELETE("/api/v1/workers/:worker", removeWorker)
 
 	// mock endpoints for authentication calls

mock/server/worker.go

@@ -45,9 +45,6 @@ const (
 				],
 				"active": true,
 				"last_checked_in": 1602612590
-			},
-			"token": {
-				"token": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiJ3b3JrZXIiLCJpYXQiOjE1MTYyMzkwMjIsInRva2VuX3R5cGUiOiJXb3JrZXJBdXRoIn0.qeULIimCJlrwsE0JykNpzBmMaHUbvfk0vkyAz2eEo38"
 			}
 		}`
 
@@ -84,6 +81,11 @@ const (
     	"token": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiJ3b3JrZXIiLCJpYXQiOjE1MTYyMzkwMjIsInRva2VuX3R5cGUiOiJXb3JrZXJBdXRoIn0.qeULIimCJlrwsE0JykNpzBmMaHUbvfk0vkyAz2eEo38"
     }`
 
+	// RefreshWorkerAuthResp represents a JSON return for refreshing a worker's authentication.
+	RefreshWorkerAuthResp = `{
+		"token": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiJ3b3JrZXIiLCJpYXQiOjE1MTYyMzkwMjIsInRva2VuX3R5cGUiOiJXb3JrZXJBdXRoIn0.qeULIimCJlrwsE0JykNpzBmMaHUbvfk0vkyAz2eEo38"
+	}`
+
 	// RegisterTokenResp represents a JSON return for an admin requesting a registration token.
 	//
 	//nolint:gosec // not actual credentials
@@ -148,12 +150,29 @@ func updateWorker(c *gin.Context) {
 
 	data := []byte(UpdateWorkerResp)
 
-	type WorkerCheckIn struct {
-		Worker *library.Worker `json:"worker,omitempty"`
-		Token  *library.Token  `json:"token,omitempty"`
+	var body library.Worker
+	_ = json.Unmarshal(data, &body)
+
+	c.JSON(http.StatusOK, body)
+}
+
+// refreshWorkerAuth has a param :worker returns mock JSON for a http PUT.
+//
+// Pass "0" to :worker to test receiving a http 404 response.
+func refreshWorkerAuth(c *gin.Context) {
+	w := c.Param("worker")
+
+	if strings.EqualFold(w, "0") {
+		msg := fmt.Sprintf("Worker %s does not exist", w)
+
+		c.AbortWithStatusJSON(http.StatusNotFound, types.Error{Message: &msg})
+
+		return
 	}
 
-	var body WorkerCheckIn
+	data := []byte(RefreshWorkerAuthResp)
+
+	var body library.Token
 	_ = json.Unmarshal(data, &body)
 
 	c.JSON(http.StatusOK, body)