Yes, it should be OK to call it from the UI thread. But I think maybe it's better to not restrict the function so much to be called in UI thread as it wouldn't have the thread safety issue. It is used everywhere in the test and we should make it easy to be called. Make sense?