Add customer unit tests