92 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			92 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
# vim: tabstop=4 shiftwidth=4 softtabstop=4
 | 
						|
 | 
						|
# Copyright 2010 United States Government as represented by the
 | 
						|
# Administrator of the National Aeronautics and Space Administration.
 | 
						|
# All Rights Reserved.
 | 
						|
#
 | 
						|
#    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 os
 | 
						|
import random
 | 
						|
import sys
 | 
						|
import unittest
 | 
						|
import zipfile
 | 
						|
 | 
						|
# If ../nova/__init__.py exists, add ../ to Python search path, so that
 | 
						|
# it will override what happens to be installed in /usr/(local/)lib/python...
 | 
						|
possible_topdir = os.path.normpath(os.path.join(os.path.abspath(sys.argv[0]),
 | 
						|
                                   os.pardir,
 | 
						|
                                   os.pardir))
 | 
						|
if os.path.exists(os.path.join(possible_topdir, 'nova', '__init__.py')):
 | 
						|
    sys.path.insert(0, possible_topdir)
 | 
						|
 | 
						|
from smoketests import flags
 | 
						|
from smoketests import base
 | 
						|
 | 
						|
 | 
						|
FLAGS = flags.FLAGS
 | 
						|
 | 
						|
# TODO(devamcar): Use random tempfile
 | 
						|
ZIP_FILENAME = '/tmp/nova-me-x509.zip'
 | 
						|
 | 
						|
TEST_PREFIX = 'test%s' % int(random.random() * 1000000)
 | 
						|
TEST_USERNAME = '%suser' % TEST_PREFIX
 | 
						|
TEST_PROJECTNAME = '%sproject' % TEST_PREFIX
 | 
						|
 | 
						|
 | 
						|
class AdminSmokeTestCase(base.SmokeTestCase):
 | 
						|
    def setUp(self):
 | 
						|
        import nova_adminclient as adminclient
 | 
						|
        self.admin = adminclient.NovaAdminClient(
 | 
						|
            access_key=os.getenv('EC2_ACCESS_KEY'),
 | 
						|
            secret_key=os.getenv('EC2_SECRET_KEY'),
 | 
						|
            clc_url=os.getenv('EC2_URL'),
 | 
						|
            region=FLAGS.region)
 | 
						|
 | 
						|
 | 
						|
class UserTests(AdminSmokeTestCase):
 | 
						|
    """ Test admin credentials and user creation. """
 | 
						|
 | 
						|
    def test_001_admin_can_connect(self):
 | 
						|
        conn = self.admin.connection_for('admin', 'admin')
 | 
						|
        self.assert_(conn)
 | 
						|
 | 
						|
    def test_002_admin_can_create_user(self):
 | 
						|
        user = self.admin.create_user(TEST_USERNAME)
 | 
						|
        self.assertEqual(user.username, TEST_USERNAME)
 | 
						|
 | 
						|
    def test_003_admin_can_create_project(self):
 | 
						|
        project = self.admin.create_project(TEST_PROJECTNAME,
 | 
						|
                                            TEST_USERNAME)
 | 
						|
        self.assertEqual(project.projectname, TEST_PROJECTNAME)
 | 
						|
 | 
						|
    def test_004_user_can_download_credentials(self):
 | 
						|
        buf = self.admin.get_zip(TEST_USERNAME, TEST_PROJECTNAME)
 | 
						|
        output = open(ZIP_FILENAME, 'w')
 | 
						|
        output.write(buf)
 | 
						|
        output.close()
 | 
						|
 | 
						|
        zip = zipfile.ZipFile(ZIP_FILENAME, 'a', zipfile.ZIP_DEFLATED)
 | 
						|
        bad = zip.testzip()
 | 
						|
        zip.close()
 | 
						|
 | 
						|
        self.failIf(bad)
 | 
						|
 | 
						|
    def test_999_tearDown(self):
 | 
						|
        self.admin.delete_project(TEST_PROJECTNAME)
 | 
						|
        self.admin.delete_user(TEST_USERNAME)
 | 
						|
        try:
 | 
						|
            os.remove(ZIP_FILENAME)
 | 
						|
        except:
 | 
						|
            pass
 |