old_contents: # pylint: disable=unused-import
  import os
  from six import PY3
  if PY3:
      from unittest.mock import patch, ANY
  else:
      from mock import patch, ANY
  
  import unittest
  
  import blivet
  import blivet.formats.fs as fs
  
  
  class SELinuxContextTestCase(unittest.TestCase):
      """Testing SELinux contexts.
      """
  
      def setUp(self):
          if not blivet.flags.flags.selinux:
              self.skipTest("SELinux disabled.")
          self.selinux_reset_fcon = blivet.flags.flags.selinux_reset_fcon
          self.selinux = blivet.flags.selinux
          super(SELinuxContextTestCase, self).setUp()
          self.addCleanup(self._clean_up)
  
      @patch("blivet.util.mount", return_value=0)
      @patch.object(fs.FS, "_pre_setup", return_value=True)
      @patch("os.access", return_value=True)
      # pylint: disable=unused-argument
      # pylint: disable=no-self-use
      def exec_mount_selinux_format(self, formt, *args):
          """ Test of correct selinux context parameter value when mounting """
  
          lost_found_context = "system_u:object_r:lost_found_t:s0"
          blivet.flags.selinux = True
          fmt = formt()
  
          import selinux
          # Patch selinux context setting
          with patch("selinux.lsetfilecon") as lsetfilecon:
              lsetfilecon.return_value = True
  
              blivet.flags.flags.selinux_reset_fcon = True
              fmt.setup(mountpoint="dummy")  # param needed to pass string check
              if isinstance(fmt, fs.Ext2FS):
                  lsetfilecon.assert_called_with(ANY, lost_found_context)
              else:
                  lsetfilecon.assert_not_called()
  
              lsetfilecon.reset_mock()
  
              blivet.flags.flags.selinux_reset_fcon = False
              fmt.setup(mountpoint="dummy")  # param needed to pass string check
              lsetfilecon.assert_not_called()
  
      def test_mount_selinux_ext2fs(self):
          """ Test of correct selinux context parameter value when mounting ext2"""
          self.exec_mount_selinux_format(fs.Ext2FS)
  
      def test_mount_selinux_xfs(self):
          """ Test of correct selinux context parameter value when mounting XFS"""
          self.exec_mount_selinux_format(fs.XFS)
  
      def _clean_up(self):
          blivet.flags.flags.selinux_reset_fcon = self.selinux_reset_fcon
          blivet.flags.selinux = self.selinux
  
new_contents: # pylint: disable=unused-import
  import os
  from six import PY3
  if PY3:
      from unittest.mock import patch, ANY
  else:
      from mock import patch, ANY
  
  import unittest
  
  import blivet
  import blivet.formats.fs as fs
  
  
  class SELinuxContextTestCase(unittest.TestCase):
      """Testing SELinux contexts.
      """
  
      def setUp(self):
          if not blivet.flags.flags.selinux:
              self.skipTest("SELinux disabled.")
          self.selinux_reset_fcon = blivet.flags.flags.selinux_reset_fcon
          self.selinux = blivet.flags.flags.selinux
          super(SELinuxContextTestCase, self).setUp()
          self.addCleanup(self._clean_up)
  
      @patch("blivet.util.mount", return_value=0)
      @patch.object(fs.FS, "_pre_setup", return_value=True)
      @patch("os.access", return_value=True)
      # pylint: disable=unused-argument
      # pylint: disable=no-self-use
      def exec_mount_selinux_format(self, formt, *args):
          """ Test of correct selinux context parameter value when mounting """
  
          lost_found_context = "system_u:object_r:lost_found_t:s0"
          blivet.flags.flags.selinux = True
          fmt = formt()
  
          import selinux
          # Patch selinux context setting
          with patch("selinux.lsetfilecon") as lsetfilecon:
              lsetfilecon.return_value = True
  
              blivet.flags.flags.selinux_reset_fcon = True
              fmt.setup(mountpoint="dummy")  # param needed to pass string check
              if isinstance(fmt, fs.Ext2FS):
                  lsetfilecon.assert_called_with(ANY, lost_found_context)
              else:
                  lsetfilecon.assert_not_called()
  
              lsetfilecon.reset_mock()
  
              blivet.flags.flags.selinux_reset_fcon = False
              fmt.setup(mountpoint="dummy")  # param needed to pass string check
              lsetfilecon.assert_not_called()
  
      def test_mount_selinux_ext2fs(self):
          """ Test of correct selinux context parameter value when mounting ext2"""
          self.exec_mount_selinux_format(fs.Ext2FS)
  
      def test_mount_selinux_xfs(self):
          """ Test of correct selinux context parameter value when mounting XFS"""
          self.exec_mount_selinux_format(fs.XFS)
  
      def _clean_up(self):
          blivet.flags.flags.selinux_reset_fcon = self.selinux_reset_fcon
          blivet.flags.flags.selinux = self.selinux
  
subject: Fix setting SELinux flag in SELinuxContextTestCase