// Context provides a mock function with given fields:
516381
func (_m *StepStream) Context() context.Context {
	ret := _m.Called()

	var r0 context.Context
	if rf, ok := ret.Get(0).(func() context.Context); ok {
		r0 = rf()
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(context.Context)
		}
	}

	return r0
}
956007
func (c *AdminGetUserCall) Context(ctx context.Context) *AdminGetUserCall {
	c.ctx_ = ctx
	return c
}
611753
func (c *StoreInlineCall) Context(ctx context.Context) *StoreInlineCall {
	c.ctx_ = ctx
	return c
}
1103621
func (c *AppUpdateCall) Context(ctx context.Context) *AppUpdateCall {
	c.ctx_ = ctx
	return c
}
1334209
func (c *AppPackagePublicListCall) Context(ctx context.Context) *AppPackagePublicListCall {
	c.ctx_ = ctx
	return c
}
245497
func (c *UnitStateListCall) Context(ctx context.Context) *UnitStateListCall {
	c.ctx_ = ctx
	return c
}
909694
func (c *AddMachinesCall) Context(ctx context.Context) *AddMachinesCall {
	c.ctx_ = ctx
	return c
}
193296
func (c *RetryCall) Context(ctx context.Context) *RetryCall {
	c.ctx_ = ctx
	return c
}
1282844
func (c *GroupListCall) Context(ctx context.Context) *GroupListCall {
	c.ctx_ = ctx
	return c
}
1135841
func (c *Core) Context() *interpolate.Context {
	return &interpolate.Context{
		TemplatePath:  c.Template.Path,
		UserVariables: c.variables,
	}
}
599911
func (c *GroupRolloutSetCall) Context(ctx context.Context) *GroupRolloutSetCall {
	c.ctx_ = ctx
	return c
}
959402
func (c *GroupRequestsVersionsRollupCall) Context(ctx context.Context) *GroupRequestsVersionsRollupCall {
	c.ctx_ = ctx
	return c
}
1340159
func (c *ValidateConfigCall) Context(ctx context.Context) *ValidateConfigCall {
	c.ctx_ = ctx
	return c
}
1000366
func (c *AddMachineCall) Context(ctx context.Context) *AddMachineCall {
	c.ctx_ = ctx
	return c
}
1055825
func (c *ClientupdateCountCall) Context(ctx context.Context) *ClientupdateCountCall {
	c.ctx_ = ctx
	return c
}
170030
func (c *BotTasksCall) Context(ctx context.Context) *BotTasksCall {
	c.ctx_ = ctx
	return c
}
560340
func (c *AckCall) Context(ctx context.Context) *AckCall {
	c.ctx_ = ctx
	return c
}