Python ConnTrackCollector Examples

Example #1

    def setUp(self):
        config = get_collector_config('ConnTrackCollector', {
            'interval': 10,
            'bin': 'true',
        })

        self.collector = ConnTrackCollector(config, None)

Example #2

File: testconntrack.py Project: Jim-Holmstroem/Diamond

    def setUp(self):
        config = get_collector_config('ConnTrackCollector', {
            'interval': 10,
            'bin': 'true',
            'dir': self.getFixtureDirPath(),
        })

        self.collector = ConnTrackCollector(config, None)

Example #3

File: testconntrack.py Project: microsigns/Diamond

    def setUp(self):
        config = get_collector_config('ConnTrackCollector', {
            'interval': 10,
            'bin': 'true',
        })

        self.collector = ConnTrackCollector(config, None)

Example #4

File: testconntrack.py Project: MechanisM/Diamond

    def setUp(self):
        config = get_collector_config('ConnTrackCollector', {
            'interval': 10
        })

        self.collector = ConnTrackCollector(config, None)
        self.collector.COMMAND[0] = 'true'

Example #5

File: testconntrack.py Project: CaesarC/Diamond

    def setUp(self):
        config = get_collector_config('ConnTrackCollector', {
            'interval': 10,
            'bin': 'true',
            'dir': self.getFixtureDirPath(),
        })

        self.collector = ConnTrackCollector(config, None)

Example #6

File: testconntrack.py Project: martindk80/Diamond

class TestConnTrackCollector(CollectorTestCase):
    def setUp(self):
        config = get_collector_config('ConnTrackCollector', {'interval': 10})

        self.collector = ConnTrackCollector(config, None)
        self.collector.COMMAND[0] = 'true'

    @patch('os.access', Mock(return_value=True))
    @patch.object(Collector, 'publish')
    def test_should_work_with_synthetic_data(self, publish_mock):
        with patch(
                'subprocess.Popen.communicate',
                Mock(return_value=('net.netfilter.nf_conntrack_count = 33',
                                   ''))):
            self.collector.collect()

        self.assertPublishedMany(publish_mock, {'nf_conntrack_count': 33.0})

    @patch('os.access', Mock(return_value=True))
    @patch.object(Collector, 'publish')
    def test_should_fail_gracefully(self, publish_mock):
        with patch(
                'subprocess.Popen.communicate',
                Mock(return_value=(
                    'sysctl: cannot stat /proc/sys/net/netfilter/nf_conntrack_count: No such file or directory',
                    ''))):
            self.collector.collect()

        self.assertPublishedMany(publish_mock, {})

    @patch('os.access', Mock(return_value=False))
    @patch.object(Collector, 'publish')
    def test_should_fail_gracefully_2(self, publish_mock):
        self.collector.collect()
        self.assertPublishedMany(publish_mock, {})

Example #7

File: testconntrack.py Project: 50onRed/Diamond

class TestConnTrackCollector(CollectorTestCase):
    def setUp(self):
        config = get_collector_config('ConnTrackCollector', {
            'interval': 10,
            'bin': 'true',
        })

        self.collector = ConnTrackCollector(config, None)

    def test_import(self):
        self.assertTrue(ConnTrackCollector)

    @patch('os.access', Mock(return_value=True))
    @patch.object(Collector, 'publish')
    def test_should_work_with_synthetic_data(self, publish_mock):
        patch_communicate = patch(
            'subprocess.Popen.communicate',
            Mock(return_value=(
                'net.netfilter.nf_conntrack_count = 33',
                '')))

        patch_communicate.start()
        self.collector.collect()
        patch_communicate.stop()

        metrics = {
            'nf_conntrack_count': 33.0
        }

        self.setDocExample(collector=self.collector.__class__.__name__,
                           metrics=metrics,
                           defaultpath=self.collector.config['path'])
        self.assertPublishedMany(publish_mock, metrics)

    @patch('os.access', Mock(return_value=True))
    @patch.object(Collector, 'publish')
    def test_should_fail_gracefully(self, publish_mock):
        patch_communicate = patch(
            'subprocess.Popen.communicate',
            Mock(
                return_value=(
                    'sysctl: cannot stat /proc/sys/net/net'
                    + 'filter/nf_conntrack_count: '
                    + 'No such file or directory', '')))

        patch_communicate.start()
        self.collector.collect()
        patch_communicate.stop()

        self.assertPublishedMany(publish_mock, {})

    @patch('os.access', Mock(return_value=False))
    @patch.object(Collector, 'publish')
    def test_should_fail_gracefully_2(self, publish_mock):
        self.collector.collect()
        self.assertPublishedMany(publish_mock, {})

Example #8

File: testconntrack.py Project: tipbit/Diamond

