Upgrade from Django 0.96 to Django 1.0.2.
Risk: high (framework change)
Visibility: medium
Signed-off-by: James Ren <jamesren@google.com>
git-svn-id: http://test.kernel.org/svn/autotest/trunk@3457 592f7852-d20e-0410-864c-8624ca9c26a4
diff --git a/frontend/afe/models_test.py b/frontend/afe/models_test.py
index e151df0..df9e964 100644
--- a/frontend/afe/models_test.py
+++ b/frontend/afe/models_test.py
@@ -6,6 +6,38 @@
from autotest_lib.frontend.afe import frontend_test_utils
from autotest_lib.frontend.afe import models
+
+class AclGroupTest(unittest.TestCase,
+ frontend_test_utils.FrontendTestMixin):
+ def setUp(self):
+ self._frontend_common_setup()
+
+
+ def tearDown(self):
+ self._frontend_common_teardown()
+
+
+ def _check_acls(self, host, acl_name_list):
+ actual_acl_names = [acl_group.name for acl_group
+ in host.aclgroup_set.all()]
+ self.assertEquals(set(actual_acl_names), set(acl_name_list))
+
+
+ def test_on_host_membership_change(self):
+ host1, host2 = self.hosts[1:3]
+ everyone_acl = models.AclGroup.objects.get(name='Everyone')
+
+ host1.aclgroup_set.clear()
+ self._check_acls(host1, [])
+ host2.aclgroup_set.add(everyone_acl)
+ self._check_acls(host2, ['Everyone', 'my_acl'])
+
+ models.AclGroup.on_host_membership_change()
+
+ self._check_acls(host1, ['Everyone'])
+ self._check_acls(host2, ['my_acl'])
+
+
class SpecialTaskUnittest(unittest.TestCase,
frontend_test_utils.FrontendTestMixin):
def setUp(self):