19aa6f7dd5
If the user set to be configured by cloudbase-init is already added, but no logon session has been created for that user, the profile registry key does not exist. As a consequence, the SetUserSSHPublicKeysPlugin fails, as it cannot retrieve the home directory from the registry key. Change-Id: I4226a1c08e940717709e65ba932ce15c8ce37aed Closes-Bug: #1415198
93 lines
3.2 KiB
Python
93 lines
3.2 KiB
Python
# Copyright 2015 Cloudbase Solutions Srl
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import unittest
|
|
|
|
try:
|
|
import unittest.mock as mock
|
|
except ImportError:
|
|
import mock
|
|
|
|
from cloudbaseinit.plugins.windows import createuser
|
|
from cloudbaseinit.tests import testutils
|
|
|
|
|
|
class CreateUserPluginTests(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self._create_user = createuser.CreateUserPlugin()
|
|
|
|
def test_create_user(self):
|
|
mock_osutils = mock.Mock()
|
|
self._create_user.create_user(
|
|
mock.sentinel.username,
|
|
mock.sentinel.password,
|
|
mock_osutils)
|
|
|
|
mock_osutils.create_user.assert_called_once_with(
|
|
mock.sentinel.username,
|
|
mock.sentinel.password)
|
|
|
|
@mock.patch('cloudbaseinit.plugins.windows.createuser.CreateUserPlugin.'
|
|
'_create_user_logon')
|
|
def test_post_create_user(self, mock_create_user_logon):
|
|
mock_osutils = mock.Mock()
|
|
self._create_user.post_create_user(
|
|
mock.sentinel.username,
|
|
mock.sentinel.password,
|
|
mock_osutils)
|
|
|
|
mock_create_user_logon.assert_called_once_with(
|
|
mock.sentinel.username,
|
|
mock.sentinel.password,
|
|
mock_osutils)
|
|
|
|
def test__create_user_logon(self):
|
|
mock_osutils = mock.Mock()
|
|
mock_token = mock.sentinel.token
|
|
mock_osutils.create_user_logon_session.return_value = mock_token
|
|
|
|
self._create_user._create_user_logon(
|
|
mock.sentinel.user_name,
|
|
mock.sentinel.password,
|
|
mock_osutils)
|
|
|
|
mock_osutils.create_user_logon_session.assert_called_once_with(
|
|
mock.sentinel.user_name,
|
|
mock.sentinel.password,
|
|
True)
|
|
mock_osutils.close_user_logon_session.assert_called_once_with(
|
|
mock_token)
|
|
|
|
def test__create_user_logon_fails(self):
|
|
mock_osutils = mock.Mock()
|
|
mock_osutils.create_user_logon_session.side_effect = Exception
|
|
|
|
with testutils.LogSnatcher('cloudbaseinit.plugins.windows.'
|
|
'createuser') as snatcher:
|
|
self._create_user._create_user_logon(
|
|
mock.sentinel.user_name,
|
|
mock.sentinel.password,
|
|
mock_osutils)
|
|
|
|
mock_osutils.create_user_logon_session.assert_called_once_with(
|
|
mock.sentinel.user_name,
|
|
mock.sentinel.password,
|
|
True)
|
|
self.assertFalse(mock_osutils.close_user_logon_session.called)
|
|
logging_message = (
|
|
"Cannot create a user logon session for user: \"%s\""
|
|
% mock.sentinel.user_name)
|
|
self.assertTrue(snatcher.output[0].startswith(logging_message))
|