with keyword doesn't yield correct result when applied to rules

[tsandall]

Expected Behavior

The with keyword should be able to mock virtual documents.

Actual Behavior

Queries that iterate over virtual documents are seeing the underlying values generated by rules instead of the mocked value.

Example:

package play

## Reference (working)
q1 = {"a": 1}
p1 = x { q1.a = x }
r1 = q1
test1 {
   p1 = x with q1 as {"a": 2} # working
   r1 = y with q1 as {"a": 2} # working
}


## Iteration (broken)
q[1]
q[2]
q[3]

p[x] { q[x] }
r = q

test {
	p = x with q as {4,5,6}  # broken
    r = y with q as {4,5,6}  # working
}

Tested on