Код работает отлично, только вот интеграционных тестов не было.