def test_getNeededDescriptorDigests_desc_have_none(self):
mock_router = mock.Mock()
mock_router.digest = 'digest'
consensus = mock.Mock()
consensus.routers = {'test': mock_router}
ret = mdm._getNeededDescriptorDigests(consensus, None)
self.assertEqual(ret, ['digest'])
def test_getNeededDescriptorDigests_desc_have_none(self):
mock_router = mock.Mock()
mock_router.digest = 'digest'
consensus = mock.Mock()
consensus.routers = {'test': mock_router}
ret = mdm._getNeededDescriptorDigests(consensus, None)
self.assertEqual(ret, ['digest'])
def test_getNeededDescriptorDigests(self):
mock_router = mock.Mock()
mock_router.digest = 'digest'
mock_router2 = mock.Mock()
mock_router2.digest = 'needed digest'
consensus = mock.Mock()
consensus.routers = {'test': mock_router, 'test2': mock_router2}
descriptors = {'test': mock_router}
ret = mdm._getNeededDescriptorDigests(consensus, descriptors)
self.assertEqual(ret, ['needed digest'])
def test_getNeededDescriptorDigests(self):
mock_router = mock.Mock()
mock_router.digest = 'digest'
mock_router2 = mock.Mock()
mock_router2.digest = 'needed digest'
consensus = mock.Mock()
consensus.routers = {'test': mock_router, 'test2': mock_router2}
descriptors = {'test': mock_router}
ret = mdm._getNeededDescriptorDigests(consensus, descriptors)
self.assertEqual(ret, ['needed digest'])
