Add tests for the worker