Add test for the AJAX function