@@ -6,6 +6,7 @@ import 'package:instabug_flutter/instabug_flutter.dart';
6
6
import 'package:instabug_flutter/src/generated/instabug.api.g.dart' ;
7
7
import 'package:instabug_flutter/src/utils/enum_converter.dart' ;
8
8
import 'package:instabug_flutter/src/utils/ibg_build_info.dart' ;
9
+ import 'package:instabug_flutter/src/utils/screen_name_masker.dart' ;
9
10
import 'package:mockito/annotations.dart' ;
10
11
import 'package:mockito/mockito.dart' ;
11
12
@@ -14,17 +15,20 @@ import 'instabug_test.mocks.dart';
14
15
@GenerateMocks ([
15
16
InstabugHostApi ,
16
17
IBGBuildInfo ,
18
+ ScreenNameMasker ,
17
19
])
18
20
void main () {
19
21
TestWidgetsFlutterBinding .ensureInitialized ();
20
22
WidgetsFlutterBinding .ensureInitialized ();
21
23
22
24
final mHost = MockInstabugHostApi ();
23
25
final mBuildInfo = MockIBGBuildInfo ();
26
+ final mScreenNameMasker = MockScreenNameMasker ();
24
27
25
28
setUpAll (() {
26
29
Instabug .$setHostApi (mHost);
27
30
IBGBuildInfo .setInstance (mBuildInfo);
31
+ ScreenNameMasker .setInstance (mScreenNameMasker);
28
32
});
29
33
30
34
test ('[setEnabled] should call host method' , () async {
@@ -76,6 +80,16 @@ void main() {
76
80
).called (1 );
77
81
});
78
82
83
+ test (
84
+ '[setScreenNameMaskingCallback] should set masking callback on screen name masker' ,
85
+ () async {
86
+ String callback (String screen) => 'REDACTED/$screen ' ;
87
+
88
+ Instabug .setScreenNameMaskingCallback (callback);
89
+
90
+ verify (mScreenNameMasker.setMaskingCallback (callback)).called (1 );
91
+ });
92
+
79
93
test ('[show] should call host method' , () async {
80
94
await Instabug .show ();
81
95
0 commit comments