Add tests for mark_complete() method