Method TestWithUserIDFilter

database/watcher/watcher_test.go:1182–1239

()

}

func (s *WatcherTestSuite) TestWithUserIDFilter() {
	producer, err := watcher.RegisterProducer(s.ctx, "test-producer")
	s.Require().NoError(err)
	s.Require().NotNil(producer)

	userID, err := uuid.NewUUID()
	s.Require().NoError(err)

	userID2, err := uuid.NewUUID()
	s.Require().NoError(err)

	consumer, err := watcher.RegisterConsumer(
		s.ctx, "test-consumer",
		watcher.WithUserIDFilter(userID.String()),
	)
	s.Require().NoError(err)
	s.Require().NotNil(consumer)
	consumeEvents(consumer)

	payload := common.ChangePayload{
		EntityType: common.UserEntityType,
		Operation:  common.UpdateOperation,
		Payload: params.User{
			ID: userID.String(),
		},
	}
	err = producer.Notify(payload)
	s.Require().NoError(err)

	receivedPayload := waitForPayload(consumer.Watch(), 100*time.Millisecond)
	s.Require().NotNil(receivedPayload)
	s.Require().Equal(payload, *receivedPayload)

	payload = common.ChangePayload{
		EntityType: common.UserEntityType,
		Operation:  common.UpdateOperation,
		Payload: params.User{
			ID: userID2.String(),
		},
	}

	err = producer.Notify(payload)
	s.Require().NoError(err)
	receivedPayload = waitForPayload(consumer.Watch(), 100*time.Millisecond)
	s.Require().Nil(receivedPayload)

	payload = common.ChangePayload{
		EntityType: common.UserEntityType,
		Operation:  common.UpdateOperation,
		// Declare as user, but payload is a pool. Filter should return false.
		Payload: params.Pool{},
	}

	err = producer.Notify(payload)
	s.Require().NoError(err)
	receivedPayload = waitForPayload(consumer.Watch(), 100*time.Millisecond)
	s.Require().Nil(receivedPayload)
}