А вы прямо реально по TDD пишите? Сначала тесты, потом код или как-то параллельно это делаете ?
А про разгрузить код самого класса, вы имели в виду, вынесение вызов каких-нить сервисных функции, или прямо бизнес логику, между классом и декоратором поделить ?