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)')
