def write_config(self, dir_ = '.', prop = 'hadoop.properties', nodes = 'nodes-byon.yaml'): ''' Writes the two configuration files to the specified directory ''' pf = open( '%s/%s' % (dir_, prop), 'w') pf.write('whirr.cluster-name=%s\n' % self._cluster.name()) pf.write('whirr.cluster-user=vagrant\n') pf.write('whirr.instance-templates=%s\n' % self._cluster.config()['hadoop']['instance-templates']) if self._cluster.config()['hadoop'].has_key('templates-namenode'): pf.write('whirr.templates.hadoop-namenode+hadoop-jobtracker.byon-instance-ids=%s\n' % self._cluster.config()['hadoop']['templates-namenode']) if self._cluster.config()['hadoop'].has_key('templates-datanode'): pf.write('whirr.templates.hadoop-datanode+hadoop-tasktracker.byon-instance-ids=%s\n' % self._cluster.config()['hadoop']['templates-datanode']) pf.write('whirr.service-name=byon\n') pf.write('whirr.provider=byon\n') pf.write('jclouds.byon.endpoint=file://%s/%s\n' % (dir_, nodes)) pf.write('whirr.provider=whirr.java.install-function = install_oracle_jdk6\n') if self._cluster.config()['hadoop'].has_key('version'): pf.write('whirr.hadoop.version=%s\n' % self._cluster.config()['hadoop']['version']) pf.write('whirr.hadoop.tarball.url=http://archive.apache.org/dist/hadoop/core/hadoop-${whirr.hadoop.version}/hadoop-${whirr.hadoop.version}.tar.gz\n') else: pf.write('whirr.hadoop.install-function=install_cdh_hadoop\n') pf.write('whirr.hadoop.configure-function=configure_cdh_hadoop\n') pf.close() nf = open( '%s/%s' % (dir_, nodes), 'w' ) nf.write('nodes:\n') for m in sorted(self._cluster.machines()): if not m == 'em1': # exclude dedicated EM server mach = self._cluster.machine(m) nf.write(' - id: %s\n' % m) nf.write(' hostname: %s\n' % mach['ip']) nf.write(' os_arch: x86_64\n') boxtype = self._cluster.config()['boxtype'] nf.write(' os_family: %s\n' % vagboxes.get_os_family(boxtype)) nf.write(' os_description: %s\n' % vagboxes.get_description(boxtype)) nf.write(' os_version: %s\n' % vagboxes.get_os_version(boxtype)) nf.write(' group: vagrant\n') nf.write(' username: vagrant\n') nf.write(' credential_url: file://%s/insecure_private_key\n' % common.props['vmi.vagrantvmi.vagrantroot']) nf.write(' sudo_password:\n') nf.close()
def testBoxes(self): self.assertEqual( vagboxes.get_default_pkgtype('cal-precise64'), 'deb') self.assertEqual( vagboxes.get_os_family('cal-precise64'), 'ubuntu') self.assertEqual( vagboxes.get_os_version('cal-precise64'), '12.04') self.assertEqual( vagboxes.get_description('cal-precise64'), 'Ubuntu 12.04.1 LTS (Precise Pangolin)')