Unit testy pro staged handler