()
| 507 | } |
| 508 | |
| 509 | func (s *MethodTestSuite) TestChats() { |
| 510 | s.Run("AcquireChats", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { |
| 511 | arg := database.AcquireChatsParams{ |
| 512 | StartedAt: dbtime.Now(), |
| 513 | WorkerID: uuid.New(), |
| 514 | NumChats: 1, |
| 515 | } |
| 516 | chat := testutil.Fake(s.T(), faker, database.Chat{}) |
| 517 | dbm.EXPECT().AcquireChats(gomock.Any(), arg).Return([]database.Chat{chat}, nil).AnyTimes() |
| 518 | check.Args(arg).Asserts(rbac.ResourceChat, policy.ActionUpdate).Returns([]database.Chat{chat}) |
| 519 | })) |
| 520 | s.Run("DeleteAllChatQueuedMessages", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { |
| 521 | chat := testutil.Fake(s.T(), faker, database.Chat{}) |
| 522 | dbm.EXPECT().GetChatByID(gomock.Any(), chat.ID).Return(chat, nil).AnyTimes() |
| 523 | dbm.EXPECT().DeleteAllChatQueuedMessages(gomock.Any(), chat.ID).Return(nil).AnyTimes() |
| 524 | check.Args(chat.ID).Asserts(chat, policy.ActionUpdate).Returns() |
| 525 | })) |
| 526 | s.Run("ArchiveChatByID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { |
| 527 | chat := testutil.Fake(s.T(), faker, database.Chat{}) |
| 528 | dbm.EXPECT().GetChatByID(gomock.Any(), chat.ID).Return(chat, nil).AnyTimes() |
| 529 | dbm.EXPECT().ArchiveChatByID(gomock.Any(), chat.ID).Return([]database.Chat{chat}, nil).AnyTimes() |
| 530 | check.Args(chat.ID).Asserts(chat, policy.ActionUpdate).Returns([]database.Chat{chat}) |
| 531 | })) |
| 532 | s.Run("UnarchiveChatByID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { |
| 533 | chat := testutil.Fake(s.T(), faker, database.Chat{}) |
| 534 | dbm.EXPECT().GetChatByID(gomock.Any(), chat.ID).Return(chat, nil).AnyTimes() |
| 535 | dbm.EXPECT().UnarchiveChatByID(gomock.Any(), chat.ID).Return([]database.Chat{chat}, nil).AnyTimes() |
| 536 | check.Args(chat.ID).Asserts(chat, policy.ActionUpdate).Returns([]database.Chat{chat}) |
| 537 | })) |
| 538 | s.Run("LinkChatFiles", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { |
| 539 | chat := testutil.Fake(s.T(), faker, database.Chat{}) |
| 540 | arg := database.LinkChatFilesParams{ |
| 541 | ChatID: chat.ID, |
| 542 | MaxFileLinks: int32(codersdk.MaxChatFileIDs), |
| 543 | FileIds: []uuid.UUID{uuid.New()}, |
| 544 | } |
| 545 | dbm.EXPECT().GetChatByID(gomock.Any(), chat.ID).Return(chat, nil).AnyTimes() |
| 546 | dbm.EXPECT().LinkChatFiles(gomock.Any(), arg).Return(int32(0), nil).AnyTimes() |
| 547 | check.Args(arg).Asserts(chat, policy.ActionUpdate).Returns(int32(0)) |
| 548 | })) |
| 549 | s.Run("PinChatByID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { |
| 550 | chat := testutil.Fake(s.T(), faker, database.Chat{}) |
| 551 | dbm.EXPECT().GetChatByID(gomock.Any(), chat.ID).Return(chat, nil).AnyTimes() |
| 552 | dbm.EXPECT().PinChatByID(gomock.Any(), chat.ID).Return(nil).AnyTimes() |
| 553 | check.Args(chat.ID).Asserts(chat, policy.ActionUpdate).Returns() |
| 554 | })) |
| 555 | s.Run("UnpinChatByID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { |
| 556 | chat := testutil.Fake(s.T(), faker, database.Chat{}) |
| 557 | dbm.EXPECT().GetChatByID(gomock.Any(), chat.ID).Return(chat, nil).AnyTimes() |
| 558 | dbm.EXPECT().UnpinChatByID(gomock.Any(), chat.ID).Return(nil).AnyTimes() |
| 559 | check.Args(chat.ID).Asserts(chat, policy.ActionUpdate).Returns() |
| 560 | })) |
| 561 | s.Run("SoftDeleteChatMessagesAfterID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { |
| 562 | chat := testutil.Fake(s.T(), faker, database.Chat{}) |
| 563 | arg := database.SoftDeleteChatMessagesAfterIDParams{ |
| 564 | ChatID: chat.ID, |
| 565 | AfterID: 123, |
| 566 | } |
nothing calls this directly
no test coverage detected