245 lines
9.3 KiB
Python
245 lines
9.3 KiB
Python
# Copyright 2015 Mirantis, Inc.
|
|
#
|
|
# 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 datetime
|
|
from oslo_serialization import jsonutils
|
|
|
|
from nailgun import consts
|
|
from nailgun import objects
|
|
from nailgun.test.base import BaseMasterNodeSettignsTest
|
|
from nailgun.test.base import fake_tasks
|
|
from nailgun.test.base import mock_rpc
|
|
from nailgun.utils import reverse
|
|
|
|
|
|
class TestActionLogs(BaseMasterNodeSettignsTest):
|
|
|
|
@fake_tasks()
|
|
def test_action_log_updating_for_tasks(self):
|
|
meta1 = self.env.generate_interfaces_in_meta(2)
|
|
mac1 = meta1['interfaces'][0]['mac']
|
|
meta2 = self.env.generate_interfaces_in_meta(2)
|
|
mac2 = meta2['interfaces'][0]['mac']
|
|
|
|
self.env.create(
|
|
cluster_kwargs={},
|
|
nodes_kwargs=[
|
|
{"api": True, "meta": meta1, "mac": mac1},
|
|
{"api": True, "meta": meta2, "mac": mac2},
|
|
]
|
|
)
|
|
|
|
task = self.env.launch_verify_networks()
|
|
self.assertEqual(task.status, consts.TASK_STATUSES.ready)
|
|
|
|
action_logs = objects.ActionLogCollection.filter_by(
|
|
None, action_type=consts.ACTION_TYPES.nailgun_task)
|
|
for al in action_logs:
|
|
task_status = al.additional_info["ended_with_status"]
|
|
self.assertEqual(task_status, consts.TASK_STATUSES.ready)
|
|
|
|
@fake_tasks()
|
|
def test_only_utc_datetime_used(self):
|
|
start_dt = datetime.datetime.utcnow()
|
|
cluster = self.env.create(
|
|
api=True,
|
|
nodes_kwargs=[
|
|
{'roles': ['controller'], 'pending_addition': True},
|
|
{'roles': ['compute'], 'pending_addition': True},
|
|
]
|
|
)
|
|
|
|
task = self.env.launch_deployment()
|
|
self.assertEqual(task.status, consts.TASK_STATUSES.ready)
|
|
|
|
self.app.delete(
|
|
reverse('ClusterHandler', kwargs={'obj_id': cluster.id}),
|
|
headers=self.default_headers
|
|
)
|
|
|
|
end_dt = datetime.datetime.utcnow()
|
|
action_logs = objects.ActionLogCollection.filter_by(None)
|
|
|
|
for action_log in action_logs:
|
|
self.assertLessEqual(start_dt, action_log.start_timestamp)
|
|
self.assertLessEqual(start_dt, action_log.end_timestamp)
|
|
self.assertGreaterEqual(end_dt, action_log.start_timestamp)
|
|
self.assertGreaterEqual(end_dt, action_log.end_timestamp)
|
|
|
|
@mock_rpc()
|
|
def test_all_action_logs_types_saved(self):
|
|
# Creating nailgun_tasks
|
|
cluster = self.env.create(
|
|
api=True,
|
|
nodes_kwargs=[
|
|
{'roles': ['controller'], 'pending_addition': True},
|
|
{'roles': ['compute'], 'pending_addition': True},
|
|
]
|
|
)
|
|
|
|
task = self.env.launch_deployment()
|
|
self.assertNotEqual(task.status, consts.TASK_STATUSES.error)
|
|
objects.Task.delete(task)
|
|
|
|
# Creating http_request
|
|
self.app.delete(
|
|
reverse('ClusterHandler', kwargs={'obj_id': cluster.id}),
|
|
headers=self.default_headers
|
|
)
|
|
|
|
# Filtering action types
|
|
action_logs = objects.ActionLogCollection.filter_by(None)
|
|
action_types = [al.action_type for al in action_logs]
|
|
self.assertSetEqual(set(consts.ACTION_TYPES), set(action_types))
|
|
|
|
def check_create_stats_user_logged(self, modification_api_call):
|
|
self.env.create(
|
|
nodes_kwargs=[
|
|
{'roles': ['controller'], 'pending_addition': True},
|
|
]
|
|
)
|
|
|
|
task = self.env.launch_deployment()
|
|
self.assertNotEqual(task.status, consts.TASK_STATUSES.error)
|
|
|
|
# Checking sending stats is not enabled yet by saving user choice
|
|
resp = self.app.get(
|
|
reverse('MasterNodeSettingsHandler'),
|
|
headers=self.default_headers)
|
|
data = resp.json_body
|
|
self.assertFalse(data['settings']['statistics']
|
|
['user_choice_saved']['value'])
|
|
self.assertTrue(data['settings']['statistics']
|
|
['send_anonymous_statistic']['value'])
|
|
|
|
# Enabling stats
|
|
stat_settings = data['settings']['statistics']
|
|
stat_settings['user_choice_saved']['value'] = True
|
|
stat_settings['send_anonymous_statistic']['value'] = True
|
|
resp = modification_api_call(
|
|
reverse('MasterNodeSettingsHandler'),
|
|
headers=self.default_headers,
|
|
params=jsonutils.dumps(data)
|
|
)
|
|
self.assertEqual(200, resp.status_code)
|
|
|
|
tasks = objects.TaskCollection.filter_by(
|
|
None, name=consts.TASK_NAMES.create_stats_user).all()
|
|
tasks_count = len(tasks)
|
|
task = tasks[-1]
|
|
action_log = objects.ActionLogCollection.filter_by(
|
|
None, task_uuid=task.uuid)
|
|
self.assertIsNotNone(action_log)
|
|
|
|
# Checking settings modified
|
|
resp = self.app.get(
|
|
reverse('MasterNodeSettingsHandler'),
|
|
headers=self.default_headers)
|
|
new_data = resp.json_body
|
|
self.assertTrue(new_data['settings']['statistics']
|
|
['user_choice_saved']['value'])
|
|
self.assertTrue(new_data['settings']['statistics']
|
|
['send_anonymous_statistic']['value'])
|
|
|
|
# Checking the second call doesn't produce task
|
|
resp = modification_api_call(
|
|
reverse('MasterNodeSettingsHandler'),
|
|
headers=self.default_headers,
|
|
params=jsonutils.dumps(data)
|
|
)
|
|
self.assertEqual(200, resp.status_code)
|
|
duplicate_call_tasks_count = objects.TaskCollection.filter_by(
|
|
None, name=consts.TASK_NAMES.create_stats_user).count()
|
|
self.assertEquals(tasks_count, duplicate_call_tasks_count)
|
|
|
|
@fake_tasks(override_state={'progress': 100,
|
|
'status': consts.TASK_STATUSES.ready})
|
|
def test_create_stats_user_logged_put(self):
|
|
self.check_create_stats_user_logged(self.app.put)
|
|
|
|
@fake_tasks(override_state={'progress': 100,
|
|
'status': consts.TASK_STATUSES.ready})
|
|
def test_create_stats_user_logged_patch(self):
|
|
self.check_create_stats_user_logged(self.app.patch)
|
|
|
|
def check_remove_stats_user_logged(self, modification_api_call):
|
|
self.env.create(
|
|
nodes_kwargs=[
|
|
{'roles': ['controller'], 'pending_addition': True},
|
|
]
|
|
)
|
|
|
|
task = self.env.launch_deployment()
|
|
self.assertEqual(task.status, consts.TASK_STATUSES.ready)
|
|
|
|
self.enable_sending_stats()
|
|
|
|
resp = self.app.get(
|
|
reverse('MasterNodeSettingsHandler'),
|
|
headers=self.default_headers)
|
|
data = resp.json_body
|
|
self.assertTrue(data['settings']['statistics']
|
|
['user_choice_saved']['value'])
|
|
self.assertTrue(data['settings']['statistics']
|
|
['send_anonymous_statistic']['value'])
|
|
|
|
# Disabling stats
|
|
stat_settings = data['settings']['statistics']
|
|
stat_settings['send_anonymous_statistic']['value'] = False
|
|
resp = modification_api_call(
|
|
reverse('MasterNodeSettingsHandler'),
|
|
headers=self.default_headers,
|
|
params=jsonutils.dumps(data)
|
|
)
|
|
self.assertEqual(200, resp.status_code)
|
|
|
|
tasks = objects.TaskCollection.filter_by(
|
|
None, name=consts.TASK_NAMES.remove_stats_user).all()
|
|
tasks_count = len(tasks)
|
|
task = tasks[-1]
|
|
action_log = objects.ActionLogCollection.filter_by(
|
|
None, task_uuid=task.uuid)
|
|
self.assertIsNotNone(action_log)
|
|
|
|
# Checking settings modified
|
|
resp = self.app.get(
|
|
reverse('MasterNodeSettingsHandler'),
|
|
headers=self.default_headers)
|
|
new_data = resp.json_body
|
|
self.assertTrue(new_data['settings']['statistics']
|
|
['user_choice_saved']['value'])
|
|
self.assertFalse(new_data['settings']['statistics']
|
|
['send_anonymous_statistic']['value'])
|
|
|
|
# Checking the second call doesn't produce task
|
|
resp = modification_api_call(
|
|
reverse('MasterNodeSettingsHandler'),
|
|
headers=self.default_headers,
|
|
params=jsonutils.dumps(data)
|
|
)
|
|
self.assertEqual(200, resp.status_code)
|
|
duplicate_call_tasks_count = objects.TaskCollection.filter_by(
|
|
None, name=consts.TASK_NAMES.remove_stats_user).count()
|
|
self.assertEquals(tasks_count, duplicate_call_tasks_count)
|
|
|
|
@fake_tasks(override_state={'progress': 100,
|
|
'status': consts.TASK_STATUSES.ready})
|
|
def test_remove_stats_user_logged_put(self):
|
|
self.check_remove_stats_user_logged(self.app.put)
|
|
|
|
@fake_tasks(override_state={'progress': 100,
|
|
'status': consts.TASK_STATUSES.ready})
|
|
def test_remove_stats_user_logged_patch(self):
|
|
self.check_remove_stats_user_logged(self.app.patch)
|