Пример #1
0
    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))
Пример #2
0
    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')
        )
Пример #3
0
    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']
Пример #4
0
 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']
     )
Пример #5
0
 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'))