def test_get_data(self, mock_api):
mock_api.service_get.return_value = 'foo_env'
update_service_row = tables.UpdateServiceRow(mock.Mock())
update_service_row.table.kwargs = {'environment_id': 'foo_env_id'}
update_service_row.get_data(None, 'foo_service_id')
mock_api.service_get.assert_called_once_with(None, 'foo_env_id',
'foo_service_id')
def test_update_service_row(self):
update_service_row = tables.UpdateServiceRow(None)
self.assertTrue(update_service_row.ajax)