diff options
| -rwxr-xr-x | google_compute_engine/accounts/accounts_daemon.py | 5 | ||||
| -rw-r--r-- | google_compute_engine/accounts/tests/accounts_daemon_test.py | 4 |
2 files changed, 5 insertions, 4 deletions
diff --git a/google_compute_engine/accounts/accounts_daemon.py b/google_compute_engine/accounts/accounts_daemon.py index a7c6939..0a84b16 100755 --- a/google_compute_engine/accounts/accounts_daemon.py +++ b/google_compute_engine/accounts/accounts_daemon.py @@ -275,10 +275,11 @@ class AccountsDaemon(object): enable_two_factor = self._GetEnableTwoFactorValue(result) if enable_oslogin: desired_users = {} - self.oslogin.UpdateOsLogin(enable=True, two_factor=enable_two_factor) + self.oslogin.UpdateOsLogin( + enable_oslogin=True, two_factor=enable_two_factor) else: desired_users = self._GetAccountsData(result) - self.oslogin.UpdateOsLogin(enable=False) + self.oslogin.UpdateOsLogin(enable_oslogin=False) remove_users = sorted(set(configured_users) - set(desired_users.keys())) self._UpdateUsers(desired_users) self._RemoveUsers(remove_users) diff --git a/google_compute_engine/accounts/tests/accounts_daemon_test.py b/google_compute_engine/accounts/tests/accounts_daemon_test.py index 2ae1250..b1bd601 100644 --- a/google_compute_engine/accounts/tests/accounts_daemon_test.py +++ b/google_compute_engine/accounts/tests/accounts_daemon_test.py @@ -514,7 +514,7 @@ class AccountsDaemonTest(unittest.TestCase): mock.call.setup._GetEnableOsLoginValue(result), mock.call.setup._GetEnableTwoFactorValue(result), mock.call.setup._GetAccountsData(result), - mock.call.oslogin.UpdateOsLogin(enable=False), + mock.call.oslogin.UpdateOsLogin(enable_oslogin=False), mock.call.setup._UpdateUsers(desired), mock.call.setup._RemoveUsers(mock.ANY), mock.call.utils.SetConfiguredUsers(mock.ANY), @@ -547,7 +547,7 @@ class AccountsDaemonTest(unittest.TestCase): mock.call.utils.GetConfiguredUsers(), mock.call.setup._GetEnableOsLoginValue(result), mock.call.setup._GetEnableTwoFactorValue(result), - mock.call.oslogin.UpdateOsLogin(enable=True, two_factor=False), + mock.call.oslogin.UpdateOsLogin(enable_oslogin=True, two_factor=False), mock.call.setup._UpdateUsers(desired), mock.call.setup._RemoveUsers(mock.ANY), mock.call.utils.SetConfiguredUsers(mock.ANY), |