test(policy_manager): update error handling to use PolicySDKException in integration tests

Author: dgarciasquare1

test/core/integration_test.dart

@@ -1,7 +1,9 @@
+import 'package:flutter_policy_engine/flutter_policy_engine.dart';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:flutter_policy_engine/src/core/policy_manager.dart';
 import 'package:flutter_policy_engine/src/core/memory_policy_storage.dart';
 import 'package:flutter_policy_engine/src/core/role_evaluator.dart';
+import 'package:flutter_policy_engine/src/exceptions/policy_sdk_exception.dart';
 
 void main() {
   group('Core Integration Tests', () {
@@ -169,7 +171,7 @@ void main() {
 
         // Should throw when storage fails
         expect(() => failingManager.initialize(jsonPolicies),
-            throwsA(isA<StateError>()));
+            throwsA(isA<PolicySDKException>()));
         expect(failingManager.isInitialized, isFalse);
       });
 

test/core/policy_manager_test.dart

@@ -207,7 +207,7 @@ void main() {
         };
 
         expect(() => policyManager.initialize(jsonPolicies),
-            throwsA(isA<StateError>()));
+            throwsA(isA<PolicySDKException>()));
         expect(policyManager.isInitialized, isFalse);
       });
 
@@ -343,7 +343,7 @@ void main() {
         mockStorage.setShouldThrowOnSave(true);
 
         expect(() => policyManager.initialize(jsonPolicies),
-            throwsA(isA<StateError>()));
+            throwsA(isA<PolicySDKException>()));
         expect(policyManager.isInitialized, isFalse);
       });
 
@@ -425,7 +425,7 @@ void main() {
 
         expect(
           () => policyManager.initialize(jsonPolicies),
-          throwsA(isA<StateError>()),
+          throwsA(isA<PolicySDKException>()),
         );
 
         expect(policyManager.isInitialized, isFalse);
@@ -444,7 +444,7 @@ void main() {
 
         expect(
           () => policyManager.initialize(jsonPolicies),
-          throwsA(isA<StateError>()),
+          throwsA(isA<PolicySDKException>()),
         );
       });
     });
@@ -931,11 +931,14 @@ void main() {
         // Make storage throw on save
         mockStorage.setShouldThrowOnSave(true);
 
-        // The method catches exceptions and doesn't rethrow them
-        await policyManager
-            .initializeFromJsonAssets('assets/policies/test.json');
+        // The method should throw PolicySDKException when storage fails
+        expect(
+          () => policyManager
+              .initializeFromJsonAssets('assets/policies/test.json'),
+          throwsA(isA<PolicySDKException>()),
+        );
 
-        // Should handle gracefully and not initialize due to storage error
+        // Should not be initialized due to storage error
         expect(policyManager.isInitialized, isFalse);
 
         // Clean up mock message handler