func (cs *clientSuite) TestClientDoReportsErrors(c *check.C) { restore := client.MockDoRetry(10*time.Millisecond, 100*time.Millisecond) defer restore() cs.err = errors.New("ouchie") err := cs.cli.Do("GET", "/", nil, nil, nil) c.Check(err, check.ErrorMatches, "cannot communicate with server: ouchie") if cs.doCalls < 2 { c.Fatalf("do did not retry") } }
func (s *SnapOpSuite) SetUpTest(c *check.C) { s.BaseSnapSuite.SetUpTest(c) restoreClientRetry := client.MockDoRetry(time.Millisecond, 10*time.Millisecond) restorePollTime := snap.MockPollTime(time.Millisecond) s.restoreAll = func() { restoreClientRetry() restorePollTime() } s.srv = snapOpTestServer{ c: c, total: 4, } }