@@ -45,6 +45,9 @@ func TestDBFunctions(t *testing.T) {
4545 t .Run ("WriteEvent succeeds" , func (t * testing.T ) {
4646 err := store .WriteEvent (ctx , debugSessionKey , "summary" , []byte (testEvent ))
4747 require .NoError (t , err )
48+
49+ err = store .WriteEvent (ctx , "another-session" , "summary" , []byte (testEvent ))
50+ require .NoError (t , err )
4851 })
4952
5053 t .Run ("QueryEvents with no filter" , func (t * testing.T ) {
@@ -110,7 +113,9 @@ func TestDBFunctions(t *testing.T) {
110113 require .NoError (t , err )
111114 err = store .WriteEvent (ctx , "session-2" , "diagnostic" , []byte (`{"kind":"diagnostic","data":"test"}` ))
112115 require .NoError (t , err )
113- // session-3 will have 0 events
116+
117+ err = store .WriteEvent (ctx , "session-3" , "diagnostic" , []byte (`{"kind":"diagnostic","data":"test"}` ))
118+ require .NoError (t , err )
114119
115120 // Query first page
116121 page , err := store .QueryDebugSessions (ctx , 2 , 0 )
@@ -131,7 +136,7 @@ func TestDBFunctions(t *testing.T) {
131136 allPage , err := store .QueryDebugSessions (ctx , 10 , 0 )
132137 require .NoError (t , err )
133138 require .NotNil (t , allPage )
134- require .Len (t , allPage .Sessions , 2 )
139+ require .Len (t , allPage .Sessions , 4 )
135140 require .Equal (t , int64 (4 ), allPage .TotalCount )
136141 require .False (t , allPage .HasMore )
137142
@@ -141,9 +146,9 @@ func TestDBFunctions(t *testing.T) {
141146 allSessionKeys [session .Key ] = session .EventCount
142147 }
143148
144- require .Equal (t , int64 (0 ), allSessionKeys ["session-3" ])
149+ require .Equal (t , int64 (1 ), allSessionKeys ["session-3" ])
145150 require .Equal (t , int64 (2 ), allSessionKeys ["session-2" ])
146- require .Equal (t , int64 (0 ), allSessionKeys ["another-session" ])
151+ require .Equal (t , int64 (1 ), allSessionKeys ["another-session" ])
147152 require .Equal (t , int64 (4 ), allSessionKeys [debugSessionKey ]) // 4 events from previous tests
148153
149154 // Test pagination - second page
0 commit comments