Спасибо, но я уже решил проблему, тесты заработали, сейчас доведу код до ума и выложу сюда, возможно кому-то пригодится.