diff options
Diffstat (limited to 'prog/encoding_test.go')
| -rw-r--r-- | prog/encoding_test.go | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/prog/encoding_test.go b/prog/encoding_test.go index 2464f35da..a38e9deab 100644 --- a/prog/encoding_test.go +++ b/prog/encoding_test.go @@ -114,9 +114,10 @@ func TestCallSet(t *testing.T) { func TestCallSetRandom(t *testing.T) { target, rs, iters := initTest(t) + ct := target.DefaultChoiceTable() for i := 0; i < iters; i++ { const ncalls = 10 - p := target.Generate(rs, ncalls, nil) + p := target.Generate(rs, ncalls, ct) calls0 := make(map[string]struct{}) for _, c := range p.Calls { calls0[c.Meta.Name] = struct{}{} @@ -328,10 +329,11 @@ func TestSerializeDeserialize(t *testing.T) { func TestSerializeDeserializeRandom(t *testing.T) { testEachTargetRandom(t, func(t *testing.T, target *Target, rs rand.Source, iters int) { + ct := target.DefaultChoiceTable() data0 := make([]byte, ExecBufferSize) data1 := make([]byte, ExecBufferSize) for i := 0; i < iters; i++ { - p0 := target.Generate(rs, 10, nil) + p0 := target.Generate(rs, 10, ct) if ok, _, _ := testSerializeDeserialize(t, p0, data0, data1); ok { continue } |