Discussion integration tests

Needs another review please...
Same comment which I had before. This code can be replaced with selenium wait api