Android: Instrumentation tests in gradle

build/android/gradle/android.jinja

 {# Copyright 2016 The Chromium Authors. All rights reserved. #}
 {# Use of this source code is governed by a BSD-style license that can be #}
 {# found in the LICENSE file. #}
+{% macro expand_sourceset(variables, prefix) %}
+{% if variables is defined %}
+        {{ prefix }} {
+            manifest.srcFile "{{ variables.android_manifest }}"
+            java.srcDirs = [
+{% for path in variables.java_dirs %}
+                "{{ path }}",
+{% endfor %}
+            ]
+            jniLibs.srcDirs = [
+{% for path in variables.jni_libs %}
+                "{{ path }}",
+{% endfor %}
+            ]
+        }
+{% endif %}
+{% endmacro %}
 // Generated by //build/android/generate_gradle.py
 
 {% if template_type in ('android_library', 'android_junit') %}
 apply plugin: "com.android.library"
 {% elif template_type == 'android_apk' %}
 apply plugin: "com.android.application"
 {% endif %}
 
 android {
     compileSdkVersion {{ compile_sdk_version }}
@@ skipping 11 matching lines @@
             aidl.srcDirs = []
             assets.srcDirs = []
             java.srcDirs = []
             jni.srcDirs = []
             renderscript.srcDirs = []
             res.srcDirs = []
             resources.srcDirs = []
         }
 {% endfor %}
 
-        main.manifest.srcFile "{{ android_manifest }}"
-
-        {{ sourceSetName }} {
-            java.srcDirs = [
-{% for path in java_dirs %}
-                "{{ path }}",
-{% endfor %}
-            ]
-            jniLibs.srcDirs = [
-{% for path in jni_libs %}
-                "{{ path }}",
-{% endfor %}
-            ]
-        }
+{{ expand_sourceset(main, 'main') }}
+{{ expand_sourceset(test, 'test') }}
+{{ expand_sourceset(android_test, 'androidTest') }}
     }
 }
 
 {% include 'dependencies.jinja' %}
 
 android.variantFilter { variant ->
     if (variant.buildType.name.equals('release')) {
         variant.setIgnore(true);
     }
 }
 
 afterEvaluate {
     def tasksToDisable = tasks.findAll {
         return (it.name.equals('generateDebugSources')  // causes unwanted AndroidManifest.java
                 || it.name.equals('generateReleaseSources')
                 || it.name.endsWith('Assets')
                 || it.name.endsWith('BuildConfig')  // causes unwanted BuildConfig.java
 {% if not use_gradle_process_resources %}
                 || it.name.endsWith('Resources')
                 || it.name.endsWith('ResValues')
 {% endif %}
                 || it.name.endsWith('Aidl')
                 || it.name.endsWith('Renderscript')
                 || it.name.endsWith('Shaders'))
     }
     tasksToDisable.each { Task task ->
       task.enabled = false
     }
 }