salt/tests/unit/beacons/test_inotify.py

# coding: utf-8

# Python libs
from __future__ import absolute_import
import os
import shutil
import tempfile

# Salt libs
import salt.utils.files
from salt.beacons import inotify

# Salt testing libs
from tests.support.unit import skipIf, TestCase
from tests.support.mixins import LoaderModuleMockMixin
# Third-party libs
try:
    import pyinotify  # pylint: disable=unused-import
    HAS_PYINOTIFY = True
except ImportError:
    HAS_PYINOTIFY = False

import logging
log = logging.getLogger(__name__)


@skipIf(not HAS_PYINOTIFY, 'pyinotify is not available')
class INotifyBeaconTestCase(TestCase, LoaderModuleMockMixin):
    '''
    Test case for salt.beacons.inotify
    '''

    def setup_loader_modules(self):
        return {inotify: {}}

    def setUp(self):
        self.tmpdir = tempfile.mkdtemp()

    def tearDown(self):
        shutil.rmtree(self.tmpdir, ignore_errors=True)

    def test_empty_config(self):
        config = [{}]
        ret = inotify.beacon(config)
        self.assertEqual(ret, [])

    def test_file_open(self):
        path = os.path.realpath(__file__)
        config = [{'files': {path: {'mask': ['open']}}}]
        ret = inotify.validate(config)
        self.assertEqual(ret, (True, 'Valid beacon configuration'))

        ret = inotify.beacon(config)
        self.assertEqual(ret, [])

        with salt.utils.files.fopen(path, 'r') as f:
            pass
        ret = inotify.beacon(config)
        self.assertEqual(len(ret), 1)
        self.assertEqual(ret[0]['path'], path)
        self.assertEqual(ret[0]['change'], 'IN_OPEN')

    def test_dir_no_auto_add(self):
        config = [{'files': {self.tmpdir: {'mask': ['create']}}}]
        ret = inotify.validate(config)
        self.assertEqual(ret, (True, 'Valid beacon configuration'))

        ret = inotify.beacon(config)
        self.assertEqual(ret, [])
        fp = os.path.join(self.tmpdir, 'tmpfile')
        with salt.utils.files.fopen(fp, 'w') as f:
            pass
        ret = inotify.beacon(config)
        self.assertEqual(len(ret), 1)
        self.assertEqual(ret[0]['path'], fp)
        self.assertEqual(ret[0]['change'], 'IN_CREATE')
        with salt.utils.files.fopen(fp, 'r') as f:
            pass
        ret = inotify.beacon(config)
        self.assertEqual(ret, [])

    def test_dir_auto_add(self):
        config = [{'files': {self.tmpdir: {'mask': ['create', 'open'], 'auto_add': True}}}]
        ret = inotify.validate(config)
        self.assertEqual(ret, (True, 'Valid beacon configuration'))

        ret = inotify.beacon(config)
        self.assertEqual(ret, [])
        fp = os.path.join(self.tmpdir, 'tmpfile')
        with salt.utils.files.fopen(fp, 'w') as f:
            pass
        ret = inotify.beacon(config)
        self.assertEqual(len(ret), 2)
        self.assertEqual(ret[0]['path'], fp)
        self.assertEqual(ret[0]['change'], 'IN_CREATE')
        self.assertEqual(ret[1]['path'], fp)
        self.assertEqual(ret[1]['change'], 'IN_OPEN')
        with salt.utils.files.fopen(fp, 'r') as f:
            pass
        ret = inotify.beacon(config)
        self.assertEqual(len(ret), 1)
        self.assertEqual(ret[0]['path'], fp)
        self.assertEqual(ret[0]['change'], 'IN_OPEN')

    def test_dir_recurse(self):
        dp1 = os.path.join(self.tmpdir, 'subdir1')
        os.mkdir(dp1)
        dp2 = os.path.join(dp1, 'subdir2')
        os.mkdir(dp2)
        fp = os.path.join(dp2, 'tmpfile')
        with salt.utils.files.fopen(fp, 'w') as f:
            pass
        config = [{'files': {self.tmpdir: {'mask': ['open'], 'recurse': True}}}]
        ret = inotify.validate(config)
        self.assertEqual(ret, (True, 'Valid beacon configuration'))

        ret = inotify.beacon(config)
        self.assertEqual(ret, [])
        with salt.utils.files.fopen(fp) as f:
            pass
        ret = inotify.beacon(config)
        self.assertEqual(len(ret), 3)
        self.assertEqual(ret[0]['path'], dp1)
        self.assertEqual(ret[0]['change'], 'IN_OPEN|IN_ISDIR')
        self.assertEqual(ret[1]['path'], dp2)
        self.assertEqual(ret[1]['change'], 'IN_OPEN|IN_ISDIR')
        self.assertEqual(ret[2]['path'], fp)
        self.assertEqual(ret[2]['change'], 'IN_OPEN')

    def test_dir_recurse_auto_add(self):
        dp1 = os.path.join(self.tmpdir, 'subdir1')
        os.mkdir(dp1)
        config = [{'files': {self.tmpdir: {'mask': ['create', 'delete'],
                                           'recurse': True,
                                           'auto_add': True}}}]
        ret = inotify.validate(config)
        self.assertEqual(ret, (True, 'Valid beacon configuration'))

        ret = inotify.beacon(config)
        self.assertEqual(ret, [])
        dp2 = os.path.join(dp1, 'subdir2')
        os.mkdir(dp2)
        ret = inotify.beacon(config)
        self.assertEqual(len(ret), 1)
        self.assertEqual(ret[0]['path'], dp2)
        self.assertEqual(ret[0]['change'], 'IN_CREATE|IN_ISDIR')
        fp = os.path.join(dp2, 'tmpfile')
        with salt.utils.files.fopen(fp, 'w') as f:
            pass
        ret = inotify.beacon(config)
        self.assertEqual(len(ret), 1)
        self.assertEqual(ret[0]['path'], fp)
        self.assertEqual(ret[0]['change'], 'IN_CREATE')
        os.remove(fp)
        ret = inotify.beacon(config)
        self.assertEqual(len(ret), 1)
        self.assertEqual(ret[0]['path'], fp)
        self.assertEqual(ret[0]['change'], 'IN_DELETE')

    def test_multi_files_exclude(self):
        dp1 = os.path.join(self.tmpdir, 'subdir1')
        dp2 = os.path.join(self.tmpdir, 'subdir2')
        os.mkdir(dp1)
        os.mkdir(dp2)
        _exclude1 = '{0}/subdir1/*tmpfile*$'.format(self.tmpdir)
        _exclude2 = '{0}/subdir2/*filetmp*$'.format(self.tmpdir)
        config = [{'files': {dp1: {'mask': ['create', 'delete'],
                                   'recurse': True,
                                   'exclude': [{_exclude1: {'regex': True}}],
                                   'auto_add': True}}},
                  {'files': {dp2: {'mask': ['create', 'delete'],
                                   'recurse': True,
                                   'exclude': [{_exclude2: {'regex': True}}],
                                   'auto_add': True}}}]
        ret = inotify.validate(config)
        self.assertEqual(ret, (True, 'Valid beacon configuration'))

        fp = os.path.join(dp1, 'tmpfile')
        with salt.utils.files.fopen(fp, 'w') as f:
            pass
        ret = inotify.beacon(config)
        self.assertEqual(len(ret), 0)
        os.remove(fp)
        ret = inotify.beacon(config)
        self.assertEqual(len(ret), 0)

        fp = os.path.join(dp2, 'tmpfile')
        with salt.utils.files.fopen(fp, 'w') as f:
            pass
        ret = inotify.beacon(config)
        self.assertEqual(len(ret), 1)
        self.assertEqual(ret[0]['path'], fp)
        self.assertEqual(ret[0]['change'], 'IN_CREATE')
        os.remove(fp)
        ret = inotify.beacon(config)
        self.assertEqual(len(ret), 1)
        self.assertEqual(ret[0]['path'], fp)
        self.assertEqual(ret[0]['change'], 'IN_DELETE')