def test_update_channels_uses_environment_variables(self):
with patch('snapcraft.storeapi.common.os.environ',
{'UBUNTU_STORE_API_ROOT_URL': 'http://example.com'}):
update_channels(
self.mock_session, 'package.name', {'stable': 2})
self.mock_post.assert_called_once_with(
'http://example.com/package-channels/package.name/',
data=json.dumps({'stable': 2}),
headers={'Content-Type': 'application/json'})
def test_update_channels(self):
self.set_channels_post_success_response()
data = update_channels(
self.mock_session, 'package.name', {'stable': 2})
expected = {
'success': True,
'errors': [],
'data': self.channels_data,
}
self.assertEqual(data, expected)
def test_update_channels_with_failed_response(self):
error_msg = 'some error'
self.set_channels_post_failed_response(error_msg)
data = update_channels(
self.mock_session, 'package.name', {'stable': 2})
expected = {
'success': True,
'errors': [error_msg],
'data': self.channels_data,
}
self.assertEqual(data, expected)
