add test

2023-08-13 20:39:36 +07:00
parent 2889f35a3d
commit 1e3f5eb91d
2 changed files with 97 additions and 2 deletions
@@ -41,6 +41,98 @@ func Test_botAll_Process(t *testing.T) {
 			},
 			wantErr: false,
 		},
+		{
+			name: "message with all and empty users",
+			messenger: func(ctrl *gomock.Controller) messenger.IMessenger {
+				m := messenger_mocks.NewMockIMessenger(ctrl)
+				m.EXPECT().SendMessage(gomock.Any(), gomock.Any()).Times(0)
+				return m
+			},
+			storage: func(ctrl *gomock.Controller) storage.IStorage {
+				m := storage_mocks.NewMockIStorage(ctrl)
+				m.EXPECT().UpsertUser(gomock.Any(), storage.User{ChatID: "123", UserID: "username"}).Times(1)
+				m.EXPECT().
+					GetAllUsersByChatID(gomock.Any(), gomock.Any()).
+					Return([]storage.User{{UserID: "username"}}, nil).
+					Times(1)
+				return m
+			},
+			msg: messenger.Message{
+				ChatID: "123",
+				UserID: "username",
+				Text:   "hello @all",
+			},
+			wantErr: false,
+		},
+		{
+			name: "message with all and one user",
+			messenger: func(ctrl *gomock.Controller) messenger.IMessenger {
+				m := messenger_mocks.NewMockIMessenger(ctrl)
+				m.EXPECT().
+					SendMessage(
+						gomock.Any(),
+						&messenger.Message{
+							ChatID: "123",
+							Text:   "username2",
+						},
+					).
+					Times(1)
+				return m
+			},
+			storage: func(ctrl *gomock.Controller) storage.IStorage {
+				m := storage_mocks.NewMockIStorage(ctrl)
+				m.EXPECT().UpsertUser(gomock.Any(), storage.User{ChatID: "123", UserID: "username"}).Times(1)
+				m.EXPECT().
+					GetAllUsersByChatID(gomock.Any(), gomock.Any()).
+					Return([]storage.User{
+						{UserID: "username"},
+						{UserID: "username2"},
+					}, nil).
+					Times(1)
+				return m
+			},
+			msg: messenger.Message{
+				ChatID: "123",
+				UserID: "username",
+				Text:   "hello @all",
+			},
+			wantErr: false,
+		},
+		{
+			name: "message with all and one user",
+			messenger: func(ctrl *gomock.Controller) messenger.IMessenger {
+				m := messenger_mocks.NewMockIMessenger(ctrl)
+				m.EXPECT().
+					SendMessage(
+						gomock.Any(),
+						&messenger.Message{
+							ChatID: "123",
+							Text:   "username2 username3",
+						},
+					).
+					Times(1)
+				return m
+			},
+			storage: func(ctrl *gomock.Controller) storage.IStorage {
+				m := storage_mocks.NewMockIStorage(ctrl)
+				m.EXPECT().UpsertUser(gomock.Any(), storage.User{ChatID: "123", UserID: "username"}).Times(1)
+				m.EXPECT().
+					GetAllUsersByChatID(gomock.Any(), gomock.Any()).
+					Return([]storage.User{
+						{UserID: "username"},
+						{UserID: "username2"},
+						{UserID: "username3"},
+					}, nil).
+					Times(1)
+				return m
+			},
+			msg: messenger.Message{
+				ChatID: "123",
+				UserID: "username",
+				Text:   "hello @all",
+			},
+			wantErr: false,
+		},
 	}
 	for _, tt := range tests {
 		tt := tt