Add test for assignTask