Added more tests under infrastructure for ui bench testing.