vm: fix test_flush()

circle/vm/tests/test_models.py

@@ -375,6 +375,7 @@ class InstanceActivityTestCase(TestCase):
         node = MagicMock(spec=Node, enabled=True)
         node = MagicMock(spec=Node, enabled=True)
         node.instance_set.all.return_value = insts
         node.instance_set.all.return_value = insts
         user = MagicMock(spec=User)
         user = MagicMock(spec=User)
+        user.is_superuser = MagicMock(return_value=True)
         flush_op = FlushOperation(node)
         flush_op = FlushOperation(node)
 
 
         with patch.object(FlushOperation, 'create_activity') as create_act:
         with patch.object(FlushOperation, 'create_activity') as create_act:
@@ -386,6 +387,7 @@ class InstanceActivityTestCase(TestCase):
             node.disable.assert_called_with(user, act)
             node.disable.assert_called_with(user, act)
             for i in insts:
             for i in insts:
                 i.migrate.assert_called()
                 i.migrate.assert_called()
+        user.is_superuser.assert_called()
 
 
     def test_flush_disabled_wo_user(self):
     def test_flush_disabled_wo_user(self):
         insts = [MagicMock(spec=Instance, migrate=MagicMock()),
         insts = [MagicMock(spec=Instance, migrate=MagicMock()),