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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
# Copyright 2009-2012 10gen, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Tests for the Binary wrapper."""
import unittest
import sys
try:
import uuid
should_test_uuid = True
except ImportError:
should_test_uuid = False
sys.path[0:0] = [""]
from bson.binary import Binary, UUIDLegacy
from bson.py3compat import b, binary_type
from nose.plugins.skip import SkipTest
from test.test_connection import get_connection
class TestBinary(unittest.TestCase):
def setUp(self):
pass
def test_binary(self):
a_string = "hello world"
a_binary = Binary(b("hello world"))
self.assertTrue(a_binary.startswith(b("hello")))
self.assertTrue(a_binary.endswith(b("world")))
self.assertTrue(isinstance(a_binary, Binary))
self.assertFalse(isinstance(a_string, Binary))
def test_exceptions(self):
self.assertRaises(TypeError, Binary, None)
self.assertRaises(TypeError, Binary, u"hello")
self.assertRaises(TypeError, Binary, 5)
self.assertRaises(TypeError, Binary, 10.2)
self.assertRaises(TypeError, Binary, b("hello"), None)
self.assertRaises(TypeError, Binary, b("hello"), "100")
self.assertRaises(ValueError, Binary, b("hello"), -1)
self.assertRaises(ValueError, Binary, b("hello"), 256)
self.assertTrue(Binary(b("hello"), 0))
self.assertTrue(Binary(b("hello"), 255))
def test_subtype(self):
one = Binary(b("hello"))
self.assertEqual(one.subtype, 0)
two = Binary(b("hello"), 2)
self.assertEqual(two.subtype, 2)
three = Binary(b("hello"), 100)
self.assertEqual(three.subtype, 100)
def test_equality(self):
two = Binary(b("hello"))
three = Binary(b("hello"), 100)
self.assertNotEqual(two, three)
self.assertEqual(three, Binary(b("hello"), 100))
self.assertEqual(two, Binary(b("hello")))
self.assertNotEqual(two, Binary(b("hello ")))
self.assertNotEqual(b("hello"), Binary(b("hello")))
def test_repr(self):
one = Binary(b("hello world"))
self.assertEqual(repr(one),
"Binary(%s, 0)" % (repr(b("hello world")),))
two = Binary(b("hello world"), 2)
self.assertEqual(repr(two),
"Binary(%s, 2)" % (repr(b("hello world")),))
three = Binary(b("\x08\xFF"))
self.assertEqual(repr(three),
"Binary(%s, 0)" % (repr(b("\x08\xFF")),))
four = Binary(b("\x08\xFF"), 2)
self.assertEqual(repr(four),
"Binary(%s, 2)" % (repr(b("\x08\xFF")),))
five = Binary(b("test"), 100)
self.assertEqual(repr(five),
"Binary(%s, 100)" % (repr(b("test")),))
def test_uuid_queries(self):
if not should_test_uuid:
raise SkipTest()
c = get_connection()
coll = c.pymongo_test.test
coll.drop()
uu = uuid.uuid4()
# Wrap uu.bytes in binary_type to work
# around http://bugs.python.org/issue7380.
coll.insert({'uuid': Binary(binary_type(uu.bytes), 3)})
self.assertEqual(1, coll.count())
# Test UUIDLegacy queries.
coll.uuid_subtype = 4
self.assertEqual(0, coll.find({'uuid': uu}).count())
cur = coll.find({'uuid': UUIDLegacy(uu)})
self.assertEqual(1, cur.count())
retrieved = cur.next()
self.assertEqual(uu, retrieved['uuid'])
# Test regular UUID queries (using subtype 4).
coll.insert({'uuid': uu})
self.assertEqual(2, coll.count())
cur = coll.find({'uuid': uu})
self.assertEqual(1, cur.count())
retrieved = cur.next()
self.assertEqual(uu, retrieved['uuid'])
# Test both.
cur = coll.find({'uuid': {'$in': [uu, UUIDLegacy(uu)]}})
self.assertEqual(2, cur.count())
coll.drop()
if __name__ == "__main__":
unittest.main()
|
