Add test to assert return value from job