AUI-1444 - add unit tests