7ac3c03212
I agree - bare except is bad - it can cause ctrl-c to stop working, which is totally evil. (I mean, unless that is the intent, in which case it's the right thing to do. None of these cases are cases when we desire ctrl-c to stop working) Change-Id: Ieffa26e8da77199c9ef7238a6c9f0704fc581ba6
114 lines
4.2 KiB
Python
114 lines
4.2 KiB
Python
# 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.
|
|
|
|
"""
|
|
test_cluster_templates
|
|
----------------------------------
|
|
|
|
Funself.ctional tests for `shade` cluster_template methods.
|
|
"""
|
|
|
|
from testtools import content
|
|
|
|
from shade.tests.functional import base
|
|
|
|
import os
|
|
import subprocess
|
|
|
|
|
|
class TestClusterTemplate(base.BaseFunctionalTestCase):
|
|
|
|
def setUp(self):
|
|
super(TestClusterTemplate, self).setUp()
|
|
if not self.user_cloud.has_service('container-infra'):
|
|
self.skipTest('Container service not supported by cloud')
|
|
self.ct = None
|
|
|
|
def test_cluster_templates(self):
|
|
'''Test cluster_templates functionality'''
|
|
name = 'fake-cluster_template'
|
|
server_type = 'vm'
|
|
public = False
|
|
image_id = 'fedora-atomic-f23-dib'
|
|
tls_disabled = False
|
|
registry_enabled = False
|
|
coe = 'kubernetes'
|
|
keypair_id = 'testkey'
|
|
|
|
self.addDetail('cluster_template', content.text_content(name))
|
|
self.addCleanup(self.cleanup, name)
|
|
|
|
# generate a keypair to add to nova
|
|
ssh_directory = '/tmp/.ssh'
|
|
if not os.path.isdir(ssh_directory):
|
|
os.mkdir(ssh_directory)
|
|
subprocess.call(
|
|
['ssh-keygen', '-t', 'rsa', '-N', '', '-f',
|
|
'%s/id_rsa_shade' % ssh_directory])
|
|
|
|
# add keypair to nova
|
|
with open('%s/id_rsa_shade.pub' % ssh_directory) as f:
|
|
key_content = f.read()
|
|
self.user_cloud.create_keypair('testkey', key_content)
|
|
|
|
# Test we can create a cluster_template and we get it returned
|
|
self.ct = self.user_cloud.create_cluster_template(
|
|
name=name, image_id=image_id,
|
|
keypair_id=keypair_id, coe=coe)
|
|
self.assertEqual(self.ct['name'], name)
|
|
self.assertEqual(self.ct['image_id'], image_id)
|
|
self.assertEqual(self.ct['keypair_id'], keypair_id)
|
|
self.assertEqual(self.ct['coe'], coe)
|
|
self.assertEqual(self.ct['registry_enabled'], registry_enabled)
|
|
self.assertEqual(self.ct['tls_disabled'], tls_disabled)
|
|
self.assertEqual(self.ct['public'], public)
|
|
self.assertEqual(self.ct['server_type'], server_type)
|
|
|
|
# Test that we can list cluster_templates
|
|
cluster_templates = self.user_cloud.list_cluster_templates()
|
|
self.assertIsNotNone(cluster_templates)
|
|
|
|
# Test we get the same cluster_template with the
|
|
# get_cluster_template method
|
|
cluster_template_get = self.user_cloud.get_cluster_template(
|
|
self.ct['uuid'])
|
|
self.assertEqual(cluster_template_get['uuid'], self.ct['uuid'])
|
|
|
|
# Test the get method also works by name
|
|
cluster_template_get = self.user_cloud.get_cluster_template(name)
|
|
self.assertEqual(cluster_template_get['name'], self.ct['name'])
|
|
|
|
# Test we can update a field on the cluster_template and only that
|
|
# field is updated
|
|
cluster_template_update = self.user_cloud.update_cluster_template(
|
|
self.ct['uuid'], 'replace', tls_disabled=True)
|
|
self.assertEqual(
|
|
cluster_template_update['uuid'], self.ct['uuid'])
|
|
self.assertEqual(cluster_template_update['tls_disabled'], True)
|
|
|
|
# Test we can delete and get True returned
|
|
cluster_template_delete = self.user_cloud.delete_cluster_template(
|
|
self.ct['uuid'])
|
|
self.assertTrue(cluster_template_delete)
|
|
|
|
def cleanup(self, name):
|
|
if self.ct:
|
|
try:
|
|
self.user_cloud.delete_cluster_template(self.ct['name'])
|
|
except Exception:
|
|
pass
|
|
|
|
# delete keypair
|
|
self.user_cloud.delete_keypair('testkey')
|
|
os.unlink('/tmp/.ssh/id_rsa_shade')
|
|
os.unlink('/tmp/.ssh/id_rsa_shade.pub')
|