Пример #1
0
func (S) TestAttemptNextHasNext(c *check.C) {
	a := aws.AttemptStrategy{}.Start()
	c.Assert(a.Next(), check.Equals, true)
	c.Assert(a.Next(), check.Equals, false)

	a = aws.AttemptStrategy{}.Start()
	c.Assert(a.Next(), check.Equals, true)
	c.Assert(a.HasNext(), check.Equals, false)
	c.Assert(a.Next(), check.Equals, false)

	a = aws.AttemptStrategy{Total: 2e8}.Start()
	c.Assert(a.Next(), check.Equals, true)
	c.Assert(a.HasNext(), check.Equals, true)
	time.Sleep(2e8)
	c.Assert(a.HasNext(), check.Equals, true)
	c.Assert(a.Next(), check.Equals, true)
	c.Assert(a.Next(), check.Equals, false)

	a = aws.AttemptStrategy{Total: 1e8, Min: 2}.Start()
	time.Sleep(1e8)
	c.Assert(a.Next(), check.Equals, true)
	c.Assert(a.HasNext(), check.Equals, true)
	c.Assert(a.Next(), check.Equals, true)
	c.Assert(a.HasNext(), check.Equals, false)
	c.Assert(a.Next(), check.Equals, false)
}