def setUp(self):
"""
Set up some items we can reuse.
"""
with mock.patch('flagon.backends.db_mongo.pymongo.MongoClient') as _mc:
self._mc = _mc
self._mc()['flagon']['features'].find().count.return_value = 1
self.backend = MongoDBBackend()
def setUp(self):
"""
Set up some items we can reuse.
"""
with mock.patch('flagon.backends.db_mongo.pymongo.MongoClient') as _mc:
self._mc = _mc
self._mc()['flagon']['features'].find().count.return_value = 1
self.backend = MongoDBBackend()
class TestMongoDBBackend(TestCase):
"""
Test the mongo database backend class.
"""
def setUp(self):
"""
Set up some items we can reuse.
"""
with mock.patch('flagon.backends.db_mongo.pymongo.MongoClient') as _mc:
self._mc = _mc
self._mc()['flagon']['features'].find().count.return_value = 1
self.backend = MongoDBBackend()
def test_exists(self):
"""
Verify Backend.exists raises.
"""
assert self.backend.exists('input') is True
self._mc()['flagon']['features'].find().count.return_value = 0
assert self.backend.exists('notthere') is False
def test_is_active(self):
"""
Verify Backend.is_active raises.
"""
self._mc()['flagon']['features'].find_one.return_value = {
'_id': '',
'name': 'input',
'active': True
}
assert self.backend.is_active('input') is True
self._mc()['flagon']['features'].find_one.return_value = {
'_id': '',
'name': 'isoff',
'active': False
}
assert self.backend.is_active('isoff') is False
def test_turn_on(self):
"""
Verify Backend.turn_on raises.
"""
self.backend.turn_on('input')
assert self._mc()['flagon']['features'].update.call_count == 1
def test_turn_off(self):
"""
Verify Backend.turn_off raises.
"""
self.backend.turn_off('input')
assert self._mc()['flagon']['features'].update.call_count == 1
def test_toggle(self):
"""
Verify Backend.toggle raises.
"""
self.backend.toggle('input')
assert self._mc()['flagon']['features'].update.call_count == 1
def test_is_off(self):
"""
Verify Backend.is_off raises.
"""
self._mc()['flagon']['features'].find_one.return_value = {
'_id': '',
'name': 'input',
'active': False
}
assert self.backend.is_off('input') is True
class TestMongoDBBackend(TestCase):
"""
Test the mongo database backend class.
"""
def setUp(self):
"""
Set up some items we can reuse.
"""
with mock.patch('flagon.backends.db_mongo.pymongo.MongoClient') as _mc:
self._mc = _mc
self._mc()['flagon']['features'].find().count.return_value = 1
self.backend = MongoDBBackend()
def test_exists(self):
"""
Verify Backend.exists raises.
"""
assert self.backend.exists('input') is True
self._mc()['flagon']['features'].find().count.return_value = 0
assert self.backend.exists('notthere') is False
def test_is_active(self):
"""
Verify Backend.is_active raises.
"""
self._mc()['flagon']['features'].find_one.return_value = {
'_id': '',
'name': 'input',
'active': True}
assert self.backend.is_active('input') is True
self._mc()['flagon']['features'].find_one.return_value = {
'_id': '',
'name': 'isoff',
'active': False}
assert self.backend.is_active('isoff') is False
def test_turn_on(self):
"""
Verify Backend.turn_on raises.
"""
self.backend.turn_on('input')
assert self._mc()['flagon']['features'].update.call_count == 1
def test_turn_off(self):
"""
Verify Backend.turn_off raises.
"""
self.backend.turn_off('input')
assert self._mc()['flagon']['features'].update.call_count == 1
def test_toggle(self):
"""
Verify Backend.toggle raises.
"""
self.backend.toggle('input')
assert self._mc()['flagon']['features'].update.call_count == 1
def test_is_off(self):
"""
Verify Backend.is_off raises.
"""
self._mc()['flagon']['features'].find_one.return_value = {
'_id': '',
'name': 'input',
'active': False}
assert self.backend.is_off('input') is True
