Add test for Omni mode.

Signed-off-by: Blaž Hrastnik <blaz.hrast@gmail.com>