fix: track was not ended when behaviour resulted in a future

Author: wim07101993

CHANGELOG.md

@@ -1,3 +1,7 @@
+## 3.0.1
+
+* fix:  track was not ended when behaviour resulted in a future
+
 ## 3.0.0
 
 * feat!: upgraded to dart v3

lib/src/behaviour_mixin.dart

@@ -55,7 +55,9 @@ mixin BehaviourMixin {
     BehaviourTrack? track,
   ) async {
     try {
-      return await future;
+      final result = await future;
+      track?.end();
+      return result;
     } catch (error, stackTrace) {
       return _catch(error, stackTrace, track);
     }

pubspec.yaml

@@ -2,7 +2,7 @@ name: behaviour
 description:
   This package adds support for behaviours. Behaviours are classes of which the
   instances are used as functions.
-version: 3.0.0
+version: 3.0.1
 repository: https://github.com/wim07101993/behaviour
 
 environment:

test/src/behaviour_mixin_test.dart

@@ -89,7 +89,7 @@ void main() {
       });
     });
 
-    test('should start and end the track', () async {
+    test('should start and end the track in case of sync', () async {
       // act
       await behaviour.executeAction((track) => const Success(Object()));
 
@@ -102,6 +102,21 @@ void main() {
       verifyNever(() => mockTrack.stopWithException(any(), any()));
     });
 
+    test('should start and end the track in case of future', () async {
+      // act
+      await behaviour.executeAction(
+        (track) => Future.value(const Success(Object())),
+      );
+
+      // assert
+      verifyInOrder([
+        () => mockTrack.start(),
+        () => mockTrack.end(),
+      ]);
+      verifyNever(() => mockTrack.stopWithError(any(), any()));
+      verifyNever(() => mockTrack.stopWithException(any(), any()));
+    });
+
     test('should execute and return result of action', () async {
       // arrange
       final expected = faker.randomGenerator.element<ExceptionOr<String>>([