// Context sets the context on this session
968945
func (session *Session) Context(ctx context.Context) *Session {
	session.ctx = ctx
	return session
}
599911
func (c *GroupRolloutSetCall) Context(ctx context.Context) *GroupRolloutSetCall {
	c.ctx_ = ctx
	return c
}
819977
func (c *GetRefsCall) Context(ctx context.Context) *GetRefsCall {
	c.ctx_ = ctx
	return c
}
141794
func (eBase *elementBase) Context() *context {
	if eBase.parent != nil {
		return eBase.parent.Context()
	}

	return eBase.ctx
}
170030
func (c *BotTasksCall) Context(ctx context.Context) *BotTasksCall {
	c.ctx_ = ctx
	return c
}
106895
func (c *HeartbeatBatchCall) Context(ctx context.Context) *HeartbeatBatchCall {
	c.ctx_ = ctx
	return c
}
992525
func (c *GetConfigCall) Context(ctx context.Context) *GetConfigCall {
	c.ctx_ = ctx
	return c
}
245497
func (c *UnitStateListCall) Context(ctx context.Context) *UnitStateListCall {
	c.ctx_ = ctx
	return c
}
78080
func Context(ctx context.Context) Option {
	return func(o *Options) {
		o.Context = ctx
	}
}
1143860
func (c *protectedContext) context() context.Context {
	c.mu.RLock()
	defer c.mu.RUnlock()
	return c.ctx
}
1361047
func (c *GroupUpdateCall) Context(ctx context.Context) *GroupUpdateCall {
	c.ctx_ = ctx
	return c
}
277996
func (c *MachinesListCall) Context(ctx context.Context) *MachinesListCall {
	c.ctx_ = ctx
	return c
}
918538
func (c *BotEventsCall) Context(ctx context.Context) *BotEventsCall {
	c.ctx_ = ctx
	return c
}
1000366
func (c *AddMachineCall) Context(ctx context.Context) *AddMachineCall {
	c.ctx_ = ctx
	return c
}
998153
func (s *PutScriptService) Context(context string) *PutScriptService {
	s.context = context
	return s
}
474517
func (c *PutBatchCall) Context(ctx context.Context) *PutBatchCall {
	c.ctx_ = ctx
	return c
}
189804
func (c *GetTaskDefCall) Context(ctx context.Context) *GetTaskDefCall {
	c.ctx_ = ctx
	return c
}