Предложение такое, поставить на время теста манекены, что бы было удобней тестить классовую механику. по окончании теста убрать ибо не близлайк.