@@ -440,12 +440,17 @@ void main() {
440440 late PerAccountStore store;
441441
442442 Future <void > prepare ({
443+ User ? selfUser,
443444 List <User > users = const [],
444445 List <UserGroup > userGroups = const [],
445446 List <RecentDmConversation > dmConversations = const [],
446447 List <Message > messages = const [],
447448 }) async {
448- store = eg.store (initialSnapshot: eg.initialSnapshot (
449+ selfUser ?? = eg.selfUser;
450+ if (! users.contains (selfUser)) {
451+ users = [...users, selfUser];
452+ }
453+ store = eg.store (selfUser: selfUser, initialSnapshot: eg.initialSnapshot (
449454 realmUsers: users,
450455 recentPrivateConversations: dmConversations));
451456 await store.addUserGroups (userGroups);
@@ -817,6 +822,7 @@ void main() {
817822 eg.user (userId: 6 , fullName: 'User Six' , isBot: true ),
818823 eg.user (userId: 7 , fullName: 'User Seven' ),
819824 ];
825+ final selfUser = users.last;
820826
821827 final userGroups = [
822828 eg.userGroup (id: 1 , name: 'User Group One' ),
@@ -825,13 +831,14 @@ void main() {
825831 eg.userGroup (id: 4 , name: 'User Group Four' ),
826832 ];
827833
828- await prepare (users: users, userGroups: userGroups, messages: [
829- eg.streamMessage (sender: users[1 - 1 ], stream: stream, topic: topic),
830- eg.streamMessage (sender: users[5 - 1 ], stream: stream, topic: 'other $topic ' ),
831- eg.dmMessage (from: users[1 - 1 ], to: [users[2 - 1 ], eg.selfUser]),
832- eg.dmMessage (from: users[1 - 1 ], to: [eg.selfUser]),
833- eg.dmMessage (from: users[4 - 1 ], to: [eg.selfUser]),
834- ]);
834+ await prepare (users: users, selfUser: selfUser, userGroups: userGroups,
835+ messages: [
836+ eg.streamMessage (sender: users[1 - 1 ], stream: stream, topic: topic),
837+ eg.streamMessage (sender: users[5 - 1 ], stream: stream, topic: 'other $topic ' ),
838+ eg.dmMessage (from: users[1 - 1 ], to: [users[2 - 1 ], selfUser]),
839+ eg.dmMessage (from: users[1 - 1 ], to: [selfUser]),
840+ eg.dmMessage (from: users[4 - 1 ], to: [selfUser]),
841+ ]);
835842
836843 // Check the ranking of the full list of mentions,
837844 // i.e. the results for an empty query.
0 commit comments