Skip to content

Commit

Permalink
tests.py: cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
vasiliyk committed Apr 30, 2024
1 parent 87c02cb commit 9f47278
Showing 1 changed file with 20 additions and 19 deletions.
39 changes: 20 additions & 19 deletions tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,17 +18,16 @@
from stat import S_ISDIR, S_ISREG, ST_MODE
import unittest
import subprocess
import time
import signal

import mocks
import tcollector
import json
import threading
try:
import flask
except ImportError:
flask = None
import mocks
import tcollector


PY3 = sys.version_info[0] > 2

Expand All @@ -48,10 +47,9 @@ def test_bool_false_converted_int(self):
This can happen if a specific collector is buggy.
"""
thread = tcollector.ReaderThread(1, 10, True) # pylint:disable=no-member
collector = tcollector.Collector("c", 1, "c") # pylint:disable=no-member
line= "mymetric 123 False a=b"
expected = "mymetric 123 0 a=b"
thread = tcollector.ReaderThread(1, 10, True) # pylint:disable=no-member
collector = tcollector.Collector("c", 1, "c") # pylint:disable=no-member
line = "mymetric 123 False a=b"
thread.process_line(collector, line)
self.assertEqual(thread.readerq.qsize(), 1, line)
self.assertEqual(thread.readerq.get(), line)
Expand All @@ -63,10 +61,9 @@ def test_bool_true_converted_int(self):
This can happen if a specific collector is buggy.
"""
thread = tcollector.ReaderThread(1, 10, True) # pylint:disable=no-member
collector = tcollector.Collector("c", 1, "c") # pylint:disable=no-member
line= "mymetric 123 True a=b"
expected = "mymetric 123 1 a=b"
thread = tcollector.ReaderThread(1, 10, True) # pylint:disable=no-member
collector = tcollector.Collector("c", 1, "c") # pylint:disable=no-member
line = "mymetric 123 True a=b"
thread.process_line(collector, line)
self.assertEqual(thread.readerq.qsize(), 1, line)
self.assertEqual(thread.readerq.get(), line)
Expand Down Expand Up @@ -103,7 +100,7 @@ def test_ok_lines(self):
class CollectorsTests(unittest.TestCase):

def test_collectorsAccessRights(self):
"""Test of collectors access rights, permissions should be 0100775."""
"""Test of collectors access rights, permissions should be 0100755."""

def check_access_rights(top):
for f in os.listdir(top):
Expand All @@ -117,9 +114,9 @@ def check_access_rights(top):
# file, check permissions
permissions = oct(os.stat(pathname)[ST_MODE])
if PY3:
self.assertEqual("0o100775", permissions)
self.assertEqual("0o100755", permissions)
else:
self.assertEqual("0100775", permissions)
self.assertEqual("0100755", permissions)
else:
# unknown file type
pass
Expand Down Expand Up @@ -216,6 +213,7 @@ def test_bad_messages(self):
sender = self.send_query_with_response_code(400)
self.assertEqual(len(sender.sendq), 0)


class NamespacePrefixTests(unittest.TestCase):
"""Tests for metric namespace prefix."""

Expand All @@ -229,6 +227,7 @@ def test_prefix_added(self):
self.assertEqual(collector.lines_received, 1)
self.assertEqual(collector.lines_invalid, 0)


class TSDBlacklistingTests(unittest.TestCase):
"""
Tests of TSD blacklisting logic
Expand Down Expand Up @@ -287,10 +286,11 @@ def test_doublePickTwoConnections(self):
sender.pick_connection()
self.assertEqual(tsd1, (sender.host, sender.port))


class UDPCollectorTests(unittest.TestCase):

def setUp(self):
if ('udp_bridge.py' not in tcollector.COLLECTORS): # pylint: disable=maybe-no-member
if 'udp_bridge.py' not in tcollector.COLLECTORS: # pylint: disable=maybe-no-member
raise unittest.SkipTest("udp_bridge unavailable")

self.saved_exit = sys.exit
Expand Down Expand Up @@ -332,7 +332,8 @@ def run_bridge_test(self, udpInputLines, stdoutLines, stderrLines):

def test_populated(self):
# assertIsInstance, assertIn, assertIsNone do not exist in Python 2.6
self.assertTrue(isinstance(self.udp_bridge, tcollector.Collector), msg="self.udp_bridge not instance of tcollector.Collector") # pylint: disable=maybe-no-member
self.assertTrue(isinstance(self.udp_bridge, tcollector.Collector),
msg="self.udp_bridge not instance of tcollector.Collector") # pylint: disable=maybe-no-member
self.assertEqual(self.udp_bridge.proc, None)
self.assertTrue('main' in self.udp_globals, msg="'main' not in self.udp_globals")

Expand Down Expand Up @@ -518,6 +519,6 @@ def test_procstats(self):
logging.basicConfig()
cdir = os.path.join(os.path.dirname(os.path.realpath(sys.argv[0])),
'collectors')
tcollector.setup_python_path(cdir) # pylint: disable=maybe-no-member
tcollector.populate_collectors(cdir) # pylint: disable=maybe-no-member
tcollector.setup_python_path(cdir) # pylint: disable=maybe-no-member
tcollector.populate_collectors(cdir) # pylint: disable=maybe-no-member
sys.exit(unittest.main())

0 comments on commit 9f47278

Please sign in to comment.