Esempio n. 1
0
    def test_search(self):
        '''
        Test if it run an arbitrary LDAP query and return the results.
        '''
        class MockConnect(object):
            '''
            Mocking _connect method
            '''
            def __init__(self):
                self.bdn = None
                self.scope = None
                self._filter = None
                self.attrs = None

            def search_s(self, bdn, scope, _filter, attrs):
                '''
                Mock function for search_s
                '''
                self.bdn = bdn
                self.scope = scope
                self._filter = _filter
                self.attrs = attrs
                return 'SALT'

        mock = MagicMock(return_value=True)
        with patch.dict(ldapmod.__salt__, {'config.option': mock}):
            with patch.object(ldapmod, '_connect',
                              MagicMock(return_value=MockConnect())):
                with patch.object(time, 'time', MagicMock(return_value=8e-04)):
                    self.assertDictEqual(ldapmod.search(filter='myhost'),
                                         {'count': 4, 'results': 'SALT',
                                          'time': {'raw': '0.0',
                                                   'human': '0.0ms'}})
Esempio n. 2
0
    def test_search(self):
        """
        Test if it run an arbitrary LDAP query and return the results.
        """
        class MockConnect(object):
            """
            Mocking _connect method
            """
            def __init__(self):
                self.bdn = None
                self.scope = None
                self._filter = None
                self.attrs = None

            def search_s(self, bdn, scope, _filter, attrs):
                """
                Mock function for search_s
                """
                self.bdn = bdn
                self.scope = scope
                self._filter = _filter
                self.attrs = attrs
                return "SALT"

        mock = MagicMock(return_value=True)
        with patch.dict(ldapmod.__salt__, {"config.option": mock}):
            with patch.object(ldapmod, "_connect",
                              MagicMock(return_value=MockConnect())):
                with patch.object(time, "time", MagicMock(return_value=8e-04)):
                    self.assertDictEqual(
                        ldapmod.search(filter="myhost"),
                        {
                            "count": 4,
                            "results": "SALT",
                            "time": {
                                "raw": "0.0",
                                "human": "0.0ms"
                            },
                        },
                    )
Esempio n. 3
0
    def test_search(self):
        '''
        Test if it run an arbitrary LDAP query and return the results.
        '''
        class MockConnect(object):
            '''
            Mocking _connect method
            '''
            def __init__(self):
                self.bdn = None
                self.scope = None
                self._filter = None
                self.attrs = None

            def search_s(self, bdn, scope, _filter, attrs):
                '''
                Mock function for search_s
                '''
                self.bdn = bdn
                self.scope = scope
                self._filter = _filter
                self.attrs = attrs
                return 'SALT'

        mock = MagicMock(return_value=True)
        with patch.dict(ldapmod.__salt__, {'config.option': mock}):
            with patch.object(ldapmod, '_connect',
                              MagicMock(return_value=MockConnect())):
                with patch.object(time, 'time', MagicMock(return_value=8e-04)):
                    self.assertDictEqual(
                        ldapmod.search(filter='myhost'), {
                            'count': 4,
                            'results': 'SALT',
                            'time': {
                                'raw': '0.0',
                                'human': '0.0ms'
                            }
                        })