c01a7a74f0
In Python 3, the mock module is now part of the unittest module. We no longer support Python 2, so there's no reason for the import checks. Change-Id: I7b4d68cba4b587ad9f0121058ba615a9bb0aec93
89 lines
3.1 KiB
Python
89 lines
3.1 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
|
|
import unittest.mock as 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))
|