[Stress test] Make fail case handle more flexible

[hugh712]

Currently in stress test the fail handle cases are hard code in the code snippet,
we should make it more flexible and easier to extend

[hugh712]

fixed in #11