add unit logic test