Add unit tests to the service