CDAP-15: added usage of check state method in existing unit-test