Esempio n. 1
0
    def test_get_server_id(self):
        def api_method(self, system_id, arg1, arg2): pass
        args = ('system_id_xml', 'arg1', 'arg2')

        # mock rhnServer.get rhnServer.get(sysid_xml).getid()
        rhnserver_got = Mock()
        rhnserver_got.getid = Mock(return_value="10001000")
        auditlog.rhnServer.get = Mock(return_value=rhnserver_got)

        self.assertEqual(auditlog._get_server_id(api_method, args),
                         ('10001000', ('10001000', 'arg1', 'arg2')))
        self.assertEqual(auditlog.rhnServer.get.call_args,
                         (('system_id_xml',), {}), )
    def test_get_server_id(self):
        def api_method(self, system_id, arg1, arg2):
            pass
        args = ('system_id_xml', 'arg1', 'arg2')

        # mock rhnServer.get rhnServer.get(sysid_xml).getid()
        rhnserver_got = Mock()
        rhnserver_got.getid = Mock(return_value="10001000")
        auditlog.rhnServer.get = Mock(return_value=rhnserver_got)

        self.assertEqual(auditlog._get_server_id(api_method, args),
                         ('10001000', ('10001000', 'arg1', 'arg2')))
        self.assertEqual(auditlog.rhnServer.get.call_args,
                         (('system_id_xml',), {}), )