class TestConnTrackCollector(CollectorTestCase):
    def setUp(self):
        config = get_collector_config('ConnTrackCollector', {
            'interval': 10,
            'bin': 'true',
        })

        self.collector = ConnTrackCollector(config, None)

    def test_import(self):
        self.assertTrue(ConnTrackCollector)

    @patch('os.access', Mock(return_value=True))
    @patch.object(Collector, 'publish')
    def test_should_work_with_synthetic_data(self, publish_mock):
        patch_communicate = patch(
            'subprocess.Popen.communicate',
            Mock(return_value=('net.netfilter.nf_conntrack_count = 33', '')))

        patch_communicate.start()
        self.collector.collect()
        patch_communicate.stop()

        metrics = {'nf_conntrack_count': 33.0}

        self.setDocExample(collector=self.collector.__class__.__name__,
                           metrics=metrics,
                           defaultpath=self.collector.config['path'])
        self.assertPublishedMany(publish_mock, metrics)

    @patch('os.access', Mock(return_value=True))
    @patch.object(Collector, 'publish')
    def test_should_fail_gracefully(self, publish_mock):
        patch_communicate = patch(
            'subprocess.Popen.communicate',
            Mock(return_value=('sysctl: cannot stat /proc/sys/net/net' +
                               'filter/nf_conntrack_count: ' +
                               'No such file or directory', '')))

        patch_communicate.start()
        self.collector.collect()
        patch_communicate.stop()

        self.assertPublishedMany(publish_mock, {})

    @patch('os.access', Mock(return_value=False))
    @patch.object(Collector, 'publish')
    def test_should_fail_gracefully_2(self, publish_mock):
        self.collector.collect()
        self.assertPublishedMany(publish_mock, {})

Example #9

File: testconntrack.py Project: pythianjoseph/Diamond

class TestConnTrackCollector(CollectorTestCase):
    def setUp(self):
        config = get_collector_config(
            "ConnTrackCollector", {"interval": 10, "bin": "true", "dir": self.getFixtureDirPath()}
        )

        self.collector = ConnTrackCollector(config, None)

    def test_import(self):
        self.assertTrue(ConnTrackCollector)

    @patch.object(Collector, "publish")
    def test_should_work_with_synthetic_data(self, publish_mock):
        self.collector.collect()

        metrics = {"ip_conntrack_count": 33.0, "ip_conntrack_max": 36.0}

        self.setDocExample(
            collector=self.collector.__class__.__name__, metrics=metrics, defaultpath=self.collector.config["path"]
        )
        self.assertPublishedMany(publish_mock, metrics)

    @patch("os.access", Mock(return_value=True))
    @patch.object(Collector, "publish")
    def test_should_fail_gracefully(self, publish_mock):
        patch_communicate = patch(
            "subprocess.Popen.communicate",
            Mock(
                return_value=(
                    "sysctl: cannot stat /proc/sys/net/net"
                    + "filter/nf_conntrack_count: "
                    + "No such file or directory",
                    "",
                )
            ),
        )

        patch_communicate.start()
        self.collector.collect()
        patch_communicate.stop()

        self.assertPublishedMany(publish_mock, {})

    @patch("os.access", Mock(return_value=False))
    @patch.object(Collector, "publish")
    def test_should_fail_gracefully_2(self, publish_mock):
        self.collector.collect()
        self.assertPublishedMany(publish_mock, {})

Example #10

File: testconntrack.py Project: philipcristiano/Diamond

class TestConnTrackCollector(CollectorTestCase):
    def setUp(self):
        config = get_collector_config('ConnTrackCollector', {
            'interval': 10,
            'bin': 'true',
        })

        self.collector = ConnTrackCollector(config, None)

    @patch('os.access', Mock(return_value=True))
    @patch.object(Collector, 'publish')
    def test_should_work_with_synthetic_data(self, publish_mock):
        with patch('subprocess.Popen.communicate', Mock(return_value =
            ( 'net.netfilter.nf_conntrack_count = 33' , '')
        )):
            self.collector.collect()
            
        self.assertPublishedMany(publish_mock, {
            'nf_conntrack_count' : 33.0
        })

    @patch('os.access', Mock(return_value=True))
    @patch.object(Collector, 'publish')
    def test_should_fail_gracefully(self, publish_mock):
        with patch('subprocess.Popen.communicate', Mock(return_value =
            ( 'sysctl: cannot stat /proc/sys/net/netfilter/nf_conntrack_count: No such file or directory' , '')
        )):
            self.collector.collect()

        self.assertPublishedMany(publish_mock, {})

    @patch('os.access', Mock(return_value=False))
    @patch.object(Collector, 'publish')
    def test_should_fail_gracefully_2(self, publish_mock):
        self.collector.collect()
        self.assertPublishedMany(publish_mock, {})

Example #11

File: testconntrack.py Project: martindk80/Diamond

    def setUp(self):
        config = get_collector_config('ConnTrackCollector', {'interval': 10})

        self.collector = ConnTrackCollector(config, None)
        self.collector.COMMAND[0] = 'true'

Example #12

File: testconntrack.py Project: pythianjoseph/Diamond

    def setUp(self):
        config = get_collector_config(
            "ConnTrackCollector", {"interval": 10, "bin": "true", "dir": self.getFixtureDirPath()}
        )

        self.collector = ConnTrackCollector(config, None)