Add tests for timeouts
