[trails] Pull data out of FIT/GPX files

tests/trails_tests/test_gpx_importer.py

@@ -8,11 +8,12 @@ from django.core.files import File
 
 from locations.models import GeoLocation
 from scrobbles.importers.trail_gpx import (
+    compute_trail_stats,
     import_trail_gpx,
     parse_trackpoints,
 )
 from scrobbles.models import Scrobble, TrailGPXImport
-from trails.models import Trail
+from trails.models import Trail, TrailLogData
 
 User = get_user_model()
 
@@ -33,9 +34,11 @@ def sample_gpx_path():
 
 class TestParseTrackpoints:
     def test_parses_gpx(self, sample_gpx_path):
-        points, name = parse_trackpoints(sample_gpx_path)
+        result = parse_trackpoints(sample_gpx_path)
+        points = result["points"]
         assert len(points) == 837
-        assert name == "Morning Run ⛅"
+        assert result["name"] == "Morning Run ⛅"
+        assert result["description"] == "Run"
         lat, lon, ele, t = points[0]
         assert round(lat, 6) == 34.190598
         assert round(lon, 6) == -118.844015
@@ -43,12 +46,23 @@ class TestParseTrackpoints:
         assert t is not None
 
     def test_first_and_last_times(self, sample_gpx_path):
-        points, _ = parse_trackpoints(sample_gpx_path)
+        result = parse_trackpoints(sample_gpx_path)
+        points = result["points"]
         first_time = points[0][3]
         last_time = points[-1][3]
         duration = (last_time - first_time).total_seconds()
         assert duration == pytest.approx(3770, abs=5)
 
+    def test_gpx_extra_metadata(self, sample_gpx_path):
+        result = parse_trackpoints(sample_gpx_path)
+        extra = result["extra"]
+        assert extra["avg_heartrate"] == 159
+        assert extra["max_heartrate"] == 183
+        assert extra["avg_speed_kmh"] == pytest.approx(9.82, abs=0.1)
+        assert extra["activity_type"] == "Run"
+        assert extra["moving_time_seconds"] == 3008
+        assert extra["total_elevation_gain_m"] == 246.4
+
 
 class TestImportTrailGPX:
     def test_creates_trail(self, user, sample_gpx_path):
@@ -100,6 +114,44 @@ class TestImportTrailGPX:
         import_trail_gpx(sample_gpx_path, user.id)
         assert Scrobble.objects.filter(source="GPX Import").count() == 1
 
+    def test_scrobble_log_has_stats(self, user, sample_gpx_path):
+        import_trail_gpx(sample_gpx_path, user.id)
+        scrobble = Scrobble.objects.filter(source="GPX Import").first()
+        log = scrobble.log
+        assert log["distance_km"] == pytest.approx(8.2, abs=0.2)
+        assert log["elevation_gain_m"] == pytest.approx(260, abs=20)
+        assert log["moving_time_seconds"] == pytest.approx(3770, abs=10)
+        assert log["avg_speed_kmh"] == pytest.approx(7.8, abs=0.5)
+        assert log["description"] == "Run"
+
+    def test_scrobble_playback_position(self, user, sample_gpx_path):
+        import_trail_gpx(sample_gpx_path, user.id)
+        scrobble = Scrobble.objects.filter(source="GPX Import").first()
+        assert scrobble.playback_position_seconds == pytest.approx(3770, abs=5)
+
+    def test_scrobble_log_extra_metadata(self, user, sample_gpx_path):
+        import_trail_gpx(sample_gpx_path, user.id)
+        scrobble = Scrobble.objects.filter(source="GPX Import").first()
+        log = scrobble.log
+        assert log["avg_heartrate"] == 159
+        assert log["max_heartrate"] == 183
+        assert log["activity_type"] == "Run"
+
+    def test_scrobble_log_no_calories_in_gpx(self, user, sample_gpx_path):
+        import_trail_gpx(sample_gpx_path, user.id)
+        scrobble = Scrobble.objects.filter(source="GPX Import").first()
+        assert scrobble.log.get("calories") is None
+
+
+class TestComputeTrailStats:
+    def test_computes_distance_and_elevation(self, sample_gpx_path):
+        result = parse_trackpoints(sample_gpx_path)
+        stats = compute_trail_stats(result["points"])
+        assert stats["distance_km"] == pytest.approx(8.2, abs=0.2)
+        assert stats["elevation_gain_m"] == pytest.approx(260, abs=20)
+        assert stats["moving_time_seconds"] == pytest.approx(3770, abs=10)
+        assert stats["avg_speed_kmh"] == pytest.approx(7.8, abs=0.5)
+
 
 class TestTrailGPXImportModel:
     def test_create_import_model(self, db, user, sample_gpx_path):