Adding tests to drafts plugin