From 8116eec17261d6dc5475966e4ad8e03c09e92342 Mon Sep 17 00:00:00 2001
From: lamnhan066 <hello@lamnhan.dev>
Date: Fri, 23 Aug 2024 13:22:11 +0700
Subject: [PATCH] Improve code coverage

---
 lib/src/utils/print_debug.dart | 18 +++++++++++++++---
 test/language_helper_test.dart |  2 ++
 2 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/lib/src/utils/print_debug.dart b/lib/src/utils/print_debug.dart
index bfabe2f..4e52114 100644
--- a/lib/src/utils/print_debug.dart
+++ b/lib/src/utils/print_debug.dart
@@ -1,7 +1,19 @@
 import 'package:flutter/foundation.dart';
 import 'package:language_helper/src/language_helper.dart';
 
+/// Improve code coverage for debug logs
+bool isTestingDebugLog = false;
+
 /// Internal function, print debug log
-void printDebug(Object? Function() object) => LanguageHelper.instance.isDebug
-    ? debugPrint('[Language Helper] ${object()}')
-    : null;
+void printDebug(Object? Function() object) {
+  assert(() {
+    if (isTestingDebugLog) {
+      object();
+    }
+    return true;
+  }());
+
+  if (LanguageHelper.instance.isDebug) {
+    debugPrint('[Language Helper] ${object()}');
+  }
+}
diff --git a/test/language_helper_test.dart b/test/language_helper_test.dart
index 868ca76..d18538e 100644
--- a/test/language_helper_test.dart
+++ b/test/language_helper_test.dart
@@ -8,6 +8,7 @@ import 'package:flutter_test/flutter_test.dart';
 import 'package:language_code/language_code.dart';
 import 'package:language_helper/language_helper.dart';
 import 'package:language_helper/src/mixins/update_language.dart';
+import 'package:language_helper/src/utils/print_debug.dart';
 import 'package:language_helper/src/utils/utils.dart';
 import 'package:shared_preferences/shared_preferences.dart';
 
@@ -26,6 +27,7 @@ void main() async {
   StreamSubscription? languageSub;
 
   setUpAll(() async {
+    isTestingDebugLog = true;
     await languageHelper.initial(
       data: dataList,
       initialCode: LanguageCodes.en,