update tests for new gomega version

Signed-off-by: Simon Leung <leungs@us.ibm.com>