1 files changed, 114 insertions, 0 deletions
diff --git a/src/account/test/TestService.py b/src/account/test/TestService.py
new file mode 100644
index 0000000..08a47ee
--- /dev/null
+++ b/src/account/test/TestService.py
@@ -0,0 +1,114 @@
+# -*- encoding: utf-8 -*-
+# Copyright (C) 2012 Red Hat, Inc.  All rights reserved.
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+#
+# Authors: Roman Rakus <rrakus@redhat.com
+#
+
+from common import AccountBase
+from methods import *
+import pywbem
+
+class TestService(AccountBase):
+    """
+    Class for testing LMI_AccountManagementService
+    """
+
+    def test_create_account(self):
+        """
+        Test create account parameters
+        """
+        # make sure the account will not exist
+        clean_account(self.user_name)
+        computer_system = self.wbemconnection.ExecQuery('WQL',
+            'select * from Linux_ComputerSystem')[0]
+        lams = self.wbemconnection.ExecQuery('WQL',
+            'select * from LMI_AccountManagementService')[0]
+
+        # create account and test all parameters
+        shell = "testshell"
+        system_account = True
+        dont_create_home = True
+        dont_create_group = True
+        gid = 0
+        gecos = "test create account gecos"
+        home_dir = "/test/home"
+        password = '$6$9Ky8vI6f$ipRcdc7rgMrtDh.sWOaRSoBck2cLz4eUom8Eze.NaY2DoMmNimuFBrXpJjlPCjMoeFTYC.FdZwj488JZcohyw1'
+        uid = 777
+        (rc, out) = self.wbemconnection.InvokeMethod("CreateAccount", lams.path,
+            Name = self.user_name,
+            System = computer_system.path,
+            Shell = shell,
+            SystemAccount = system_account,
+            DontCreateHome = dont_create_home,
+            DontCreateGroup = dont_create_group,
+            GID = pywbem.Uint32(gid),
+            GECOS = gecos,
+            HomeDirectory = home_dir,
+            Password = password,
+            UID = pywbem.Uint32(uid))
+
+        acc = out["Account"]
+        idents = out["Identities"]
+
+        # check return values
+        # account
+        self.assertEqual(rc, 0)
+        self.assertEqual(acc["Name"], self.user_name)
+        # identities
+        for identity in idents:
+            if identity["InstanceID"].find("UID") != -1:
+                # user identity
+                self.assertEqual(identity["InstanceID"], "LMI:UID:%d" %uid)
+            else:
+                # group identity
+                self.assertEqual(identity["InstanceID"], "LMI:GID:%d" %gid)
+
+        # check with system info
+        self.assertEqual(field_in_passwd(self.user_name, 2), str(uid))
+        self.assertEqual(field_in_passwd(self.user_name, 3), str(gid))
+        self.assertEqual(field_in_passwd(self.user_name, 4), gecos)
+        self.assertEqual(field_in_passwd(self.user_name, 5), home_dir)
+        self.assertEqual(field_in_passwd(self.user_name, 6), shell)
+        self.assertEqual(field_in_shadow(self.user_name, 1), password)
+        clean_account(self.user_name)
+
+    def test_create_group(self):
+        """
+        Test create group parameters
+        """
+        clean_group(self.group_name)
+        computer_system = self.wbemconnection.ExecQuery('WQL',
+            'select * from Linux_ComputerSystem')[0]
+        lams = self.wbemconnection.ExecQuery('WQL',
+            'select * from LMI_AccountManagementService')[0]
+        system_account = True
+        gid = 666
+        (rc, out) = self.wbemconnection.InvokeMethod("CreateGroup", lams.path,
+            Name=self.group_name,
+            System=computer_system.path,
+            SystemAccount=system_account,
+            GID=pywbem.Uint32(gid))
+
+        group = out["Group"]
+        idents = out["Identities"]
+        self.assertEqual(rc, 0)
+        self.assertEqual(group["Name"], self.group_name)
+        for identity in idents:
+            self.assertEqual(identity["InstanceID"], "LMI:GID:%d" %gid)
+
+        self.assertEqual(field_in_group(self.group_name, 2), str(gid))
+        clean_group(self.group_name)