func (s *HookContextSuite) AssertRelationContext(c *gc.C, ctx *context.HookContext, relId int, remoteUnit string) *context.ContextRelation { actualRemoteUnit, _ := ctx.RemoteUnitName() c.Assert(actualRemoteUnit, gc.Equals, remoteUnit) rel, err := ctx.HookRelation() c.Assert(err, jc.ErrorIsNil) c.Assert(rel.Id(), gc.Equals, relId) return rel.(*context.ContextRelation) }