AUI-1092 - Add tests