def test_host_group_add(self):
'''Test that a host can be added to a HostGroup'''
import budgie
import tests
dummy_group1 = budgie.HostGroup()
dummy_group1.add('localhost')
localhost1 = tests.the_baker(budgie.ssh('localhost1'))
dummy_group1.add(localhost1)
self.assertTrue(dummy_group1.has_key('localhost'))
self.assertTrue(isinstance(dummy_group1['localhost'], budgie.ssh))
self.assertTrue(dummy_group1.has_key('localhost1'))
self.assertTrue(isinstance(dummy_group1['localhost1'], budgie.ssh))
dummy_group2 = budgie.HostGroup()
localhost = the_baker(budgie.ssh('localhost'))
dummy_group2.add([localhost, localhost1])
self.assertTrue(dummy_group2.has_key('localhost'))
self.assertTrue(isinstance(dummy_group2['localhost'], budgie.ssh))
self.assertTrue(dummy_group2.has_key('localhost1'))
self.assertTrue(isinstance(dummy_group2['localhost1'], budgie.ssh))
def test_host_group_run(self):
'''Test that a command can be run agains a hostgroup'''
import budgie
localhost = the_baker(budgie.ssh('localhost'))
localhost1 = the_baker(budgie.ssh('localhost1'))
dummy_group = budgie.HostGroup([localhost, localhost1])
whoami_result = dummy_group.whoami()
self.assertEqual(
whoami_result['localhost'].strip('\n'),
whoami_result['localhost1'].strip('\n')
)
def test_host_group_remove(self):
'''Test that a host can be removed from HostGroup'''
import budgie
localhost = the_baker(budgie.ssh('localhost'))
localhost1 = the_baker(budgie.ssh('localhost1'))
dummy_group = budgie.HostGroup([localhost, localhost1])
del dummy_group['localhost']
if sys.version_info[:2] == (2,6):
try:
dummy_group['localhost']
except:
self.assertTrue(isinstance(sys.exc_info()[1], KeyError))
else:
with self.assertRaises(KeyError):
dummy_group['localhost']
def test_ssh_run_cmd(self):
'''Test that a command can be executed on the remote host'''
import budgie
self.assertEqual(
str(the_baker(budgie.ssh('127.0.0.1')).whoami()).strip('\n'),
os.environ['USER']
)
def test_ssh_call(self):
'''Test the __call__ that raises the NotImplementedError'''
import budgie
# with self.assertRaises(NotImplementedError):
# budgie.ssh('localhost')()
self.assertRaises(NotImplementedError,budgie.ssh('localhost'))