pipe_test.go 1.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. package mongo
  2. import (
  3. "testing"
  4. "github.com/stretchr/testify/assert"
  5. "github.com/tal-tech/go-zero/core/breaker"
  6. )
  7. func TestRejectedPipe_All(t *testing.T) {
  8. assert.Equal(t, breaker.ErrServiceUnavailable, new(rejectedPipe).All(nil))
  9. }
  10. func TestRejectedPipe_AllowDiskUse(t *testing.T) {
  11. var p rejectedPipe
  12. assert.Equal(t, p, p.AllowDiskUse())
  13. }
  14. func TestRejectedPipe_Batch(t *testing.T) {
  15. var p rejectedPipe
  16. assert.Equal(t, p, p.Batch(1))
  17. }
  18. func TestRejectedPipe_Collation(t *testing.T) {
  19. var p rejectedPipe
  20. assert.Equal(t, p, p.Collation(nil))
  21. }
  22. func TestRejectedPipe_Explain(t *testing.T) {
  23. assert.Equal(t, breaker.ErrServiceUnavailable, new(rejectedPipe).Explain(nil))
  24. }
  25. func TestRejectedPipe_Iter(t *testing.T) {
  26. assert.EqualValues(t, rejectedIter{}, new(rejectedPipe).Iter())
  27. }
  28. func TestRejectedPipe_One(t *testing.T) {
  29. assert.Equal(t, breaker.ErrServiceUnavailable, new(rejectedPipe).One(nil))
  30. }
  31. func TestRejectedPipe_SetMaxTime(t *testing.T) {
  32. var p rejectedPipe
  33. assert.Equal(t, p, p.SetMaxTime(0))
  34. }