Updated UT for flow api

Change-Id: I8600f0fac69d7e3b52655e6e8f2f98e193c3c9b9
Signed-off-by: ananthip <ananthip@hcl.com>