Add test for IfAction