Added tests for DraftClassRepository