def test_set_or_create(self):
"""Test the internal set or create method."""
key = '{"_id": 4096, "key": "bigger"}'
obj = Keys()
# pylint: disable=protected-access
obj._set_or_create(key)
keys = '[{"_id": 4097, "key": "blah"}]'
obj._set_or_create(keys)
obj._set_or_create(key)
hit_exception = False
try:
obj._set_or_create('{ bad json }')
except ValueError:
hit_exception = True
self.assertTrue(hit_exception)
obj._meta.database.close()
