Unit test

Signed-off-by: OwenPatey <owen_patey@hotmail.com>