Add integration tests support