Results: improve admin page

Change-Id: I2db0196ca6e79c29157e49b7496a13badd189fab

1 files changed, 40 insertions, 1 deletions

diff --git a/lava_results_app/admin.py b/lava_results_app/admin.py
index d941ddb78..1b7bdca19 100644
--- a/lava_results_app/admin.py
+++ b/lava_results_app/admin.py
@@ -23,11 +23,50 @@ Administration interface of the LAVA Results application.
 from django.contrib import admin
 
 from lava_results_app.models import (
-    Query
+    ActionData,
+    Query,
+    TestCase,
+    TestSet,
+    TestSuite,
 )
 
+class ActionDataAdmin(admin.ModelAdmin):
+    list_display = ('job_pk', 'action_level', 'action_name')
+    ordering = ('-testdata__testjob__pk', '-action_level', )
+
+    def job_pk(self, action):
+        return action.testdata.testjob.pk
+
 
 class QueryAdmin(admin.ModelAdmin):
     save_as = True
 
+
+class TestCaseAdmin(admin.ModelAdmin):
+    list_display = ('job_pk', 'suite_name', 'name', 'result')
+    ordering = ('-suite__job__pk', 'suite__name', 'name')
+
+    def job_pk(self, testcase):
+        return testcase.suite.job.pk
+
+    def suite_name(self, testcase):
+        return testcase.suite.name
+
+
+class TestSetAdmin(admin.ModelAdmin):
+    list_display = ('suite', 'name')
+
+
+class TestSuiteAdmin(admin.ModelAdmin):
+    list_display = ('job_pk', 'name')
+    ordering = ('-job__pk', 'name')
+
+    def job_pk(self, testsuite):
+        return testsuite.job.pk
+
+
+admin.site.register(ActionData, ActionDataAdmin)
 admin.site.register(Query, QueryAdmin)
+admin.site.register(TestCase, TestCaseAdmin)
+admin.site.register(TestSet, TestSetAdmin)
+admin.site.register(TestSuite, TestSuiteAdmin)