Add test for non-generation of draft.

- Refactor later.