def test_get_conversation_bad_deserialize(get_mock):
# type: (Mock) -> None
get_mock.side_effect = [
Response(True, '[' + conversation_contents + ']', 200),
Response(True, '', 200)
]
request = Fulfillment(id='PR-0000-0000-0000')
conversation = request.get_conversation()
assert get_mock.call_count == 2
get_mock.assert_has_calls([
call(headers={
'Content-Type': 'application/json',
'Authorization': 'ApiKey XXXX:YYYYY'
},
params={'instance_id': request.id},
url='http://localhost:8080/api/public/v1/conversations/'),
call(
headers={
'Content-Type': 'application/json',
'Authorization': 'ApiKey XXXX:YYYYY'
},
url=
'http://localhost:8080/api/public/v1/conversations/CO-750-033-356')
])
assert conversation is None
def test_get_conversation_ok(get_mock):
# type: (Mock) -> None
get_mock.side_effect = [
Response(True, '[' + conversation_contents + ']', 200),
Response(True, conversation_contents, 200)
]
request = Fulfillment(id='PR-0000-0000-0000')
conversation = request.get_conversation()
assert get_mock.call_count == 2
get_mock.assert_has_calls([
call(headers={
'Content-Type': 'application/json',
'Authorization': 'ApiKey XXXX:YYYYY'
},
params={'instance_id': request.id},
timeout=300,
url='http://localhost:8080/api/public/v1/conversations'),
call(headers={
'Content-Type': 'application/json',
'Authorization': 'ApiKey XXXX:YYYYY'
},
timeout=300,
url='http://localhost:8080/api/public/v1/conversations/' +
conversation.id)
])
assert isinstance(conversation, Conversation)
def test_needs_migration():
# type: () -> None
handler = connect_migration.MigrationHandler()
# No migration needed
response_no_migration = _load_str('response.json')
requests_no_migration = Fulfillment.deserialize(response_no_migration)
assert isinstance(requests_no_migration, list)
assert len(requests_no_migration) == 1
assert isinstance(requests_no_migration[0], Fulfillment)
assert not handler._needs_migration(requests_no_migration[0])
# Migration needed
response_migration = _load_str('request.migrate.valid.json')
request = Fulfillment.deserialize(response_migration)
assert isinstance(request, Fulfillment)
assert handler._needs_migration(request)
def test_get_conversation_empty(get_mock):
# type: (Mock) -> None
get_mock.return_value = Response(True, '[]', 200)
request = Fulfillment(id='PR-0000-0000-0000')
conversation = request.get_conversation()
assert get_mock.call_count == 1
get_mock.assert_has_calls([
call(headers={
'Content-Type': 'application/json',
'Authorization': 'ApiKey XXXX:YYYYY'
},
params={'instance_id': request.id},
url='http://localhost:8080/api/public/v1/conversations/')
])
assert conversation is None
def test_no_migration(info_mock):
# type: (Mock) -> None
response = _load_str('response.json')
requests = Fulfillment.deserialize(response)
assert isinstance(requests, list)
assert len(requests) == 1
handler = connect_migration.MigrationHandler()
request = handler.migrate(requests[0])
info_mock.assert_called_once_with('[MIGRATION::PR-5852-1608-0000] '
'Request does not need migration.')
assert request == requests[0]
def test_migration_direct(info_mock, debug_mock):
# type: (Mock, Mock) -> None
response = _load_str('request.migrate.direct.success.json')
request = Fulfillment.deserialize(response)
handler = connect_migration.MigrationHandler()
request_out = handler.migrate(request)
assert request_out != request
assert request_out.id == 'PR-7001-1234-5678'
assert request_out.asset.id == 'AS-146-621-424-3'
assert len(request_out.asset.params) == 6
assert request_out.asset.get_param_by_id(
'email').value == '*****@*****.**'
assert request_out.asset.get_param_by_id(
'num_licensed_users').value == '10'
assert request_out.asset.get_param_by_id('reseller_id').value == ''
assert request_out.asset.get_param_by_id('team_id').value == 'dbtid:AADaQq_'\
'w53nMDQbIPM_X123456PuzpcM2BI'
assert request_out.asset.get_param_by_id(
'team_name').value == 'Migration Team'
assert info_mock.call_count == 2
info_mock.assert_has_calls([
call(
'[MIGRATION::PR-7001-1234-5678] Running migration operations for request '
'PR-7001-1234-5678'),
call(
'[MIGRATION::PR-7001-1234-5678] 5 processed, 4 succeeded '
'(email, num_licensed_users, team_id, team_name), 0 failed, 1 skipped (reseller_id).'
)
])
assert debug_mock.call_count == 2
debug_mock.assert_has_calls([
call(
'[MIGRATION::PR-7001-1234-5678] Migration data `migration_info`: {'
'"email":"*****@*****.**",'
'"team_id":"dbtid:AADaQq_w53nMDQbIPM_X123456PuzpcM2BI",'
'"team_name":"Migration Team",'
'"num_licensed_users":"10"}'),
call(
'[MIGRATION::PR-7001-1234-5678] Migration data `migration_info` parsed correctly'
)
])
def test_migration_transform_manual_fail(info_mock, debug_mock, error_mock):
# type: (Mock, Mock, Mock) -> None
response = _load_str('request.migrate.transformation.json')
request = Fulfillment.deserialize(response)
handler = connect_migration.MigrationHandler({'email': _raise_error})
with pytest.raises(SkipRequest):
handler.migrate(request)
assert info_mock.call_count == 3
info_mock.assert_has_calls([
call(
'[MIGRATION::PR-7001-1234-5678] Running migration operations for request '
'PR-7001-1234-5678'),
call(
'[MIGRATION::PR-7001-1234-5678] Running transformation for parameter email'
),
call(
'[MIGRATION::PR-7001-1234-5678] 5 processed, 0 succeeded, 1 failed (email), '
'4 skipped (num_licensed_users, reseller_id, team_id, team_name).')
])
assert debug_mock.call_count == 2
debug_mock.assert_has_calls([
call(
'[MIGRATION::PR-7001-1234-5678] Migration data `migration_info`: {'
'"teamAdminEmail":"*****@*****.**",'
'"teamId":"dbtid:AADaQq_w53nMDQbIPM_X123456PuzpcM2BI",'
'"teamName":"Migration Team",'
'"licNumber":"10"}'),
call(
'[MIGRATION::PR-7001-1234-5678] Migration data `migration_info` parsed correctly'
)
])
assert error_mock.call_count == 2
error_mock.assert_has_calls([
call('[MIGRATION::PR-7001-1234-5678] Manual fail.'),
call(
'[MIGRATION::PR-7001-1234-5678] Processing of parameters email failed, '
'unable to complete migration.')
])
def test_migration_skip_all(info_mock, debug_mock):
# type: (Mock, Mock) -> None
response = _load_str('request.migrate.valid.json')
request = Fulfillment.deserialize(response)
handler = connect_migration.MigrationHandler()
request_out = handler.migrate(request)
assert info_mock.call_count == 2
info_mock.assert_has_calls([
call(
'[MIGRATION::PR-7001-1234-5678] Running migration operations for request '
'PR-7001-1234-5678'),
call(
'[MIGRATION::PR-7001-1234-5678] 5 processed, 0 succeeded, 0 failed, 5 skipped '
'(email, num_licensed_users, reseller_id, team_id, team_name).')
])
assert debug_mock.call_count == 2
debug_mock.assert_has_calls([
call(
'[MIGRATION::PR-7001-1234-5678] Migration data `migration_info`: {'
'"teamAdminEmail":"*****@*****.**",'
'"teamId":"dbtid:AADaQq_w53nMDQbIPM_X123456PuzpcM2BI",'
'"resellerId":["3ONEYO1234"],'
'"teamName":"Migration Team",'
'"licNumber":"10"}'),
call(
'[MIGRATION::PR-7001-1234-5678] Migration data `migration_info` parsed correctly'
)
])
assert request_out != request
assert request_out.id == 'PR-7001-1234-5678'
assert request_out.asset.id == 'AS-146-621-424-3'
assert len(request_out.asset.params) == 6
for i, _ in enumerate(request_out.asset.params):
assert request.asset.params[i].id == request_out.asset.params[i].id
assert request.asset.params[i].value == request_out.asset.params[
i].value
def test_migration_direct_serialize(info_mock, debug_mock):
# type: (Mock, Mock) -> None
response = _load_str('request.migrate.direct.notserialized.json')
request = Fulfillment.deserialize(response)
handler = connect_migration.MigrationHandler(serialize=True)
request_out = handler.migrate(request)
assert request_out != request
assert request_out.id == 'PR-7001-1234-5678'
assert request_out.asset.id == 'AS-146-621-424-3'
assert len(request_out.asset.params) == 6
assert request_out.asset.get_param_by_id(
'email').value == '*****@*****.**'
team_name = request_out.asset.get_param_by_id('team_name').value
assert isinstance(team_name, six.string_types)
assert team_name == '["Some name"]'
assert info_mock.call_count == 2
info_mock.assert_has_calls([
call(
'[MIGRATION::PR-7001-1234-5678] Running migration operations for request '
'PR-7001-1234-5678'),
call(
'[MIGRATION::PR-7001-1234-5678] 5 processed, 2 succeeded '
'(email, team_name), 0 failed, 3 skipped (num_licensed_users, reseller_id, team_id).'
)
])
assert debug_mock.call_count == 2
debug_mock.assert_has_calls([
call(
'[MIGRATION::PR-7001-1234-5678] Migration data `migration_info`: {'
'"email":"*****@*****.**",'
'"team_name":["Some name"]}'),
call(
'[MIGRATION::PR-7001-1234-5678] Migration data `migration_info` parsed correctly'
)
])
def test_migration_direct_no_serialize(info_mock, debug_mock, error_mock):
# type: (Mock, Mock, Mock) -> None
response = _load_str('request.migrate.direct.notserialized.json')
request = Fulfillment.deserialize(response)
handler = connect_migration.MigrationHandler()
with pytest.raises(SkipRequest):
handler.migrate(request)
assert info_mock.call_count == 2
info_mock.assert_has_calls([
call('[MIGRATION::PR-7001-1234-5678] Running migration operations '
'for request PR-7001-1234-5678'),
call(
'[MIGRATION::PR-7001-1234-5678] 5 processed, 1 succeeded (email), '
'1 failed (team_name), 3 skipped (num_licensed_users, reseller_id, team_id).'
)
])
assert debug_mock.call_count == 2
debug_mock.assert_has_calls([
call(
'[MIGRATION::PR-7001-1234-5678] Migration data `migration_info`: {'
'"email":"*****@*****.**",'
'"team_name":["Some name"]}'),
call(
'[MIGRATION::PR-7001-1234-5678] Migration data `migration_info` parsed correctly'
)
])
assert error_mock.call_count == 2
error_mock.assert_has_calls([
call(
'[MIGRATION::PR-7001-1234-5678] Parameter team_name type must be str, '
'but list was given'),
call(
'[MIGRATION::PR-7001-1234-5678] Processing of parameters team_name failed, '
'unable to complete migration.')
])
def test_migration_wrong_info(info_mock, debug_mock, error_mock):
# type: (Mock, Mock, Mock) -> None
response = _load_str('request.migrate.invalid.json')
request = Fulfillment.deserialize(response)
handler = connect_migration.MigrationHandler()
with pytest.raises(SkipRequest):
handler.migrate(request)
assert info_mock.call_count == 1
info_mock.assert_called_with(
'[MIGRATION::PR-7001-1234-5678] Running migration operations '
'for request PR-7001-1234-5678')
assert debug_mock.call_count == 1
debug_mock.assert_called_with(
'[MIGRATION::PR-7001-1234-5678] Migration data `migration_info`: '
'"teamAdminEmail":"*****@*****.**",'
'"teamId":"dbtid:AADaQq_w53nMDQbIPM_X123456PuzpcM2BI",'
'"resellerId":["3ONEYO1234"],'
'"teamName":"Migration Team",'
'"licNumber":"10"}')
assert error_mock.call_count == 1
def make_object(self, data):
from connect.models import Fulfillment
return Fulfillment(**data)
def test_migration_transform(info_mock, debug_mock):
# type: (Mock, Mock) -> None
response = _load_str('request.migrate.transformation.json')
request = Fulfillment.deserialize(response)
handler = connect_migration.MigrationHandler({
'email':
lambda data, request_id: data['teamAdminEmail'].upper(),
'team_id':
lambda data, request_id: data['teamId'].upper(),
'team_name':
lambda data, request_id: data['teamName'].upper(),
'num_licensed_users':
lambda data, request_id: int(data['licNumber']) * 10
})
request_out = handler.migrate(request)
assert request_out != request
assert request_out.id == 'PR-7001-1234-5678'
assert request_out.asset.id == 'AS-146-621-424-3'
assert len(request_out.asset.params) == 6
assert request_out.asset.get_param_by_id(
'email').value == '*****@*****.**'
assert request_out.asset.get_param_by_id('num_licensed_users').value == 100
assert request_out.asset.get_param_by_id('reseller_id').value == ''
assert request_out.asset.get_param_by_id('team_id').value == 'DBTID:AADAQQ_'\
'W53NMDQBIPM_X123456PUZPCM2BI'
assert request_out.asset.get_param_by_id(
'team_name').value == 'MIGRATION TEAM'
assert info_mock.call_count == 6
info_mock.assert_has_calls([
call(
'[MIGRATION::PR-7001-1234-5678] Running migration operations for request '
'PR-7001-1234-5678'),
call(
'[MIGRATION::PR-7001-1234-5678] Running transformation for parameter email'
),
call(
'[MIGRATION::PR-7001-1234-5678] Running transformation for parameter '
'num_licensed_users'),
call(
'[MIGRATION::PR-7001-1234-5678] Running transformation for parameter team_id'
),
call(
'[MIGRATION::PR-7001-1234-5678] Running transformation for parameter team_name'
),
call(
'[MIGRATION::PR-7001-1234-5678] 5 processed, 4 succeeded '
'(email, num_licensed_users, team_id, team_name), 0 failed, 1 skipped (reseller_id).'
)
])
assert debug_mock.call_count == 2
debug_mock.assert_has_calls([
call(
'[MIGRATION::PR-7001-1234-5678] Migration data `migration_info`: {'
'"teamAdminEmail":"*****@*****.**",'
'"teamId":"dbtid:AADaQq_w53nMDQbIPM_X123456PuzpcM2BI",'
'"teamName":"Migration Team",'
'"licNumber":"10"}'),
call(
'[MIGRATION::PR-7001-1234-5678] Migration data `migration_info` parsed correctly'
)
])