Add test for new action
Add tests for edit action