Integration test with Android gradle DSL

I'm trying to have a separate folder and configuration for integration tests in the same fashion as this:

http://ift.tt/2guicyQ

The problem with that is it relies on the Java plugin being used in the build graph, not the Android one. The android plugin does the configurations in it's own way and can't figure out how to make this work for and Android project.