1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
"""
"""
# Created on 2013.06.06
#
# Author: Giovanni Cannata
#
# Copyright 2013 - 2020 Giovanni Cannata
#
# This file is part of ldap3.
#
# ldap3 is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published
# by the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# ldap3 is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with ldap3 in the COPYING and COPYING.LESSER files.
# If not, see <http://www.gnu.org/licenses/>.
import unittest
from copy import deepcopy
from time import sleep
from test.config import get_connection, drop_connection, add_user, random_id, get_add_user_attributes,\
test_user_password, generate_dn, test_base
testcase_id = ''
class Test(unittest.TestCase):
def setUp(self):
global testcase_id
testcase_id = random_id()
self.connection = get_connection()
self.delete_at_teardown = []
def tearDown(self):
drop_connection(self.connection, self.delete_at_teardown)
self.assertFalse(self.connection.bound)
def test_add(self):
self.delete_at_teardown.append(add_user(self.connection, testcase_id, 'add-operation-1'))
self.assertEqual('success', self.delete_at_teardown[0][1]['description'])
def test_add_bytes(self):
self.delete_at_teardown.append(add_user(self.connection, testcase_id, 'add-operation-2', test_bytes=True))
self.assertEqual('success', self.delete_at_teardown[0][1]['description'])
def test_unmodified_attributes_dict(self):
attributes = get_add_user_attributes(testcase_id, 'add-operation-3', test_user_password)
object_class = attributes.pop('objectClass')
copy_of_attributes = deepcopy(attributes)
dn = generate_dn(test_base, testcase_id, 'add-operation-3')
self.connection.add(dn, object_class, attributes)
sleep(2)
self.connection.delete(dn)
self.assertDictEqual(copy_of_attributes, attributes)
def test_add_binary(self):
try:
bin1 = open('512b-rsa-example-cert.der', 'rb')
except Exception:
bin1 = open('test/512b-rsa-example-cert.der', 'rb')
try:
bin2 = open('1024b-rsa-example-cert.der', 'rb')
except Exception:
bin2 = open('test/1024b-rsa-example-cert.der', 'rb')
der_certificates = [bin1.read(), bin2.read()]
bin1.close()
bin2.close()
self.delete_at_teardown.append(add_user(self.connection, testcase_id, 'add-operation-4', attributes={'userCertificate;binary': der_certificates}))
self.assertEqual('success', self.delete_at_teardown[0][1]['description'])
|
