def test_pk_repr_no_leak(self):
"""Representation of PK shouldn't leak DB data."""
value = 'pk-value-1234'
pk = m.PartitionKey(Session, value)
self.assertNotIn(value, repr(pk))
def setUp(self):
self._pk = m.PartitionKey(User, 'alice')
self._sk = m.SortKey(Project, 'foo')
self._primary = m.PrimaryKey(self._pk, self._sk)
def test_hash_eq(self):
pk_1 = m.PartitionKey(User, 'value')
pk_2 = m.PartitionKey(User, 'value')
self.assertEqual(hash(pk_1), hash(pk_2))
def test_pk_eq_sq(self):
pk = m.PartitionKey(User, 'value')
sk = m.SortKey(User, 'value')
self.assertEqual(pk, sk)
def test_eq(self):
pk_1 = m.PartitionKey(User, 'value')
pk_2 = m.PartitionKey(User, 'value')
self.assertEqual(pk_1, pk_2)
def test_self(self):
pk = m.PartitionKey(User, 'value')
self.assertEqual(pk, pk)
def test_pk_prefix(self):
pk = m.PartitionKey(User, 'val')
self.assertEqual(User.to_prefix(), pk.prefix)