class WatchNodeTestCase(TestCase):
def setUp(self):
self.mock_api = MagicMock()
self.now = utcnow()
self.later = self.now + timedelta(seconds=1)
self.test_data = [b'line_1\n', b'line_2\n']
self.inst = WatchNode('test_type', self.mock_api, timedelta(seconds=1))
self.inst.last_send = self.now
# Creates a temporary file in a known location
self.tmpdir = mkdtemp()
def fixed_temp_file(*args, **kwargs):
return NamedTemporaryFile(delete=False, dir=self.tmpdir)
self.fixed_temp_file = fixed_temp_file
def tearDown(self):
rmtree(self.tmpdir)
def test_flush_data_compressed(self):
patch_src = 'ona_service.log_watcher.NamedTemporaryFile'
with patch(patch_src, self.fixed_temp_file):
self.inst.flush_data(self.test_data, self.later, compress=True)
# Gzip-read of the file should give back the input data
for file_path in iglob(join(self.tmpdir, '*')):
with gzip.open(file_path, 'rb') as infile:
self.assertEqual(infile.read(), ''.join(self.test_data))
def test_flush_data_uncompressed(self):
patch_src = 'ona_service.log_watcher.NamedTemporaryFile'
with patch(patch_src, self.fixed_temp_file):
self.inst.flush_data(self.test_data, self.later)
# Direct read of the file should give back the input data
for file_path in iglob(join(self.tmpdir, '*')):
with open(file_path, 'r') as infile:
self.assertEqual(infile.read(), ''.join(self.test_data))
def test_flush_data_calls(self):
# No data -> no calls
self.inst.flush_data([], self.later)
self.assertEqual(self.mock_api.send_signal.call_count, 0)
# Not enough time has passed -> no calls
self.inst.flush_data(self.test_data, self.now)
self.assertEqual(self.mock_api.send_signal.call_count, 0)
# Data is present, enough time has passed -> one call
self.inst.flush_data(self.test_data, self.later)
self.assertEqual(self.mock_api.send_signal.call_count, 1)
class WatchNodeTestCase(TestCase):
def setUp(self):
self.mock_api = MagicMock()
self.now = utcnow()
self.later = self.now + timedelta(seconds=1)
self.test_data = [b'line_1\n', b'line_2\n']
self.inst = WatchNode('test_type', self.mock_api, timedelta(seconds=1))
self.inst.last_send = self.now
# Creates a temporary file in a known location
self.tmpdir = mkdtemp()
def fixed_temp_file(*args, **kwargs):
return NamedTemporaryFile(delete=False, dir=self.tmpdir)
self.fixed_temp_file = fixed_temp_file
def tearDown(self):
rmtree(self.tmpdir)
def test_flush_data_compressed(self):
patch_src = 'ona_service.log_watcher.NamedTemporaryFile'
with patch(patch_src, self.fixed_temp_file):
self.inst.flush_data(self.test_data, self.later, compress=True)
# Gzip-read of the file should give back the input data
for file_path in iglob(join(self.tmpdir, '*')):
with gzip.open(file_path, 'rb') as infile:
self.assertEqual(infile.read(), ''.join(self.test_data))
def test_flush_data_uncompressed(self):
patch_src = 'ona_service.log_watcher.NamedTemporaryFile'
with patch(patch_src, self.fixed_temp_file):
self.inst.flush_data(self.test_data, self.later)
# Direct read of the file should give back the input data
for file_path in iglob(join(self.tmpdir, '*')):
with open(file_path, 'r') as infile:
self.assertEqual(infile.read(), ''.join(self.test_data))
def test_flush_data_calls(self):
# No data -> no calls
self.inst.flush_data([], self.later)
self.assertEqual(self.mock_api.send_signal.call_count, 0)
# Not enough time has passed -> no calls
self.inst.flush_data(self.test_data, self.now)
self.assertEqual(self.mock_api.send_signal.call_count, 0)
# Data is present, enough time has passed -> one call
self.inst.flush_data(self.test_data, self.later)
self.assertEqual(self.mock_api.send_signal.call_count, 1)
def setUp(self):
self.mock_api = MagicMock()
self.now = utcnow()
self.later = self.now + timedelta(seconds=1)
self.test_data = [b'line_1\n', b'line_2\n']
self.inst = WatchNode('test_type', self.mock_api, timedelta(seconds=1))
self.inst.last_send = self.now
# Creates a temporary file in a known location
self.tmpdir = mkdtemp()
def fixed_temp_file(*args, **kwargs):
return NamedTemporaryFile(delete=False, dir=self.tmpdir)
self.fixed_temp_file = fixed_temp_file
def setUp(self):
self.mock_api = MagicMock()
self.now = utcnow()
self.later = self.now + timedelta(seconds=1)
self.test_data = [b'line_1\n', b'line_2\n']
self.inst = WatchNode('test_type', self.mock_api, timedelta(seconds=1))
self.inst.last_send = self.now
# Creates a temporary file in a known location
self.tmpdir = mkdtemp()
def fixed_temp_file(*args, **kwargs):
return NamedTemporaryFile(delete=False, dir=self.tmpdir)
self.fixed_temp_file = fixed_temp_file