- setup state using Gitlab::Database::Partitioning.registered_models
- move the responsibility to ensure that the registered models have
  partitioning strategy to register_models
- make sure we do not leak state outside tests

https://gitlab.com/gitlab-org/gitlab/-/issues/343955