@@ -44,7 +44,7 @@ func TestGetFeeds(t *testing.T) {
4444 assert .NoError (t , unittest .PrepareTestDatabase ())
4545 user := unittest .AssertExistsAndLoadBean (t , & user_model.User {ID : 2 })
4646
47- actions , err := activities_model .GetFeeds (db .DefaultContext , activities_model.GetFeedsOptions {
47+ actions , count , err := activities_model .GetFeeds (db .DefaultContext , activities_model.GetFeedsOptions {
4848 RequestedUser : user ,
4949 Actor : user ,
5050 IncludePrivate : true ,
@@ -56,15 +56,17 @@ func TestGetFeeds(t *testing.T) {
5656 assert .EqualValues (t , 1 , actions [0 ].ID )
5757 assert .EqualValues (t , user .ID , actions [0 ].UserID )
5858 }
59+ assert .Equal (t , int64 (1 ), count )
5960
60- actions , err = activities_model .GetFeeds (db .DefaultContext , activities_model.GetFeedsOptions {
61+ actions , count , err = activities_model .GetFeeds (db .DefaultContext , activities_model.GetFeedsOptions {
6162 RequestedUser : user ,
6263 Actor : user ,
6364 IncludePrivate : false ,
6465 OnlyPerformedBy : false ,
6566 })
6667 assert .NoError (t , err )
6768 assert .Len (t , actions , 0 )
69+ assert .Equal (t , int64 (0 ), count )
6870}
6971
7072func TestGetFeedsForRepos (t * testing.T ) {
@@ -74,38 +76,42 @@ func TestGetFeedsForRepos(t *testing.T) {
7476 pubRepo := unittest .AssertExistsAndLoadBean (t , & repo_model.Repository {ID : 8 })
7577
7678 // private repo & no login
77- actions , err := activities_model .GetFeeds (db .DefaultContext , activities_model.GetFeedsOptions {
79+ actions , count , err := activities_model .GetFeeds (db .DefaultContext , activities_model.GetFeedsOptions {
7880 RequestedRepo : privRepo ,
7981 IncludePrivate : true ,
8082 })
8183 assert .NoError (t , err )
8284 assert .Len (t , actions , 0 )
85+ assert .Equal (t , int64 (0 ), count )
8386
8487 // public repo & no login
85- actions , err = activities_model .GetFeeds (db .DefaultContext , activities_model.GetFeedsOptions {
88+ actions , count , err = activities_model .GetFeeds (db .DefaultContext , activities_model.GetFeedsOptions {
8689 RequestedRepo : pubRepo ,
8790 IncludePrivate : true ,
8891 })
8992 assert .NoError (t , err )
9093 assert .Len (t , actions , 1 )
94+ assert .Equal (t , int64 (1 ), count )
9195
9296 // private repo and login
93- actions , err = activities_model .GetFeeds (db .DefaultContext , activities_model.GetFeedsOptions {
97+ actions , count , err = activities_model .GetFeeds (db .DefaultContext , activities_model.GetFeedsOptions {
9498 RequestedRepo : privRepo ,
9599 IncludePrivate : true ,
96100 Actor : user ,
97101 })
98102 assert .NoError (t , err )
99103 assert .Len (t , actions , 1 )
104+ assert .Equal (t , int64 (1 ), count )
100105
101106 // public repo & login
102- actions , err = activities_model .GetFeeds (db .DefaultContext , activities_model.GetFeedsOptions {
107+ actions , count , err = activities_model .GetFeeds (db .DefaultContext , activities_model.GetFeedsOptions {
103108 RequestedRepo : pubRepo ,
104109 IncludePrivate : true ,
105110 Actor : user ,
106111 })
107112 assert .NoError (t , err )
108113 assert .Len (t , actions , 1 )
114+ assert .Equal (t , int64 (1 ), count )
109115}
110116
111117func TestGetFeeds2 (t * testing.T ) {
@@ -114,7 +120,7 @@ func TestGetFeeds2(t *testing.T) {
114120 org := unittest .AssertExistsAndLoadBean (t , & user_model.User {ID : 3 })
115121 user := unittest .AssertExistsAndLoadBean (t , & user_model.User {ID : 2 })
116122
117- actions , err := activities_model .GetFeeds (db .DefaultContext , activities_model.GetFeedsOptions {
123+ actions , count , err := activities_model .GetFeeds (db .DefaultContext , activities_model.GetFeedsOptions {
118124 RequestedUser : org ,
119125 Actor : user ,
120126 IncludePrivate : true ,
@@ -127,8 +133,9 @@ func TestGetFeeds2(t *testing.T) {
127133 assert .EqualValues (t , 2 , actions [0 ].ID )
128134 assert .EqualValues (t , org .ID , actions [0 ].UserID )
129135 }
136+ assert .Equal (t , int64 (1 ), count )
130137
131- actions , err = activities_model .GetFeeds (db .DefaultContext , activities_model.GetFeedsOptions {
138+ actions , count , err = activities_model .GetFeeds (db .DefaultContext , activities_model.GetFeedsOptions {
132139 RequestedUser : org ,
133140 Actor : user ,
134141 IncludePrivate : false ,
@@ -137,6 +144,7 @@ func TestGetFeeds2(t *testing.T) {
137144 })
138145 assert .NoError (t , err )
139146 assert .Len (t , actions , 0 )
147+ assert .Equal (t , int64 (0 ), count )
140148}
141149
142150func TestActivityReadable (t * testing.T ) {
@@ -224,13 +232,14 @@ func TestGetFeedsCorrupted(t *testing.T) {
224232 RepoID : 1700 ,
225233 })
226234
227- actions , err := activities_model .GetFeeds (db .DefaultContext , activities_model.GetFeedsOptions {
235+ actions , count , err := activities_model .GetFeeds (db .DefaultContext , activities_model.GetFeedsOptions {
228236 RequestedUser : user ,
229237 Actor : user ,
230238 IncludePrivate : true ,
231239 })
232240 assert .NoError (t , err )
233241 assert .Len (t , actions , 0 )
242+ assert .Equal (t , int64 (0 ), count )
234243}
235244
236245func TestConsistencyUpdateAction (t * testing.T ) {
0 commit comments