Add unit tests for Edit and Add controllers