Add tests for model's mark_complete method