def test_update_new_deployment(self, get): get.return_value = None annotations = MagicMock() manager.update('datapushers', False, {'router-type': 'traefik'}, annotations, {'root-domain': 'ckan.io'}) get.assert_called_once_with('deployment router-traefik-datapushers', required=False) self.assertEqual(annotations.update_status.call_count, 1)
def test_update_old_deployment(self, get): get.side_effect = [{ 'metadata': { 'generation': 1 } }, { 'metadata': { 'generation': 2 } }] annotations = MagicMock() manager.update('datapushers', False, {'router-type': 'traefik'}, annotations, {'root-domain': 'ckan.io'}) self.assertEqual(get.call_count, 2) self.assertEqual(annotations.update_status.call_count, 1)
def test_update_old_deployment_wrong_generation(self, get): get.side_effect = [{ 'metadata': { 'generation': 1 } }, { 'metadata': { 'generation': 4 } }] annotations = MagicMock() with self.assertRaisesRegex(Exception, 'Invalid generation: 4 \(expected: 2\)'): manager.update('datapushers', False, {'router-type': 'traefik'}, annotations, {'root-domain': 'ckan.io'}) self.assertEqual(get.call_count, 2) self.assertEqual(annotations.update_status.call_count, 1)