prog: fix isDefaultArg

Test that isDefaultArg returns true for result of DefaultArg.
Fix few bugs uncovered by this test.

1 files changed, 8 insertions, 4 deletions

diff --git a/prog/prog_test.go b/prog/prog_test.go
index 1dc310456..2dc4f6bbd 100644
--- a/prog/prog_test.go
+++ b/prog/prog_test.go
@@ -23,10 +23,14 @@ func TestGeneration(t *testing.T) {
 
 func TestDefault(t *testing.T) {
 	target, _, _ := initTest(t)
-	for _, meta := range target.SyscallMap {
-		for _, t := range meta.Args {
-			target.defaultArg(t)
-		}
+	for _, meta := range target.Syscalls {
+		ForeachType(meta, func(typ Type) {
+			arg := target.defaultArg(typ)
+			if !target.isDefaultArg(arg) {
+				t.Errorf("default arg is not default: %s\ntype: %#v\narg: %#v",
+					typ, typ, arg)
+			}
+		})
 	}
 }