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'])