def test_replace(): with warnings.catch_warnings(): fname = os.tmpnam() t1 = 'some data' t2 = 'other text' header = '-' * 50 footer = '=' * 50 mkfl = lambda x : header + '\n' + x + '\n' + footer core.put_rf(fname, mkfl(t1)) try: ok(core.replace_in_file(fname, t2, t1)) == False ok(core.replace_in_file(fname, t1, t2)) == True ok(core.get_rf(fname)) == mkfl(t2) sudo('chown root.root ' + fname) sudo('chmod o-w ' + fname) ok(lambda : core.replace_in_file(fname, t2, t1)).raises(SystemExit) ok(core.replace_in_file(fname, t1, t2, use_sudo=True)) == False ok(core.replace_in_file(fname, t2, t1, use_sudo=True)) == True ok(core.get_rf(fname)) == mkfl(t1) finally: sudo('rm ' + fname)
def prepare_node(): with settings(hide('warnings', 'stderr'), warn_only=True): stop_riak() ip = env.host_string.split('@')[1] if replace_in_file('/etc/riak/vm.args', '-name [email protected]', '-name riak@' + ip, use_sudo=True): replace_in_file('/etc/riak/app.config', '{http, [ {"127.0.0.1", 8098 } ]}', '{{http, [ {{"{0}", 8098 }} ]}}'.format(ip), use_sudo=True) with settings(hide('warnings', 'stderr'), warn_only=True): sudo("riak-admin reip [email protected] riak@{0}".format( env.host_string.split('@')[1]))
def install_nova(ip, net, net_prefix, lvm_dev, ext_network, proj_name, nova_adm='ubuntu', dbpasswd='nova', compute_backend='lxc'): install('bridge-utils,postgresql,python-psycopg2') replace_in_file("/etc/postgresql/9.1/main/postgresql.conf", "#listen_addresses\s+=\s+'localhost'(.*)", r"listen_addresses = '*'\1", use_sudo=True) pg_hba = '/etc/postgresql/9.1/main/pg_hba.conf' add = "host all all 0.0.0.0/0 md5" sudo("chmod a+rw " + pg_hba) try: append(pg_hba, add) finally: sudo("chmod 640 " + pg_hba) sudo("/etc/init.d/postgresql restart") install('glance') cmd = "CREATE user glancedbadmin;\n" + \ "ALTER user glancedbadmin with password '{0}';\n" + \ 'CREATE DATABASE glance;\n' + \ 'GRANT ALL PRIVILEGES ON database glance TO glancedbadmin;\n' psql(cmd.format(dbpasswd)) replace_in_file("/etc/glance/glance-registry.conf", 'sql_connection = .*?$', 'sql_connection = postgresql://glancedbadmin:{0}@{1}/glance'\ .format(dbpasswd,ip), use_sudo=True) sudo('restart glance-registry') install('rabbitmq-server,nova-common,nova-doc,python-nova,nova-api,' + \ 'nova-network,nova-volume,nova-objectstore,nova-scheduler,' + \ 'nova-compute,euca2ools,unzip,nova-compute-' + compute_backend) cmd = 'CREATE user novadbadmin;\n' + \ "ALTER user novadbadmin with password '{0}';\n" + \ 'CREATE DATABASE nova;\n' + \ 'GRANT ALL PRIVILEGES ON database nova TO novadbadmin;\n' psql(cmd.format(dbpasswd)) cfg = nova_config.format(ip, net, '8', dbpasswd, net_prefix) put_rf('/etc/nova/nova.conf', cfg , use_sudo=True) use_lvm = False if use_lvm: install('iscsitarget,iscsitarget-dkms') sudo("sed -i 's/false/true/g' /etc/default/iscsitarget") sudo("service iscsitarget restart") sudo("pvcreate " + lvm_dev) sudo("vgcreate nova-volumes " + lvm_dev) sudo("chown -R root:nova /etc/nova") sudo("chmod 644 /etc/nova/nova.conf") all_nova_services = ('libvirt-bin,nova-network,nova-compute,nova-api,' + \ 'nova-objectstore,nova-scheduler,' + \ 'glance-api,glance-registry,nova-volume').split(',') for service in all_nova_services: upstart_restart(service) sudo('nova-manage db sync') sudo('nova-manage network create private {0} 1 255'.format(net)) sudo('nova-manage floating create --ip_range=' + ext_network) sudo('nova-manage user admin novaadmin') sudo('nova-manage project create {0} novaadmin'.format(proj_name)) for service in all_nova_services: upstart_restart(service) sudo('mkdir -p /home/{0}/creds'.format(nova_adm)) sudo('nova-manage project zipfile proj novaadmin ' + \ '/home/{0}/creds/novacreds.zip'.format(nova_adm)) with cd('/home/{0}/creds'.format(nova_adm)): sudo('unzip novacreds.zip') sudo('chown {0}:{0} /home/{0}/creds -R'.format(nova_adm)) res = sudo('nova-manage user exports novaadmin') EC2_ACCESS_KEY = None EC2_SECRET_KEY = None for string in str(res).split('\n'): string = string.strip() if string.startswith('export EC2_ACCESS_KEY='): EC2_ACCESS_KEY = string[len('export EC2_ACCESS_KEY='):] elif string.startswith('export EC2_SECRET_KEY='): EC2_SECRET_KEY = string[len('export EC2_SECRET_KEY='):] replace_in_file('/home/{0}/creds/novarc'.format(nova_adm), 'export EC2_ACCESS_KEY="novaadmin:proj"'.format(proj_name), 'export EC2_ACCESS_KEY="{0}:{1}"'.format(EC2_ACCESS_KEY, proj_name)) replace_in_file('/home/{0}/creds/novarc'.format(nova_adm), 'export NOVA_PROJECT_ID="proj"', 'export NOVA_PROJECT_ID="{0}"'.format(proj_name)) return