diff --git a/ninjalog_uploader.py b/ninjalog_uploader.py index 80f72698b..6b89826fe 100755 --- a/ninjalog_uploader.py +++ b/ninjalog_uploader.py @@ -51,10 +51,10 @@ ALLOWLISTED_CONFIGS = ( def IsGoogler(): """Check whether this user is Googler or not.""" p = subprocess.run( - "goma_auth info", + "cipd auth-info", stdout=subprocess.PIPE, stderr=subprocess.PIPE, - universal_newlines=True, + text=True, shell=True, ) if p.returncode != 0: @@ -63,8 +63,9 @@ def IsGoogler(): if len(lines) == 0: return False l = lines[0] - # |l| will be like 'Login as @google.com' for googler using goma. - return l.startswith("Login as ") and l.endswith("@google.com") + # |l| will be like 'Logged in as @google.com.' for googler using + # reclient. + return l.startswith("Logged in as ") and l.endswith("@google.com.") def ParseGNArgs(gn_args): diff --git a/tests/ninjalog_uploader_test.py b/tests/ninjalog_uploader_test.py index 94aaea58e..020fbacc6 100755 --- a/tests/ninjalog_uploader_test.py +++ b/tests/ninjalog_uploader_test.py @@ -19,8 +19,7 @@ class NinjalogUploaderTest(unittest.TestCase): def test_IsGoogler(self): with unittest.mock.patch('subprocess.run') as run_mock: run_mock.return_value.returncode = 0 - run_mock.return_value.stdout = ('Login as foo@google.com\n' - 'goma is ready to use') + run_mock.return_value.stdout = 'Logged in as foo@google.com.\n' self.assertTrue(ninjalog_uploader.IsGoogler()) with unittest.mock.patch('subprocess.run') as run_mock: @@ -34,7 +33,7 @@ class NinjalogUploaderTest(unittest.TestCase): with unittest.mock.patch('subprocess.run') as run_mock: run_mock.return_value.returncode = 0 - run_mock.return_value.stdout = 'Login as foo@example.com\n' + run_mock.return_value.stdout = 'Logged in as foo@example.com.\n' self.assertFalse(ninjalog_uploader.IsGoogler()) def test_parse_gn_args(self):