Add tests for installutils.set_directive

Part of: https://pagure.io/freeipa/issue/7288

Reviewed-By: Florence Blanc-Renaud <frenaud@redhat.com>

ipatests/test_ipaserver/test_install/test_installutils.py

@@ -0,0 +1,57 @@
+#
+# Copyright (C) 2017  FreeIPA Contributors.  See COPYING for license
+#
+
+import os
+import tempfile
+
+from ipaserver.install import installutils
+
+EXAMPLE_CONFIG = [
+    'foo=1\n',
+    'foobar=2\n',
+]
+
+
+class test_set_directive_lines(object):
+    def test_remove_directive(self):
+        lines = installutils.set_directive_lines(
+            False, '=', 'foo', None, EXAMPLE_CONFIG)
+        assert list(lines) == ['foobar=2\n']
+
+    def test_add_directive(self):
+        lines = installutils.set_directive_lines(
+            False, '=', 'baz', '4', EXAMPLE_CONFIG)
+        assert list(lines) == ['foo=1\n', 'foobar=2\n', 'baz=4\n']
+
+    def test_set_directive_does_not_clobber_suffix_key(self):
+        lines = installutils.set_directive_lines(
+            False, '=', 'foo', '3', EXAMPLE_CONFIG)
+        assert list(lines) == ['foo=3\n', 'foobar=2\n']
+
+
+class test_set_directive(object):
+    def test_set_directive(self):
+        """Check that set_directive writes the new data and preserves mode."""
+        fd, filename = tempfile.mkstemp()
+        try:
+            os.close(fd)
+            stat_pre = os.stat(filename)
+
+            with open(filename, 'w') as f:
+                for line in EXAMPLE_CONFIG:
+                    f.write(line)
+
+            installutils.set_directive(filename, 'foo', '3', False, '=')
+
+            stat_post = os.stat(filename)
+            with open(filename, 'r') as f:
+                lines = list(f)
+
+            assert lines == ['foo=3\n', 'foobar=2\n']
+            assert stat_pre.st_mode == stat_post.st_mode
+            assert stat_pre.st_uid == stat_post.st_uid
+            assert stat_pre.st_gid == stat_post.st_gid
+
+        finally:
+            os.remove(filename)