SentryIOOverridesIntegration

CHANGELOG.md

@@ -5,6 +5,7 @@
 ### Features
 
 - SentryUserInteractionWidget: add support for PopupMenuButton and PopupMenuItem ([#1361](https://github.com/getsentry/sentry-dart/pull/1361))
+- SentryIOOverridesIntegration ([#1362](https://github.com/getsentry/sentry-dart/pull/1362))
 
 ### Fixes
 

file/lib/sentry_file.dart

@@ -1,2 +1,3 @@
 export 'src/sentry_file.dart';
 export 'src/sentry_file_extension.dart';
+export 'src/sentry_io_overrides_integration.dart';

file/lib/src/sentry_io_overrides.dart

@@ -0,0 +1,18 @@
+import 'dart:io';
+import 'package:sentry/sentry.dart';
+
+import 'sentry_file.dart';
+
+class SentryIOOverrides extends IOOverrides {
+  final Hub _hub;
+
+  SentryIOOverrides(this._hub);
+
+  @override
+  File createFile(String path) {
+    return SentryFile(
+      super.createFile(path),
+      hub: _hub,
+    );
+  }
+}

file/lib/src/sentry_io_overrides_integration.dart

@@ -0,0 +1,32 @@
+import 'dart:async';
+import 'dart:io';
+
+import 'package:sentry/sentry.dart';
+import 'sentry_io_overrides.dart';
+
+/// When installed, every new file will be created as [SentryFile].
+/// When installed, operations will use [SentryFile] instead of dart:io's [File]
+/// implementation whenever [File] is used.
+///
+/// Must be called outside the RunZoneGuardedIntegration.
+/// Otherwise the FileOverrides aren't attached to the root zone.
+class SentryIOOverridesIntegration extends Integration<SentryOptions> {
+  IOOverrides? _previousOverrides;
+  SentryOptions? _options;
+  @override
+  FutureOr<void> call(Hub hub, SentryOptions options) {
+    _options = options;
+    if (options.isTracingEnabled()) {
+      _previousOverrides = IOOverrides.current;
+      IOOverrides.global = SentryIOOverrides(hub);
+      options.sdk.addIntegration('sentryIOOverridesIntegration');
+    }
+  }
+
+  @override
+  FutureOr<void> close() {
+    if (_options?.isTracingEnabled() ?? false) {
+      IOOverrides.global = _previousOverrides;
+    }
+  }
+}

file/test/sentry_io_overrides_integration_test.dart

@@ -0,0 +1,68 @@
+import 'dart:io';
+
+import 'package:sentry/sentry.dart';
+import 'package:sentry_file/sentry_file.dart';
+import 'package:sentry_file/src/sentry_io_overrides.dart';
+import 'package:test/expect.dart';
+import 'package:test/scaffolding.dart';
+
+import 'mock_sentry_client.dart';
+
+void main() {
+  late IOOverrides? current;
+  late Fixture fixture;
+
+  setUp(() {
+    current = IOOverrides.current;
+    fixture = Fixture();
+  });
+
+  tearDown(() {
+    IOOverrides.global = current;
+  });
+
+  test('adding integration installs io overrides', () {
+    fixture.options.tracesSampleRate = 1.0;
+
+    final sut = fixture.getSut();
+    sut.call(fixture.hub, fixture.options);
+
+    expect(
+      fixture.options.sdk.integrations.contains('sentryIOOverridesIntegration'),
+      isTrue,
+    );
+    expect(IOOverrides.current is SentryIOOverrides, isTrue);
+  });
+
+  test('not installed when tracing disabled', () {
+    final sut = fixture.getSut();
+    sut.call(fixture.hub, fixture.options);
+
+    expect(
+      fixture.options.sdk.integrations.contains('sentryIOOverridesIntegration'),
+      isFalse,
+    );
+    expect(IOOverrides.current is SentryIOOverrides, isFalse);
+  });
+
+  test('global overrides restored', () {
+    final previous = IOOverrides.current;
+
+    fixture.options.tracesSampleRate = 1.0;
+
+    final sut = fixture.getSut();
+    sut.call(fixture.hub, fixture.options);
+    sut.close();
+
+    expect(IOOverrides.current, previous);
+  });
+}
+
+class Fixture {
+  final options = SentryOptions(dsn: fakeDsn);
+  late final hub = Hub(options);
+
+  SentryIOOverridesIntegration getSut() {
+    return SentryIOOverridesIntegration();
+  }
+}