@@ -30,16 +30,16 @@ impl Writer for VecWriter {
30
30
#[ macro_export]
31
31
macro_rules! test_msg {
32
32
( $MsgType: path, $data: ident) => { {
33
- use lightning:: util:: ser:: { Readable , Writeable } ;
34
- let mut r = :: lightning :: io :: Cursor :: new ( $data) ;
35
- if let Ok ( msg) = <$MsgType as Readable >:: read ( & mut r) {
36
- let p = r . position ( ) as usize ;
33
+ use lightning:: util:: ser:: { LengthReadable , Writeable } ;
34
+ let mut r = & $data[ .. ] ;
35
+ if let Ok ( msg) = <$MsgType as LengthReadable >:: read_from_fixed_length_buffer ( & mut r) {
36
+ let p = $data . len ( ) - r . len ( ) as usize ;
37
37
let mut w = VecWriter ( Vec :: new( ) ) ;
38
38
msg. write( & mut w) . unwrap( ) ;
39
39
40
40
assert_eq!( w. 0 . len( ) , p) ;
41
41
assert_eq!( msg. serialized_length( ) , p) ;
42
- assert_eq!( & r . into_inner ( ) [ ..p] , & w. 0 [ ..p] ) ;
42
+ assert_eq!( & $data [ ..p] , & w. 0 [ ..p] ) ;
43
43
}
44
44
} } ;
45
45
}
@@ -87,17 +87,16 @@ macro_rules! test_msg_exact {
87
87
#[ macro_export]
88
88
macro_rules! test_msg_hole {
89
89
( $MsgType: path, $data: ident, $hole: expr, $hole_len: expr) => { {
90
- use lightning:: util:: ser:: { Readable , Writeable } ;
91
- let mut r = :: lightning:: io:: Cursor :: new( $data) ;
92
- if let Ok ( msg) = <$MsgType as Readable >:: read( & mut r) {
90
+ use lightning:: util:: ser:: { LengthReadable , Writeable } ;
91
+ if let Ok ( msg) = <$MsgType as LengthReadable >:: read_from_fixed_length_buffer( & mut & $data[ ..] ) {
93
92
let mut w = VecWriter ( Vec :: new( ) ) ;
94
93
msg. write( & mut w) . unwrap( ) ;
95
94
let p = w. 0 . len( ) as usize ;
96
95
assert_eq!( msg. serialized_length( ) , p) ;
97
96
98
97
assert_eq!( w. 0 . len( ) , p) ;
99
- assert_eq!( & r . get_ref ( ) [ ..$hole] , & w. 0 [ ..$hole] ) ;
100
- assert_eq!( & r . get_ref ( ) [ $hole + $hole_len..p] , & w. 0 [ $hole + $hole_len..] ) ;
98
+ assert_eq!( & $data [ ..$hole] , & w. 0 [ ..$hole] ) ;
99
+ assert_eq!( & $data [ $hole + $hole_len..p] , & w. 0 [ $hole + $hole_len..] ) ;
101
100
}
102
101
} } ;
103
102
}
0 commit comments