From 31a9e217b1f65fb35097482289caccdcfaf2106f Mon Sep 17 00:00:00 2001 From: Anand Shanmugam Date: Thu, 23 Jul 2015 03:55:16 -0700 Subject: [PATCH] Unit Test cases for cpulse objects Partially implements: blueprint unit-tests Change-Id: Ic68007d6e2aea77b8370cd8a674862f16339c350 --- cloudpulse/tests/unit/db/utils.py | 2 + cloudpulse/tests/unit/objects/__init__.py | 0 cloudpulse/tests/unit/objects/test_cpulse.py | 150 +++++++++++++++++++ 3 files changed, 152 insertions(+) create mode 100644 cloudpulse/tests/unit/objects/__init__.py create mode 100644 cloudpulse/tests/unit/objects/test_cpulse.py diff --git a/cloudpulse/tests/unit/db/utils.py b/cloudpulse/tests/unit/db/utils.py index 83366a0..6c81561 100644 --- a/cloudpulse/tests/unit/db/utils.py +++ b/cloudpulse/tests/unit/db/utils.py @@ -24,6 +24,8 @@ def get_cpulse_test(**kw): 'state': kw.get('state', 'created'), 'result': kw.get('state', 'success'), 'testtype': kw.get('testtype', 'periodic'), + 'created_at': kw.get('created_at'), + 'updated_at': kw.get('updated_at'), } diff --git a/cloudpulse/tests/unit/objects/__init__.py b/cloudpulse/tests/unit/objects/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/cloudpulse/tests/unit/objects/test_cpulse.py b/cloudpulse/tests/unit/objects/test_cpulse.py new file mode 100644 index 0000000..4e7646c --- /dev/null +++ b/cloudpulse/tests/unit/objects/test_cpulse.py @@ -0,0 +1,150 @@ +# 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. +from cloudpulse.common import exception +from cloudpulse.common import utils as cloudpulse_utils +from cloudpulse import objects +from cloudpulse.tests.unit.db import base +from cloudpulse.tests.unit.db import utils + +import mock + + +class TestCpulseObject(base.DbTestCase): + + def setUp(self): + super(TestCpulseObject, self).setUp() + self.cpulsetest = utils.get_cpulse_test() + + def test_get_with_id(self): + cpulse_id = self.cpulsetest['id'] + with mock.patch.object(self.dbapi, 'get_test_by_id', + autospec=True) as mock_get_test: + mock_get_test.return_value = self.cpulsetest + cpulse = objects.Cpulse.get(self.context, + cpulse_id) + mock_get_test.assert_called_once_with(self.context, + cpulse_id) + self.assertEqual(self.context, cpulse._context) + + def test_get_with_uuid(self): + cpulse_id = self.cpulsetest['uuid'] + with mock.patch.object(self.dbapi, 'get_test_by_uuid', + autospec=True) as mock_get_test: + mock_get_test.return_value = self.cpulsetest + cpulse = objects.Cpulse.get(self.context, + cpulse_id) + mock_get_test.assert_called_once_with(self.context, + cpulse_id) + self.assertEqual(self.context, cpulse._context) + + def test_get_exception(self): + cpulse_id = "cpulsetest" + self.assertRaises(exception.InvalidIdentity, objects.Cpulse.get, + self.context, cpulse_id) + + def test_get_by_uuid(self): + cpulse_id = self.cpulsetest['uuid'] + with mock.patch.object(self.dbapi, 'get_test_by_uuid', + autospec=True) as mock_get_test: + mock_get_test.return_value = self.cpulsetest + cpulse = objects.Cpulse.get_by_uuid(self.context, + cpulse_id) + mock_get_test.assert_called_once_with(self.context, + cpulse_id) + self.assertEqual(self.context, cpulse._context) + + def test_get_by_name(self): + cpulse_name = self.cpulsetest['name'] + with mock.patch.object(self.dbapi, 'get_test_by_name', + autospec=True) as mock_get_test: + mock_get_test.return_value = self.cpulsetest + cpulse = objects.Cpulse.get_by_name(self.context, + cpulse_name) + mock_get_test.assert_called_once_with(self.context, + cpulse_name) + self.assertEqual(self.context, cpulse._context) + + def test_get_by_id(self): + cpulse_id = self.cpulsetest['id'] + with mock.patch.object(self.dbapi, 'get_test_by_id', + autospec=True) as mock_get_test: + mock_get_test.return_value = self.cpulsetest + cpulse = objects.Cpulse.get_by_id(self.context, + cpulse_id) + mock_get_test.assert_called_once_with(self.context, + cpulse_id) + self.assertEqual(self.context, cpulse._context) + + def test_list(self): + with mock.patch.object(self.dbapi, 'get_test_list', + autospec=True) as mock_get_list: + mock_get_list.return_value = [self.cpulsetest] + cpulse = objects.Cpulse.list(self.context) + self.assertEqual(mock_get_list.call_count, 1) + self.assertEqual(len(cpulse), 1) + self.assertIsInstance(cpulse[0], objects.Cpulse) + self.assertEqual(self.context, cpulse[0]._context) + + def test_cpulse_create(self): + with mock.patch.object(self.dbapi, 'create_test', + autospec=True) as mock_create_test: + mock_create_test.return_value = self.cpulsetest + cpulse = objects.Cpulse(self.context, **self.cpulsetest) + cpulse.create() + mock_create_test.assert_called_once_with(self.cpulsetest) + self.assertEqual(self.context, cpulse._context) + + def test_cpulse_destroy(self): + uuid = self.cpulsetest['uuid'] + with mock.patch.object(self.dbapi, 'get_test_by_uuid', + autospec=True) as mock_get_cpulse: + mock_get_cpulse.return_value = self.cpulsetest + with mock.patch.object(self.dbapi, 'destroy_test', + autospec=True) as mock_destroy_cpulse: + cpulse = objects.Cpulse.get_by_uuid(self.context, uuid) + cpulse.destroy() + mock_get_cpulse.assert_called_once_with(self.context, uuid) + mock_destroy_cpulse.assert_called_once_with(uuid) + self.assertEqual(self.context, cpulse._context) + + def test_save(self): + uuid = self.cpulsetest['uuid'] + with mock.patch.object(self.dbapi, 'get_test_by_uuid', + autospec=True) as mock_get_cpulse: + mock_get_cpulse.return_value = self.cpulsetest + with mock.patch.object(self.dbapi, 'update_test', + autospec=True) as mock_update_cpulse: + cpulse = objects.Cpulse.get_by_uuid(self.context, uuid) + cpulse.state = 'running' + cpulse.save() + + mock_get_cpulse.assert_called_once_with(self.context, uuid) + mock_update_cpulse.assert_called_once_with( + uuid, {'state': 'running'}) + self.assertEqual(self.context, cpulse._context) + + def test_refresh(self): + uuid = self.cpulsetest['uuid'] + new_uuid = cloudpulse_utils.generate_uuid() + returns = [dict(self.cpulsetest, uuid=uuid), + dict(self.cpulsetest, uuid=new_uuid)] + expected = [mock.call(self.context, uuid), + mock.call(self.context, uuid)] + with mock.patch.object(self.dbapi, 'get_test_by_uuid', + side_effect=returns, + autospec=True) as mock_get_cpulse: + cpulse = objects.Cpulse.get_by_uuid(self.context, uuid) + self.assertEqual(uuid, cpulse.uuid) + cpulse.refresh() + self.assertEqual(new_uuid, cpulse.uuid) + self.assertEqual(expected, mock_get_cpulse.call_args_list) + self.assertEqual(self.context, cpulse._context)