From af01f55da19f54b1f96b3bda12fea2359a942267 Mon Sep 17 00:00:00 2001
From: sunilkumarn417 <sunnagar@redhat.com>
Date: Fri, 19 Jun 2020 13:44:54 +0530
Subject: [PATCH] orchestra/daemon/systemd: fix proc_regex in pid() method

Signed-off-by: sunilkumarn417 <sunnagar@redhat.com>
---
 teuthology/orchestra/daemon/systemd.py        |  2 +-
 .../daemon-systemdstate-pid-ps-ef.output      |  5 ++
 teuthology/orchestra/test/test_systemd.py     | 54 +++++++++++++++++++
 3 files changed, 60 insertions(+), 1 deletion(-)
 create mode 100644 teuthology/orchestra/test/files/daemon-systemdstate-pid-ps-ef.output
 create mode 100644 teuthology/orchestra/test/test_systemd.py

diff --git a/teuthology/orchestra/daemon/systemd.py b/teuthology/orchestra/daemon/systemd.py
index c202669fbe..d7069a6489 100644
--- a/teuthology/orchestra/daemon/systemd.py
+++ b/teuthology/orchestra/daemon/systemd.py
@@ -106,7 +106,7 @@ class SystemDState(DaemonState):
         # process regex to match RADOSGW process command string
         # eg. "/usr/bin/radosgw -f --cluster ceph --name <daemon-id=self.id_>"
         if self.type_ == "rgw":
-            proc_regex = "{}.* --name .*{}".format(self.daemon_type, self.id_)
+            proc_regex = '"{}.*--name.*{}"'.format(self.daemon_type, self.id_)
 
         args = ['ps', '-ef',
                 run.Raw('|'),
diff --git a/teuthology/orchestra/test/files/daemon-systemdstate-pid-ps-ef.output b/teuthology/orchestra/test/files/daemon-systemdstate-pid-ps-ef.output
new file mode 100644
index 0000000000..ddddf571c9
--- /dev/null
+++ b/teuthology/orchestra/test/files/daemon-systemdstate-pid-ps-ef.output
@@ -0,0 +1,5 @@
+ceph       658     1  0 Jun08 ?        00:07:43 /usr/bin/ceph-mgr -f --cluster ceph --id host1 --setuser ceph --setgroup ceph
+ceph      1634     1  0 Jun08 ?        00:02:17 /usr/bin/ceph-mds -f --cluster ceph --id host1 --setuser ceph --setgroup ceph
+ceph     31555     1  0 Jun08 ?        01:13:50 /usr/bin/ceph-mon -f --cluster ceph --id host1 --setuser ceph --setgroup ceph
+ceph     31765     1  0 Jun08 ?        00:48:42 /usr/bin/radosgw -f --cluster ceph --name client.rgw.host1.rgw0 --setuser ceph --setgroup ceph
+ceph     97427     1  0 Jun17 ?        00:41:39 /usr/bin/ceph-osd -f --cluster ceph --id 0 --setuser ceph --setgroup ceph
\ No newline at end of file
diff --git a/teuthology/orchestra/test/test_systemd.py b/teuthology/orchestra/test/test_systemd.py
new file mode 100644
index 0000000000..c7cb3425f7
--- /dev/null
+++ b/teuthology/orchestra/test/test_systemd.py
@@ -0,0 +1,54 @@
+import argparse
+import os
+
+from logging import debug
+from teuthology import misc
+from teuthology.orchestra import cluster
+from teuthology.orchestra.run import quote
+from teuthology.orchestra.daemon.group import DaemonGroup
+import subprocess
+
+
+class FakeRemote(object):
+    pass
+
+
+def test_pid():
+    ctx = argparse.Namespace()
+    ctx.daemons = DaemonGroup(use_systemd=True)
+    remote = FakeRemote()
+
+    ps_ef_output_path = os.path.join(
+        os.path.dirname(__file__),
+        "files/daemon-systemdstate-pid-ps-ef.output"
+    )
+
+    # patching ps -ef command output using a file
+    def sh(args):
+        args[0:2] = ["cat", ps_ef_output_path]
+        debug(args)
+        return subprocess.getoutput(quote(args))
+
+    remote.sh = sh
+    remote.init_system = 'systemd'
+    remote.shortname = 'host1'
+
+    ctx.cluster = cluster.Cluster(
+        remotes=[
+            (remote, ['rgw.0', 'mon.a', 'mgr.a', 'mds.a', 'osd.0'])
+        ],
+    )
+
+    for remote, roles in ctx.cluster.remotes.items():
+        for role in roles:
+            _, rol, id_ = misc.split_role(role)
+            if any(rol.startswith(x) for x in ['mon', 'mgr', 'mds']):
+                ctx.daemons.register_daemon(remote, rol, remote.shortname)
+            else:
+                ctx.daemons.register_daemon(remote, rol, id_)
+
+    for _, daemons in ctx.daemons.daemons.items():
+        for daemon in daemons.values():
+            pid = daemon.pid
+            debug(pid)
+            assert pid
-- 
2.39.5