watchtower/wtwire/wtwire_test: test failcodes and Data field

watchtower/wtwire/wtwire_test.go

@@ -78,14 +78,16 @@ func TestWatchtowerWireProtocol(t *testing.T) {
 		},
 		wtwire.MsgSessionInitReply: func(v []reflect.Value, r *rand.Rand) {
 			req := &wtwire.SessionInitReply{
-				Code: uint16(r.Int31()),
+				Code: wtwire.SessionInitCode(r.Int31()),
 			}
 
-			rewardAddrLen := rand.Int31n(96)
+			dataLen := rand.Int31n(
+				wtwire.MaxSessionInitReplyDataLength,
+			)
 
-			req.RewardPkScript = make([]byte, rewardAddrLen)
-			if _, err := r.Read(req.RewardPkScript); err != nil {
-				t.Fatalf("unable to generate reward pk script: %v",
+			req.Data = make([]byte, dataLen)
+			if _, err := r.Read(req.Data); err != nil {
+				t.Fatalf("unable to generate data: %v",
 					err)
 				return
 			}
@@ -95,6 +97,7 @@ func TestWatchtowerWireProtocol(t *testing.T) {
 		wtwire.MsgStateUpdate: func(v []reflect.Value, r *rand.Rand) {
 			req := &wtwire.StateUpdate{
 				SeqNum:      uint16(r.Int31()),
+				IsComplete:  uint8(r.Int31()),
 				LastApplied: uint16(r.Int31()),
 			}
 
@@ -117,7 +120,7 @@ func TestWatchtowerWireProtocol(t *testing.T) {
 		},
 		wtwire.MsgStateUpdateReply: func(v []reflect.Value, r *rand.Rand) {
 			req := &wtwire.StateUpdateReply{
-				Code:        uint16(r.Int31()),
+				Code:        wtwire.StateUpdateCode(r.Int31()),
 				LastApplied: uint16(r.Int31()),
 			}