github.com/lingyao2333/mo-zero@v1.4.1/core/stores/mongo/query_test.go (about) 1 package mongo 2 3 import ( 4 "testing" 5 6 "github.com/globalsign/mgo" 7 "github.com/lingyao2333/mo-zero/core/breaker" 8 "github.com/stretchr/testify/assert" 9 ) 10 11 func Test_rejectedQuery_All(t *testing.T) { 12 assert.Equal(t, breaker.ErrServiceUnavailable, new(rejectedQuery).All(nil)) 13 } 14 15 func Test_rejectedQuery_Apply(t *testing.T) { 16 info, err := new(rejectedQuery).Apply(mgo.Change{}, nil) 17 assert.Equal(t, breaker.ErrServiceUnavailable, err) 18 assert.Nil(t, info) 19 } 20 21 func Test_rejectedQuery_Batch(t *testing.T) { 22 var q rejectedQuery 23 assert.Equal(t, q, q.Batch(1)) 24 } 25 26 func Test_rejectedQuery_Collation(t *testing.T) { 27 var q rejectedQuery 28 assert.Equal(t, q, q.Collation(nil)) 29 } 30 31 func Test_rejectedQuery_Comment(t *testing.T) { 32 var q rejectedQuery 33 assert.Equal(t, q, q.Comment("")) 34 } 35 36 func Test_rejectedQuery_Count(t *testing.T) { 37 n, err := new(rejectedQuery).Count() 38 assert.Equal(t, breaker.ErrServiceUnavailable, err) 39 assert.Equal(t, 0, n) 40 } 41 42 func Test_rejectedQuery_Distinct(t *testing.T) { 43 assert.Equal(t, breaker.ErrServiceUnavailable, new(rejectedQuery).Distinct("", nil)) 44 } 45 46 func Test_rejectedQuery_Explain(t *testing.T) { 47 assert.Equal(t, breaker.ErrServiceUnavailable, new(rejectedQuery).Explain(nil)) 48 } 49 50 func Test_rejectedQuery_For(t *testing.T) { 51 assert.Equal(t, breaker.ErrServiceUnavailable, new(rejectedQuery).For(nil, nil)) 52 } 53 54 func Test_rejectedQuery_Hint(t *testing.T) { 55 var q rejectedQuery 56 assert.Equal(t, q, q.Hint()) 57 } 58 59 func Test_rejectedQuery_Iter(t *testing.T) { 60 assert.EqualValues(t, rejectedIter{}, new(rejectedQuery).Iter()) 61 } 62 63 func Test_rejectedQuery_Limit(t *testing.T) { 64 var q rejectedQuery 65 assert.Equal(t, q, q.Limit(1)) 66 } 67 68 func Test_rejectedQuery_LogReplay(t *testing.T) { 69 var q rejectedQuery 70 assert.Equal(t, q, q.LogReplay()) 71 } 72 73 func Test_rejectedQuery_MapReduce(t *testing.T) { 74 info, err := new(rejectedQuery).MapReduce(nil, nil) 75 assert.Equal(t, breaker.ErrServiceUnavailable, err) 76 assert.Nil(t, info) 77 } 78 79 func Test_rejectedQuery_One(t *testing.T) { 80 assert.Equal(t, breaker.ErrServiceUnavailable, new(rejectedQuery).One(nil)) 81 } 82 83 func Test_rejectedQuery_Prefetch(t *testing.T) { 84 var q rejectedQuery 85 assert.Equal(t, q, q.Prefetch(1)) 86 } 87 88 func Test_rejectedQuery_Select(t *testing.T) { 89 var q rejectedQuery 90 assert.Equal(t, q, q.Select(nil)) 91 } 92 93 func Test_rejectedQuery_SetMaxScan(t *testing.T) { 94 var q rejectedQuery 95 assert.Equal(t, q, q.SetMaxScan(0)) 96 } 97 98 func Test_rejectedQuery_SetMaxTime(t *testing.T) { 99 var q rejectedQuery 100 assert.Equal(t, q, q.SetMaxTime(0)) 101 } 102 103 func Test_rejectedQuery_Skip(t *testing.T) { 104 var q rejectedQuery 105 assert.Equal(t, q, q.Skip(0)) 106 } 107 108 func Test_rejectedQuery_Snapshot(t *testing.T) { 109 var q rejectedQuery 110 assert.Equal(t, q, q.Snapshot()) 111 } 112 113 func Test_rejectedQuery_Sort(t *testing.T) { 114 var q rejectedQuery 115 assert.Equal(t, q, q.Sort()) 116 } 117 118 func Test_rejectedQuery_Tail(t *testing.T) { 119 assert.EqualValues(t, rejectedIter{}, new(rejectedQuery).Tail(0)) 120 }