def test_get_batch_change(mocked_responses, vinyldns_client):
arc = AddRecordChange(forward_zone.id, forward_zone.name,
'foo', 'foo.bar.com', RecordType.A, 200,
AData('1.2.3.4'), 'Complete', 'id1', [],
'system-message', 'rchangeid1', 'rsid1')
drc = DeleteRecordSetChange(forward_zone.id, forward_zone.name, 'baz',
'baz.bar.com', RecordType.A, 'Complete', 'id2',
[], 'system-message', 'rchangeid2', 'rsid2')
drc_with_data = DeleteRecordSetChange(forward_zone.id, forward_zone.name,
'biz', 'biz.bar.com', RecordType.A,
'Complete', 'id3', [],
'system-message', 'rchangeid3',
'rsid3', AData("5.6.7.8"))
bc = BatchChange('user-id',
'user-name',
datetime.utcnow(), [arc, drc, drc_with_data],
'bcid',
'Complete',
'AutoApproved',
comments='batch change test',
owner_group_id='owner-group-id')
mocked_responses.add(responses.GET,
'http://test.com/zones/batchrecordchanges/bcid',
body=to_json_string(bc),
status=200)
r = vinyldns_client.get_batch_change('bcid')
check_batch_changes_are_same(r, bc)
def test_create_batch_change(mocked_responses, vinyldns_client):
ar = AddRecord('foo.bar.com', RecordType.A, 100, AData('1.2.3.4'))
drs = DeleteRecordSet('baz.bar.com', RecordType.A)
arc = AddRecordChange(forward_zone.id, forward_zone.name, 'foo', 'foo.bar.com', RecordType.A, 200,
AData('1.2.3.4'), 'Complete', 'id1', 'system-message', 'rchangeid1', 'rsid1')
drc = DeleteRecordSetChange(forward_zone.id, forward_zone.name, 'baz', 'baz.bar.com', RecordType.A, 'Complete',
'id2', 'system-message', 'rchangeid2', 'rsid2')
bc = BatchChange('user-id', 'user-name', 'batch change test', datetime.utcnow(), [arc, drc],
'bcid', 'owner-group-id')
mocked_responses.add(
responses.POST, 'http://test.com/zones/batchrecordchanges',
body=to_json_string(bc), status=200
)
r = vinyldns_client.create_batch_change(
BatchChangeRequest(
changes=[ar, drs],
comments='batch change test',
owner_group_id='owner-group-id'
))
assert r.user_id == bc.user_id
assert r.user_name == bc.user_name
assert r.comments == bc.comments
assert r.created_timestamp == bc.created_timestamp
assert r.id == bc.id
for l, r in zip(r.changes, bc.changes):
check_single_changes_are_same(l, r)
def test_create_batch_change(mocked_responses, vinyldns_client):
ar = AddRecord('foo.baa.com', RecordType.A, 100, AData('1.2.3.4'))
drs = DeleteRecordSet('baz.bar.com', RecordType.A)
drs_with_data = DeleteRecordSet('baz-with-data.bar.com', RecordType.A,
AData('5.6.7.8'))
arc = AddRecordChange(forward_zone.id, forward_zone.name,
'foo', 'foo.bar.com', RecordType.A, 200,
AData('1.2.3.4'), 'Complete', 'id1', [],
'system-message', 'rchangeid1', 'rsid1')
drc = DeleteRecordSetChange(forward_zone.id, forward_zone.name, 'baz',
'baz.bar.com', RecordType.A, 'Complete', 'id2',
[], 'system-message', 'rchangeid2', 'rsid2')
drc_with_data = DeleteRecordSetChange(forward_zone.id, forward_zone.name,
'baz-with-data',
'baz-with-data.bar.com',
RecordType.A, 'Complete', 'id2', [],
'system-message', 'rchangeid3',
'rsid3', AData('5.6.7.8'))
# Python 2/3 compatibility
try:
tomorrow = datetime.now().astimezone() + timedelta(1)
except TypeError:
tomorrow = datetime.now(tzlocal()).astimezone(tzlocal()) + timedelta(1)
bc = BatchChange('user-id',
'user-name',
datetime.utcnow(), [arc, drc, drc_with_data],
'bcid',
'Scheduled',
'PendingReview',
comments='batch change test',
owner_group_id='owner-group-id',
scheduled_time=tomorrow)
mocked_responses.add(responses.POST,
'http://test.com/zones/batchrecordchanges',
body=to_json_string(bc),
status=200)
r = vinyldns_client.create_batch_change(
BatchChangeRequest(changes=[ar, drs, drs_with_data],
comments='batch change test',
owner_group_id='owner-group-id',
scheduled_time=tomorrow))
check_batch_changes_are_same(r, bc)
def test_create_batch_change(vinyldns_test_context):
changes = [
AddRecord('change-test.vinyldns.', RecordType.A, 200,
AData('192.0.2.111')),
AddRecord('192.0.2.111', RecordType.PTR, 200,
PTRData('change-test.vinyldns.'))
]
r = vinyldns_test_context.client.create_batch_change(
BatchChangeRequest(changes, 'comments',
vinyldns_test_context.group.id))
assert r.user_id == 'ok'
assert r.user_name == 'ok'
assert r.comments == 'comments'
assert r.approval_status == 'AutoApproved'
change1 = r.changes[0]
assert change1.input_name == 'change-test.vinyldns.'
assert change1.ttl == 200
assert change1.record.address == '192.0.2.111'
assert change1.type == RecordType.A
change2 = r.changes[1]
assert change2.input_name == '192.0.2.111'
assert change2.ttl == 200
assert change2.record.ptrdname == 'change-test.vinyldns.'
assert change2.type == RecordType.PTR
def test_record_sets(vinyldns_test_context):
rs = RecordSet(vinyldns_test_context.zone.id,
'a-test',
RecordType.A,
200,
records=[AData('1.2.3.4')])
change = vinyldns_test_context.client.create_record_set(rs)
assert change.record_set.zone_id == rs.zone_id
assert change.record_set.type == rs.type
assert change.record_set.ttl == rs.ttl
assert change.record_set.name == rs.name
assert all([
l.__dict__ == r.__dict__
for l, r in zip(change.record_set.records, rs.records)
])
rs = change.record_set
wait_until_record_set_exists(vinyldns_test_context.client, rs.zone_id,
rs.id)
r = vinyldns_test_context.client.get_record_set(rs.zone_id, rs.id)
assert r.id == rs.id
assert r.name == rs.name
assert r.ttl == rs.ttl
assert all(
[l.__dict__ == r.__dict__ for l, r in zip(rs.records, r.records)])
def test_cancel_batch_change(mocked_responses, vinyldns_client):
error_message = (
"Zone Discovery Failed: zone for 'foo.bar.com' does not exist in"
" VinylDNS. If zone exists, then it must be connected to in VinylDNS.")
error = ValidationError('ZoneDiscoveryError', error_message)
arc = AddRecordChange(forward_zone.id, forward_zone.name,
'cancel', 'cancel.bar.com', RecordType.A, 200,
AData('1.2.3.4'), 'PendingReview', 'id1', [error],
'system-message', 'cchangeid1', 'csid1')
drc = DeleteRecordSetChange(forward_zone.id, forward_zone.name, 'cancel2',
'cancel2.bar.com', RecordType.A, 'Complete',
'id2', [], 'system-message', 'cchangeid2',
'csid2')
bc = BatchChange('user-id',
'user-name',
datetime.utcnow(), [arc, drc],
'bcid',
'Cancelled',
'Cancelled',
owner_group_id='owner-group-id')
mocked_responses.add(
responses.POST,
'http://test.com/zones/batchrecordchanges/bcid/cancel',
body=to_json_string(bc),
status=200)
c = vinyldns_client.cancel_batch_change('bcid')
check_batch_changes_are_same(c, bc)
def test_approve_batch_change(mocked_responses, vinyldns_client):
arc = AddRecordChange(forward_zone.id, forward_zone.name,
'foo', 'foo.bar.com', RecordType.A, 200,
AData('1.2.3.4'), 'PendingReview', 'id1', [],
'system-message', 'rchangeid1', 'rsid1')
drc = DeleteRecordSetChange(forward_zone.id, forward_zone.name, 'baz',
'baz.bar.com', RecordType.A, 'PendingReview',
'id2', [], 'system-message', 'rchangeid2',
'rsid2')
bc = BatchChange('user-id',
'user-name',
datetime.utcnow(), [arc, drc],
'bcid',
'Complete',
'ManuallyApproved',
comments='batch change test',
owner_group_id='owner-group-id',
reviewer_id='admin-id',
reviewer_user_name='admin',
review_comment='looks good',
review_timestamp=datetime.utcnow())
mocked_responses.add(
responses.POST,
'http://test.com/zones/batchrecordchanges/bcid/approve',
body=to_json_string(bc),
status=200)
r = vinyldns_client.approve_batch_change('bcid', 'looks good')
check_batch_changes_are_same(r, bc)
def test_batch_change_review_process_cancel(vinyldns_test_context):
changes = [
AddRecord('test-approve-success.not.loaded.', RecordType.A, 200,
AData("4.3.2.1"))
]
bc = vinyldns_test_context.client.create_batch_change(
BatchChangeRequest(changes, 'comments',
vinyldns_test_context.group.id))
assert bc.status == 'PendingReview'
assert bc.approval_status == 'PendingReview'
assert len(bc.changes[0].validation_errors) == 1
cancelled_bc = vinyldns_test_context.client.cancel_batch_change(bc.id)
assert cancelled_bc.status == 'Cancelled'
assert cancelled_bc.approval_status == 'Cancelled'
def test_reject_batch_change(mocked_responses, vinyldns_client):
error_message = "Zone Discovery Failed: zone for \"foo.bar.com\" does not exist in VinylDNS. \
If zone exists, then it must be connected to in VinylDNS."
error = ValidationError('ZoneDiscoveryError', error_message)
arc = AddRecordChange(forward_zone.id, forward_zone.name,
'reject', 'reject.bar.com', RecordType.A, 200,
AData('1.2.3.4'), 'PendingReview', 'id1', [error],
'system-message', 'rchangeid1', 'rsid1')
drc = DeleteRecordSetChange(forward_zone.id, forward_zone.name, 'reject2',
'reject2.bar.com', RecordType.A, 'Complete',
'id2', [], 'system-message', 'rchangeid2',
'rsid2')
bc = BatchChange('user-id',
'user-name',
datetime.utcnow(), [arc, drc],
'bcid',
'Rejected',
'Rejected',
comments='batch change test',
owner_group_id='owner-group-id',
reviewer_id='admin-id',
reviewer_user_name='admin',
review_comment='not good',
review_timestamp=datetime.utcnow())
mocked_responses.add(
responses.POST,
'http://test.com/zones/batchrecordchanges/bcid/reject',
body=to_json_string(bc),
status=200)
r = vinyldns_client.reject_batch_change('bcid', 'not good')
check_batch_changes_are_same(r, bc)
def test_batch_change_review_process_approve(vinyldns_test_context):
approver = vinyldns_test_context.support_client
changes = [
AddRecord('test-approve-success.not.loaded.', RecordType.A, 200,
AData("4.3.2.1"))
]
bc = vinyldns_test_context.client.create_batch_change(
BatchChangeRequest(changes, 'comments',
vinyldns_test_context.group.id))
assert bc.status == 'PendingReview'
assert bc.approval_status == 'PendingReview'
assert len(bc.changes[0].validation_errors) == 1
# need to create the zone so the change can succeed
zone = {
'name': 'not.loaded.',
'email': '*****@*****.**',
'adminGroupId': vinyldns_test_context.group.id,
'backendId': 'func-test-backend',
'shared': True
}
zone_create = approver.connect_zone(zone)
to_disconnect = zone_create.zone
wait_until_zone_exists(approver, to_disconnect.id)
vinyldns_test_context.support_client.approve_batch_change(
bc.id, 'all good!')
completed_bc = vinyldns_test_context.client.get_batch_change(bc.id)
assert completed_bc.approval_status == 'ManuallyApproved'
assert completed_bc.reviewer_id == 'support-user-id'
assert completed_bc.reviewer_user_name == 'support-user'
assert completed_bc.review_comment == 'all good!'
def test_batch_change_review_process_reject(vinyldns_test_context):
changes = [
AddRecord('test-approve-success.not.loaded.', RecordType.A, 200,
AData("4.3.2.1"))
]
bc = vinyldns_test_context.client.create_batch_change(
BatchChangeRequest(changes, 'comments',
vinyldns_test_context.group.id))
assert bc.status == 'PendingReview'
assert bc.approval_status == 'PendingReview'
assert len(bc.changes[0].validation_errors) == 1
vinyldns_test_context.support_client.reject_batch_change(
bc.id, 'cannot create the zone')
completed_bc = vinyldns_test_context.client.get_batch_change(bc.id)
assert completed_bc.status == 'Rejected'
assert completed_bc.approval_status == 'ManuallyRejected'
assert completed_bc.reviewer_id == 'support-user-id'
assert completed_bc.reviewer_user_name == 'support-user'
assert completed_bc.review_comment == 'cannot create the zone'
sample_zone_change = ZoneChange(zone=forward_zone,
user_id='some-user',
change_type='Create',
status='Pending',
created=datetime.utcnow(),
system_message=None,
id='zone-change-id')
record_sets = {
RecordType.A:
RecordSet(forward_zone.id,
'a-test',
RecordType.A,
200,
records=[AData('1.2.3.4')],
owner_group_id='owner-group-id'),
RecordType.AAAA:
RecordSet(forward_zone.id,
'aaaa-test',
RecordType.AAAA,
200,
records=[AAAAData('1:2:3:4:5:6:7:8')],
owner_group_id='owner-group-id'),
RecordType.CNAME:
RecordSet(forward_zone.id,
'cname-test',
RecordType.CNAME,
200,
records=[CNAMEData('cname')],
owner_group_id='owner-group-id'),
