def test_get_flags_for_site(self):
self.assertTrue(len(Flag.get_flags_for_site(self.site1)) == 0)
name1 = "foo"
Flag.objects.create(name=name1, everyone=True, site=self.site1)
self.assertEqual([name1], [f.name for f in Flag.get_flags_for_site(self.site1)])
# by default sample is sites-global
self.assertEqual([name1], [f.name for f in Flag.get_flags_for_site(self.site2)])
name2 = "bar"
Flag.objects.create(name=name2, everyone=True, site=self.site2, all_sites_override=False)
self.assertEqual({name1, name2}, set([f.name for f in Flag.get_flags_for_site(self.site2)]))
self.assertEqual([name1], [f.name for f in Flag.get_flags_for_site(self.site1)])
