From e01ad034bf16da8bf48edea9dcd5630e3eea89ac Mon Sep 17 00:00:00 2001
From: Mario Splivalo <mario.splivalo@canonical.com>
Date: Wed, 10 Dec 2014 12:18:45 +0100
Subject: Syncing to get closer with
 lp:~mariosplivalo/charms/trusty/mongodb/charmhelpers-sync

---
 unit_tests/test_write_log_rotate_config.py | 35 ++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100644 unit_tests/test_write_log_rotate_config.py

(limited to 'unit_tests')

diff --git a/unit_tests/test_write_log_rotate_config.py b/unit_tests/test_write_log_rotate_config.py
new file mode 100644
index 0000000..e1d6154
--- /dev/null
+++ b/unit_tests/test_write_log_rotate_config.py
@@ -0,0 +1,35 @@
+import mock
+import os
+import unittest
+import tempfile
+import sys
+sys.path.append('hooks')
+import hooks
+
+
+class TestWriteLogrotateConfigFile(unittest.TestCase):
+
+    def test_success(self):
+        logpath = '/tmp/foo/foo.log'
+        config_data = {
+            'logpath': logpath,
+            'logrotate-frequency': 'daily',
+            'logrotate-maxsize': '5G',
+            'logrotate-rotate': 5,
+        }
+        fd, temp_fn = tempfile.mkstemp()
+        os.close(fd)
+        with mock.patch('hooks.juju_log') as mock_juju_log:
+            with mock.patch('hooks.open', create=True) as mock_open:
+                mock_open.return_value = mock.MagicMock(spec=file)
+                hooks.write_logrotate_config(config_data, temp_fn)
+                os.unlink(temp_fn)
+        mock_juju_log.assert_called_once_with('Writing {}.'.format(temp_fn))
+        mock_open.assert_called_once_with(temp_fn, 'w')
+        mock_file = mock_open().__enter__()
+        call_args = mock_file.write.call_args[0][0]
+        self.assertTrue(mock_file.write.called)
+        self.assertIn(logpath, call_args)
+        self.assertIn('daily', call_args)
+        self.assertIn('maxsize 5G', call_args)
+        self.assertIn('rotate 5', call_args)
-- 
cgit v1.2.3