def test_mutation(self):
bithandler = BitHandler(0, ('FLAG_0', 'FLAG_1', 'FLAG_2', 'FLAG_3'))
self.assertEquals(bool(bithandler.FLAG_0), False)
self.assertEquals(bool(bithandler.FLAG_1), False)
self.assertEquals(bool(bithandler.FLAG_2), False)
self.assertEquals(bool(bithandler.FLAG_3), False)
bithandler = BitHandler(bithandler | 1, bithandler._keys)
self.assertEquals(bool(bithandler.FLAG_0), True)
self.assertEquals(bool(bithandler.FLAG_1), False)
self.assertEquals(bool(bithandler.FLAG_2), False)
self.assertEquals(bool(bithandler.FLAG_3), False)
bithandler ^= 3
self.assertEquals(int(bithandler), 2)
self.assertEquals(bool(bithandler & 1), False)
bithandler.FLAG_0 = False
self.assertEquals(bithandler.FLAG_0, False)
bithandler.FLAG_1 = True
self.assertEquals(bithandler.FLAG_0, False)
self.assertEquals(bithandler.FLAG_1, True)
bithandler.FLAG_2 = False
self.assertEquals(bithandler.FLAG_0, False)
self.assertEquals(bithandler.FLAG_1, True)
self.assertEquals(bithandler.FLAG_2, False)
def to_python(self, value):
if isinstance(value, Bit):
value = value.mask
if not isinstance(value, BitHandler):
value = BitHandler(value, self.flags, self.labels)
else:
# Ensure flags are consistent for unpickling
value._keys = self.flags
return value
def test_mutation(self):
bithandler = BitHandler(0, ('FLAG_0', 'FLAG_1', 'FLAG_2', 'FLAG_3'))
self.assertEquals(bool(bithandler.FLAG_0), False)
self.assertEquals(bool(bithandler.FLAG_1), False)
self.assertEquals(bool(bithandler.FLAG_2), False)
self.assertEquals(bool(bithandler.FLAG_3), False)
bithandler = BitHandler(bithandler | 1, bithandler._keys)
self.assertEquals(bool(bithandler.FLAG_0), True)
self.assertEquals(bool(bithandler.FLAG_1), False)
self.assertEquals(bool(bithandler.FLAG_2), False)
self.assertEquals(bool(bithandler.FLAG_3), False)
bithandler ^= 3
self.assertEquals(int(bithandler), 2)
self.assertEquals(bool(bithandler & 1), False)
bithandler.FLAG_0 = False
self.assertEquals(bithandler.FLAG_0, False)
bithandler.FLAG_1 = True
self.assertEquals(bithandler.FLAG_0, False)
self.assertEquals(bithandler.FLAG_1, True)
bithandler.FLAG_2 = False
self.assertEquals(bithandler.FLAG_0, False)
self.assertEquals(bithandler.FLAG_1, True)
self.assertEquals(bithandler.FLAG_2, False)
def to_python(self, value):
if isinstance(value, Bit):
value = value.mask
if not isinstance(value, BitHandler):
# Regression for #1425: fix bad data that was created resulting
# in negative values for flags. Compute the value that would
# have been visible ot the application to preserve compatibility.
if isinstance(value, six.integer_types) and value < 0:
new_value = 0
for bit_number, _ in enumerate(self.flags):
new_value |= (value & (2 ** bit_number))
value = new_value
value = BitHandler(value, self.flags, self.labels)
else:
# Ensure flags are consistent for unpickling
value._keys = self.flags
return value
def to_python(self, value):
if isinstance(value, Bit):
value = value.mask
if not isinstance(value, BitHandler):
# Regression for #1425: fix bad data that was created resulting
# in negative values for flags. Compute the value that would
# have been visible ot the application to preserve compatibility.
if isinstance(value, six.integer_types) and value < 0:
new_value = 0
for bit_number, _ in enumerate(self.flags):
new_value |= (value & (2**bit_number))
value = new_value
value = BitHandler(value, self.flags, self.labels)
else:
# Ensure flags are consistent for unpickling
value._keys = self.flags
return value
def to_python(self, value):
# Assume an iterable which contains an item per flag that's enabled
result = BitHandler(0, [k for k, v in self.choices])
for k in value:
try:
setattr(result, str(k), True)
except AttributeError:
raise ValidationError('Unknown choice: %r' % (k, ))
return result
def clean(self, value):
if not value:
return 0
# Assume an iterable which contains an item per flag that's enabled
result = BitHandler(0, [k for k, v in self.choices])
for k in value:
try:
setattr(result, six.text_type(k), True)
except AttributeError:
raise ValidationError('Unknown choice: %r' % (k, ))
return int(result)
def test_nonzero_default(self):
bithandler = BitHandler(1, ('FLAG_0', 'FLAG_1', 'FLAG_2', 'FLAG_3'))
self.assertEquals(bool(bithandler.FLAG_0), True)
self.assertEquals(bool(bithandler.FLAG_1), False)
self.assertEquals(bool(bithandler.FLAG_2), False)
self.assertEquals(bool(bithandler.FLAG_3), False)
bithandler = BitHandler(2, ('FLAG_0', 'FLAG_1', 'FLAG_2', 'FLAG_3'))
self.assertEquals(bool(bithandler.FLAG_0), False)
self.assertEquals(bool(bithandler.FLAG_1), True)
self.assertEquals(bool(bithandler.FLAG_2), False)
self.assertEquals(bool(bithandler.FLAG_3), False)
bithandler = BitHandler(3, ('FLAG_0', 'FLAG_1', 'FLAG_2', 'FLAG_3'))
self.assertEquals(bool(bithandler.FLAG_0), True)
self.assertEquals(bool(bithandler.FLAG_1), True)
self.assertEquals(bool(bithandler.FLAG_2), False)
self.assertEquals(bool(bithandler.FLAG_3), False)
bithandler = BitHandler(4, ('FLAG_0', 'FLAG_1', 'FLAG_2', 'FLAG_3'))
self.assertEquals(bool(bithandler.FLAG_0), False)
self.assertEquals(bool(bithandler.FLAG_1), False)
self.assertEquals(bool(bithandler.FLAG_2), True)
self.assertEquals(bool(bithandler.FLAG_3), False)
def test_defaults(self):
bithandler = BitHandler(0, ('FLAG_0', 'FLAG_1', 'FLAG_2', 'FLAG_3'))
# Default value of 0.
self.assertEquals(int(bithandler), 0)
# Test bit numbers.
self.assertEquals(int(bithandler.FLAG_0.number), 0)
self.assertEquals(int(bithandler.FLAG_1.number), 1)
self.assertEquals(int(bithandler.FLAG_2.number), 2)
self.assertEquals(int(bithandler.FLAG_3.number), 3)
# Negative test non-existant key.
self.assertRaises(AttributeError, lambda: bithandler.FLAG_4)
# Test bool().
self.assertEquals(bool(bithandler.FLAG_0), False)
self.assertEquals(bool(bithandler.FLAG_1), False)
self.assertEquals(bool(bithandler.FLAG_2), False)
self.assertEquals(bool(bithandler.FLAG_3), False)
def __xor__(self, value):
return BitHandler(self._value ^ int(value), self._keys)
def __rshift__(self, value):
return BitHandler(self._value >> int(value), self._keys)
def __lshift__(self, value):
return BitHandler(self._value << int(value), self._keys)
def __sub__(self, value):
return BitHandler(self._value - int(value), self._keys)
def __add__(self, value):
return BitHandler(self._value + int(value), self._keys)
