add unit integration tests