add generate unit test