omi/app/test/unit/app_session_telemetry_test.dart at main · ghzhost/omi · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
import 'package:flutter_test/flutter_test.dart';
import 'package:omi/backend/preferences.dart';
import 'package:omi/utils/analytics/analytics_adapter.dart';
import 'package:omi/utils/analytics/analytics_manager.dart';
import 'package:omi/utils/analytics/app_session_telemetry.dart';
import 'package:shared_preferences/shared_preferences.dart';

void main() {
  TestWidgetsFlutterBinding.ensureInitialized();

  late _FakeAnalyticsAdapter adapter;
  late AppSessionTelemetry telemetry;
  var nextId = 0;

  setUp(() async {
    nextId = 0;
    AnalyticsManager.resetForTesting();
    SharedPreferences.setMockInitialValues({});
    await SharedPreferencesUtil.init();
    adapter = _FakeAnalyticsAdapter();
    AnalyticsManager.configure(adapter);
    await AnalyticsManager.init();
    telemetry = AppSessionTelemetry(
      createSessionId: () => 'session-${++nextId}',
    );
  });

  tearDown(AnalyticsManager.resetForTesting);

  test('records one cold-start session', () async {
    telemetry.recordColdStart();
    telemetry.recordColdStart();
    await AnalyticsManager.flushPending(force: true);

    expect(adapter.events, hasLength(1));
    expect(adapter.events.single.eventName, 'App Session Started');
    expect(adapter.events.single.properties, containsPair('app_session_id', 'session-1'));
    expect(adapter.events.single.properties, containsPair('start_kind', 'coldStart'));
  });

  test('records a fresh session only after backgrounding', () async {
    telemetry.recordColdStart();
    telemetry.recordResumed();
    telemetry.recordBackgrounded();
    telemetry.recordResumed();
    telemetry.recordResumed();
    await AnalyticsManager.flushPending(force: true);

    expect(adapter.events.map((event) => event.eventName), ['App Session Started', 'App Session Started']);
    expect(adapter.events[0].properties, containsPair('app_session_id', 'session-1'));
    expect(adapter.events[0].properties, containsPair('start_kind', 'coldStart'));
    expect(adapter.events[1].properties, containsPair('app_session_id', 'session-2'));
    expect(adapter.events[1].properties, containsPair('start_kind', 'foreground'));
  });
}

class _FakeAnalyticsAdapter implements AnalyticsAdapter {
  final List<_RecordedEvent> events = [];

  @override
  bool get isInitialized => true;

  @override
  Future<void> init() async {}

  @override
  void identify({
    required String userId,
    Map<String, Object>? userProperties,
  }) {}

  @override
  void alias({required String newUserId}) {}

  @override
  void track({required String eventName, Map<String, Object>? properties}) {
    events.add(_RecordedEvent(eventName, properties ?? const {}));
  }

  @override
  void setInteractionContext({String? screenName, required String target}) {}

  @override
  void enable() {}

  @override
  void disable() {}

  @override
  void reset() {}
}

class _RecordedEvent {
  const _RecordedEvent(this.eventName, this.properties);

  final String eventName;
  final Map<String, Object> properties;
}