Adding a specific tests set for ubuntu-desktop

Audio, bluetooth and graphics should only run at the ubuntu-desktop
image, as nano would require a bunch of extra packages to be installed.

The desktop test set should also cover the ones used and executed at
nano, with additional ones on top.

Signed-off-by: Ricardo Salveti de Araujo <ricardo.salveti@linaro.org>

 2 files changed, 20 insertions(+), 9 deletions(-)
diff --git a/lava-submit.py b/lava-submit.py
index fb84104..4ae4a89 100755
--- a/lava-submit.py
+++ b/lava-submit.py
@@ -18,6 +18,19 @@
     'beaglexm': 'beaglexm',
 }
 
+tests_nano = [
+    'pwrmgmt',
+    'gatortests',
+    'perf',
+    'wifi-enablement',
+]
+
+tests_desktop = [
+    'e2eaudiotest',
+    'bluetooth-enablement',
+    'leb-basic-graphics',
+]
+
 def get_install_action(tests):
     return {
        'command': 'lava_test_install',
@@ -77,16 +90,14 @@
                    help='URL of image')
     p.add_argument('-d', dest='device_type', required=True,
                    help='The device type to execute on. ie "panda"')
+    p.add_argument('-t', dest='image_type', default='nano',
+                   help='The image type to execute on, nano (default) or ubuntu-desktop')
     args = p.parse_args()
 
-    tests = [
-        'pwrmgmt',
-        'gatortests',
-        'perf',
-        'e2eaudiotest',
-        'bluetooth-enablement',
-        'wifi-enablement',
-    ]
+    if args.image_type == 'ubuntu-desktop'
+        tests = tests_nano + tests_desktop
+    else:
+        tests = tests_nano
 
     job = get_job(args.job_name, args.device_type, args.img_url, tests)