def test_verify_block_returns_inputted_data_on_unverifiable_block(
self, mock_get_keys, mock_registration):
mock_block = MagicMock(dc_id=123, block_id=124)
clouds, regions, ddss, l2_count = level_3_actions.verify_block(
mock_block, set(), set(), 0, 0)
mock_get_keys.assert_called_once_with(123)
mock_registration.assert_not_called()
self.assertEqual(clouds, set())
self.assertEqual(regions, set())
self.assertEqual(ddss, 0)
self.assertEqual(l2_count, 0)
def test_verify_block_returns_what_was_passed_in_on_invalid_block(
self, mock_get_keys, mock_registration):
mock_block = MagicMock(dc_id=123, block_id=123)
clouds, regions, ddss, l2_count = level_3_actions.verify_block(
mock_block, set(), set(), 0, 0)
mock_get_keys.assert_called_once_with(123)
mock_registration.assert_not_called()
self.assertEqual(clouds, set())
self.assertEqual(regions, set())
self.assertEqual(ddss, 0)
self.assertEqual(l2_count, 0)
def test_verify_block_returns_data_on_valid_block(self, mock_registration,
mock_get_keys):
mock_block = MagicMock(dc_id=123, block_id=124, current_ddss="123432")
clouds, regions, ddss, l2_count = level_3_actions.verify_block(
mock_block, set(), set(), 0, 0)
mock_get_keys.assert_called_once_with(123)
mock_registration.assert_called_once_with(123)
self.assertEqual(clouds, {"aws"})
self.assertEqual(regions, {"us-west-2"})
self.assertEqual(ddss, 123432)
self.assertEqual(l2_count, 1)