code: func WithState(c context.Context, s State) context.Context {
  	return context.WithValue(c, stateContextKey(0), s)
  }
docs: // WithState injects State into the context.
  //
  // Mostly useful from tests. Must not be normally used from production code,
  // 'Authenticate' sets the state itself.