add testplan flag for show in total chart

cdws_api/tests.py

@@ -138,7 +138,8 @@ def _create_testplan(self, name, project_id,
     def _update_testplan(self, testplan_id, name,
                          hidden=None, main=None,
                          statistic_filter=None, description=None,
-                         variable_name=None, variable_value_regexp=None):
+                         variable_name=None, variable_value_regexp=None,
+                         summary=None):
         data = {'name': name}
         if main is not None:
             data['main'] = main
@@ -152,6 +153,8 @@ def _update_testplan(self, testplan_id, name,
             data['variable_name'] = variable_name
         if variable_value_regexp is not None:
             data['variable_value_regexp'] = variable_value_regexp
+        if summary is not None:
+            data['summary'] = summary
         return self._call_rest('patch',
                                'testplans/{0}/'.format(testplan_id), data)
 
@@ -194,9 +197,14 @@ def test_hidden_flag(self):
 
     def test_main_flag(self):
         project = Project.objects.get(name='DummyTestProject')
-        data = self._create_testplan('HiddenByDefaultTrue', project.id)
+        data = self._create_testplan('MainByDefaultFalse', project.id)
         self.assertFalse(data['main'])
 
+    def test_summary_flag(self):
+        project = Project.objects.get(name='DummyTestProject')
+        data = self._create_testplan('SummaryByDefaultFalse', project.id)
+        self.assertFalse(data['summary'])
+
     def test_statistic_filter(self):
         testplan = TestPlan.objects.get(name='DummyTestPlan')
         self.assertEquals(testplan.filter, '')
@@ -265,6 +273,14 @@ def test_update_main_flag(self):
         data = self._update_testplan(testplan.id, testplan.name, main=False)
         self.assertFalse(data['main'])
 
+    def test_update_summary_flag(self):
+        testplan = TestPlan.objects.get(name='DummyTestPlan')
+        self.assertFalse(testplan.summary)
+        data = self._update_testplan(testplan.id, testplan.name, summary=True)
+        self.assertTrue(data['summary'])
+        data = self._update_testplan(testplan.id, testplan.name, summary=False)
+        self.assertFalse(data['summary'])
+
     def test_update_hidden_flag(self):
         testplan = TestPlan.objects.get(name='DummyTestPlan')
         self.assertTrue(testplan.hidden)

testreport/migrations/0035_testplan_summary.py

@@ -0,0 +1,20 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from django.db import models, migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('testreport', '0034_extuser_dashboards'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='testplan',
+            name='summary',
+            field=models.BooleanField(verbose_name='Use for total chart', default=False),
+            preserve_default=True,
+        ),
+    ]

testreport/models.py

@@ -55,6 +55,8 @@ class TestPlan(models.Model):
     variable_value_regexp = models.CharField(_('Regexp for variable value'),
                                              max_length=255, default='',
                                              blank=True)
+    summary = models.BooleanField(_('Use for total chart'),
+                                  blank=True, null=False, default=False)
 
     def __str__(self):
         return '{0} -> TestPlan: {1}'.format(self.project, self.name)