[test] Added test cases for upcoming demo