Add tests for worker class.