Add tests for cluster operations