avocado-framework/avocado

import os
import tempfile
import unittest

from avocado.utils.filelock import AlreadyLocked, FileLock
from selftests.utils import temp_dir_prefix


class TestFileLock(unittest.TestCase):
    def setUp(self):
        prefix = temp_dir_prefix(self)
        self.tmpdir = tempfile.TemporaryDirectory(prefix=prefix)
        self.filename = os.path.join(self.tmpdir.name, "file.img")
        self.content = "Foo bar"
        with open(self.filename, "w", encoding="utf-8") as f:
            f.write(self.content)

    def _readfile(self):
        with FileLock(self.filename):
            with open(self.filename, "r", encoding="utf-8") as f:
                return f.read()

    def test_readfile(self):
        self.assertEqual(self._readfile(), self.content)

    def test_locked_by_me(self):
        with FileLock(self.filename):
            self.assertRaises(AlreadyLocked, self._readfile)

    def test_locked_by_other(self):
        with open(self.filename + ".lock", "w", encoding="utf-8") as f:
            f.write("1")
        self.assertRaises(AlreadyLocked, self._readfile)

    def tearDown(self):
        self.tmpdir.cleanup()


if __name__ == "__main__":
    unittest.main()