Add a test for nil-ing out request tokenentry

Crazybus · Jun 8, 2018 · 8eee1da · 8eee1da
1 parent 6ec09a0
commit 8eee1da
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/vault/router_test.go b/vault/router_test.go
@@ -31,6 +31,10 @@ type NoopBackend struct {
 }
 
 func (n *NoopBackend) HandleRequest(ctx context.Context, req *logical.Request) (*logical.Response, error) {
+	if req.TokenEntry() != nil {
+		panic("got a non-nil TokenEntry")
+	}
+
 	var err error
 	resp := n.Response
 	if n.RequestHandler != nil {
@@ -171,13 +175,19 @@ func TestRouter_Mount(t *testing.T) {
 	req := &logical.Request{
 		Path: "prod/aws/foo",
 	}
+	req.SetTokenEntry(&logical.TokenEntry{
+		ID: "foo",
+	})
 	resp, err := r.Route(context.Background(), req)
 	if err != nil {
 		t.Fatalf("err: %v", err)
 	}
 	if resp != nil {
 		t.Fatalf("bad: %v", resp)
 	}
+	if req.TokenEntry() == nil || req.TokenEntry().ID != "foo" {
+		t.Fatalf("unexpected value for token entry: %v", req.TokenEntry())
+	}
 
 	// Verify the path
 	if len(n.Paths) != 1 || n.Paths[0] != "foo" {