GackAwareIterable

GackAwareIterable.cls

public class GackAwareIterable implements Iterable<Object> {
	private final Iterable<Object> iterable;
	
	public GackAwareIterable(Iterable<Object> iterable) {
		this.iterable = iterable;
	}
	
	public Iterator<Object> iterator() {
		Iterator<Object> result;
		
		if(iterable instanceof Database.QueryLocator) {
			result = (Iterator<Object>) ((Database.QueryLocator) iterable).iterator();
		}
		else {
			result = iterable.iterator();
		}
		
		return result;
	}
}

GackAwareIterable_Test.cls

@IsTest
private class GackAwareIterable_Test {
	
	@IsTest
	private static void happyPath() {
		// Setup
		insert new Account(Name = 'test');
		
		Database.QueryLocator queryLocator = Database.getQueryLocator([SELECT Name FROM Account]);
		Iterable<SObject> iterable = queryLocator;
		
		// Execute
		Iterator<SObject> iterator = (Iterator<SObject>) new GackAwareIterable(iterable).iterator();
		
		// Verify
		System.assertEquals([SELECT Name FROM Account], iterator.next());
	}
}