def setUp(self):
super(TestClusterRootController, self).setUp()
self.controller = ClusterRootController()
class TestClusterRootController(trove_testtools.TestCase):
def setUp(self):
super(TestClusterRootController, self).setUp()
self.controller = ClusterRootController()
@patch.object(ClusterRootController, "cluster_root_index")
def test_root_index_cluster(self, mock_cluster_root_index):
req = Mock()
tenant_id = Mock()
uuid = utils.generate_uuid()
is_cluster = True
self.controller.root_index(req, tenant_id, uuid, is_cluster)
mock_cluster_root_index.assert_called_with(req, tenant_id, uuid)
@patch.object(ClusterRootController, "instance_root_index")
def test_root_index_instance(self, mock_instance_root_index):
req = Mock()
tenant_id = Mock()
uuid = utils.generate_uuid()
is_cluster = False
self.controller.root_index(req, tenant_id, uuid, is_cluster)
mock_instance_root_index.assert_called_with(req, tenant_id, uuid)
@patch.object(ClusterRootController, "cluster_root_create")
def test_root_create_cluster(self, mock_cluster_root_create):
req = Mock()
body = Mock()
tenant_id = Mock()
uuid = utils.generate_uuid()
is_cluster = True
self.controller.root_create(req, body, tenant_id, uuid, is_cluster)
mock_cluster_root_create.assert_called_with(req, body, tenant_id, uuid)
@patch.object(ClusterRootController, "check_cluster_instance_actions")
@patch.object(ClusterRootController, "instance_root_create")
def test_root_create_instance(self, mock_instance_root_create, mock_check):
req = Mock()
body = Mock()
tenant_id = Mock()
uuid = utils.generate_uuid()
is_cluster = False
self.controller.root_create(req, body, tenant_id, uuid, is_cluster)
mock_check.assert_called_with(uuid)
mock_instance_root_create.assert_called_with(req, body, uuid)
@patch.object(models.ClusterRoot, "load")
def test_instance_root_index(self, mock_cluster_root_load):
context = Mock()
req = Mock()
req.environ = Mock()
req.environ.__getitem__ = Mock(return_value=context)
tenant_id = Mock()
instance_id = utils.generate_uuid()
self.controller.instance_root_index(req, tenant_id, instance_id)
mock_cluster_root_load.assert_called_with(context, instance_id)
@patch.object(models.ClusterRoot, "load",
side_effect=exception.UnprocessableEntity())
def test_instance_root_index_exception(self, mock_cluster_root_load):
context = Mock()
req = Mock()
req.environ = Mock()
req.environ.__getitem__ = Mock(return_value=context)
tenant_id = Mock()
instance_id = utils.generate_uuid()
self.assertRaises(
exception.UnprocessableEntity,
self.controller.instance_root_index,
req, tenant_id, instance_id
)
mock_cluster_root_load.assert_called_with(context, instance_id)
@patch.object(ClusterRootController, "instance_root_index")
@patch.object(ClusterRootController, "_get_cluster_instance_id")
def test_cluster_root_index(self, mock_get_cluster_instance,
mock_instance_root_index):
req = Mock()
tenant_id = Mock()
cluster_id = utils.generate_uuid()
single_instance_id = Mock()
mock_get_cluster_instance.return_value = (single_instance_id, Mock())
self.controller.cluster_root_index(req, tenant_id, cluster_id)
mock_get_cluster_instance.assert_called_with(tenant_id, cluster_id)
mock_instance_root_index.assert_called_with(req, tenant_id,
single_instance_id)
@patch.object(ClusterRootController, "instance_root_create")
@patch.object(ClusterRootController, "_get_cluster_instance_id")
def test_cluster_root_create(self, mock_get_cluster_instance,
mock_instance_root_create):
req = Mock()
body = Mock()
tenant_id = Mock()
cluster_id = utils.generate_uuid()
single_instance_id = Mock()
cluster_instances = Mock()
mock_get_cluster_instance.return_value = (single_instance_id,
cluster_instances)
self.controller.cluster_root_create(req, body, tenant_id, cluster_id)
mock_get_cluster_instance.assert_called_with(tenant_id, cluster_id)
mock_instance_root_create.assert_called_with(req, body,
single_instance_id,
cluster_instances)
@patch.object(DBInstance, "find_all")
def test_get_cluster_instance_id(self, mock_find_all):
tenant_id = Mock()
cluster_id = Mock()
db_inst_1 = Mock()
db_inst_1.id.return_value = utils.generate_uuid()
db_inst_2 = Mock()
db_inst_2.id.return_value = utils.generate_uuid()
cluster_instances = [db_inst_1, db_inst_2]
mock_find_all.return_value.all.return_value = cluster_instances
ret = self.controller._get_cluster_instance_id(tenant_id, cluster_id)
self.assertTrue(db_inst_1.id, ret[0])
self.assertTrue(cluster_instances, ret[1])
@patch.object(models.ClusterRoot, "create")
def test_instance_root_create(self, mock_cluster_root_create):
user = Mock()
context = Mock()
context.user = Mock()
context.user.__getitem__ = Mock(return_value=user)
req = Mock()
req.environ = Mock()
req.environ.__getitem__ = Mock(return_value=context)
password = Mock()
body = {'password': password}
instance_id = utils.generate_uuid()
cluster_instances = Mock()
self.controller.instance_root_create(
req, body, instance_id, cluster_instances)
mock_cluster_root_create.assert_called_with(
context, instance_id, context.user, password, cluster_instances)
@patch.object(models.ClusterRoot, "create")
def test_instance_root_create_no_body(self, mock_cluster_root_create):
user = Mock()
context = Mock()
context.user = Mock()
context.user.__getitem__ = Mock(return_value=user)
req = Mock()
req.environ = Mock()
req.environ.__getitem__ = Mock(return_value=context)
password = None
body = None
instance_id = utils.generate_uuid()
cluster_instances = Mock()
self.controller.instance_root_create(
req, body, instance_id, cluster_instances)
mock_cluster_root_create.assert_called_with(
context, instance_id, context.user, password, cluster_instances)
def setUp(self):
super(TestClusterRootController, self).setUp()
self.controller = ClusterRootController()
class TestClusterRootController(trove_testtools.TestCase):
def setUp(self):
super(TestClusterRootController, self).setUp()
self.controller = ClusterRootController()
@patch.object(ClusterRootController, "cluster_root_index")
def test_root_index_cluster(self, mock_cluster_root_index):
req = Mock()
tenant_id = Mock()
uuid = utils.generate_uuid()
is_cluster = True
self.controller.root_index(req, tenant_id, uuid, is_cluster)
mock_cluster_root_index.assert_called_with(req, tenant_id, uuid)
@patch.object(ClusterRootController, "instance_root_index")
def test_root_index_instance(self, mock_instance_root_index):
req = Mock()
tenant_id = Mock()
uuid = utils.generate_uuid()
is_cluster = False
self.controller.root_index(req, tenant_id, uuid, is_cluster)
mock_instance_root_index.assert_called_with(req, tenant_id, uuid)
@patch.object(ClusterRootController, "cluster_root_create")
def test_root_create_cluster(self, mock_cluster_root_create):
req = Mock()
body = Mock()
tenant_id = Mock()
uuid = utils.generate_uuid()
is_cluster = True
self.controller.root_create(req, body, tenant_id, uuid, is_cluster)
mock_cluster_root_create.assert_called_with(req, body, tenant_id, uuid)
@patch.object(ClusterRootController, "check_cluster_instance_actions")
@patch.object(ClusterRootController, "instance_root_create")
def test_root_create_instance(self, mock_instance_root_create, mock_check):
req = Mock()
body = Mock()
tenant_id = Mock()
uuid = utils.generate_uuid()
is_cluster = False
self.controller.root_create(req, body, tenant_id, uuid, is_cluster)
mock_check.assert_called_with(uuid)
mock_instance_root_create.assert_called_with(req, body, uuid)
@patch.object(models.ClusterRoot, "load")
def test_instance_root_index(self, mock_cluster_root_load):
context = Mock()
req = Mock()
req.environ = Mock()
req.environ.__getitem__ = Mock(return_value=context)
tenant_id = Mock()
instance_id = utils.generate_uuid()
self.controller.instance_root_index(req, tenant_id, instance_id)
mock_cluster_root_load.assert_called_with(context, instance_id)
@patch.object(models.ClusterRoot,
"load",
side_effect=exception.UnprocessableEntity())
def test_instance_root_index_exception(self, mock_cluster_root_load):
context = Mock()
req = Mock()
req.environ = Mock()
req.environ.__getitem__ = Mock(return_value=context)
tenant_id = Mock()
instance_id = utils.generate_uuid()
self.assertRaises(exception.UnprocessableEntity,
self.controller.instance_root_index, req, tenant_id,
instance_id)
mock_cluster_root_load.assert_called_with(context, instance_id)
@patch.object(ClusterRootController, "instance_root_index")
@patch.object(ClusterRootController, "_get_cluster_instance_id")
def test_cluster_root_index(self, mock_get_cluster_instance,
mock_instance_root_index):
req = Mock()
tenant_id = Mock()
cluster_id = utils.generate_uuid()
single_instance_id = Mock()
mock_get_cluster_instance.return_value = (single_instance_id, Mock())
self.controller.cluster_root_index(req, tenant_id, cluster_id)
mock_get_cluster_instance.assert_called_with(tenant_id, cluster_id)
mock_instance_root_index.assert_called_with(req, tenant_id,
single_instance_id)
@patch.object(ClusterRootController, "instance_root_create")
@patch.object(ClusterRootController, "_get_cluster_instance_id")
def test_cluster_root_create(self, mock_get_cluster_instance,
mock_instance_root_create):
req = Mock()
body = Mock()
tenant_id = Mock()
cluster_id = utils.generate_uuid()
single_instance_id = Mock()
cluster_instances = Mock()
mock_get_cluster_instance.return_value = (single_instance_id,
cluster_instances)
self.controller.cluster_root_create(req, body, tenant_id, cluster_id)
mock_get_cluster_instance.assert_called_with(tenant_id, cluster_id)
mock_instance_root_create.assert_called_with(req, body,
single_instance_id,
cluster_instances)
@patch.object(DBInstance, "find_all")
def test_get_cluster_instance_id(self, mock_find_all):
tenant_id = Mock()
cluster_id = Mock()
db_inst_1 = Mock()
db_inst_1.id.return_value = utils.generate_uuid()
db_inst_2 = Mock()
db_inst_2.id.return_value = utils.generate_uuid()
cluster_instances = [db_inst_1, db_inst_2]
mock_find_all.return_value.all.return_value = cluster_instances
ret = self.controller._get_cluster_instance_id(tenant_id, cluster_id)
self.assertTrue(db_inst_1.id, ret[0])
self.assertTrue(cluster_instances, ret[1])
@patch.object(models.ClusterRoot, "create")
def test_instance_root_create(self, mock_cluster_root_create):
user = Mock()
context = Mock()
context.user = Mock()
context.user.__getitem__ = Mock(return_value=user)
req = Mock()
req.environ = Mock()
req.environ.__getitem__ = Mock(return_value=context)
password = Mock()
body = {'password': password}
instance_id = utils.generate_uuid()
cluster_instances = Mock()
self.controller.instance_root_create(req, body, instance_id,
cluster_instances)
mock_cluster_root_create.assert_called_with(context, instance_id,
context.user, password,
cluster_instances)
@patch.object(models.ClusterRoot, "create")
def test_instance_root_create_no_body(self, mock_cluster_root_create):
user = Mock()
context = Mock()
context.user = Mock()
context.user.__getitem__ = Mock(return_value=user)
req = Mock()
req.environ = Mock()
req.environ.__getitem__ = Mock(return_value=context)
password = None
body = None
instance_id = utils.generate_uuid()
cluster_instances = Mock()
self.controller.instance_root_create(req, body, instance_id,
cluster_instances)
mock_cluster_root_create.assert_called_with(context, instance_id,
context.user, password,
cluster_instances)
def setUp(self):
super(TestClusterRootController, self).setUp()
self.context = trove_testtools.TroveTestContext(self)
self.controller = ClusterRootController()
def setUp(self):
super(TestClusterRootController, self).setUp()
self.context = trove_testtools.TroveTestContext(self)
self.controller = ClusterRootController()
