feat: Persist Cotacol ride values

1 files changed, 55 insertions, 8 deletions

diff --git a/rides/rides_test.go b/rides/rides_test.go
index 1ba23f9..cbe5670 100644
--- a/rides/rides_test.go
+++ b/rides/rides_test.go
@@ -2,9 +2,12 @@ package rides
 
 import (
 	"database/sql"
+	"os"
+	"path/filepath"
 	"testing"
 	"time"
 
+	"github.com/martinlehoux/biking_home/ride"
 	"github.com/martinlehoux/kagamigo/kcore"
 	_ "github.com/mattn/go-sqlite3"
 	"github.com/stretchr/testify/assert"
@@ -29,6 +32,8 @@ func newTestDB(t *testing.T) *sql.DB {
 			elapsed_time_s integer not null,
 			total_elevation_gain_m real not null,
 			average_speed_mps real not null,
+			cotacol_score real,
+			cotacol_algo_version text,
 			created_at text not null default (strftime('%Y-%m-%dT%H:%M:%SZ', 'now')),
 			updated_at text not null default (strftime('%Y-%m-%dT%H:%M:%SZ', 'now'))
 		)
@@ -37,10 +42,13 @@ func newTestDB(t *testing.T) *sql.DB {
 	return db
 }
 
-func sampleRide() Ride {
+func sampleRide(t *testing.T) Ride {
+	t.Helper()
+	gpxPath := filepath.Join(t.TempDir(), "ride.gpx")
+	require.NoError(t, os.WriteFile(gpxPath, []byte(testGPX), 0o600))
 	return Ride{
 		ExternalID:          "strava:1234",
-		GPXPath:             "rides/activity_1234.gpx",
+		GPXPath:             gpxPath,
 		Name:                "Morning Ride",
 		Type:                "Ride",
 		StartDate:           time.Date(2026, 8, 1, 7, 30, 0, 0, time.UTC),
@@ -52,27 +60,66 @@ func sampleRide() Ride {
 	}
 }
 
+const testGPX = `<?xml version="1.0"?><gpx xmlns="http://www.topografix.com/GPX/1/1" version="1.1"><trk><trkseg><trkpt lat="43.0" lon="5.0"><ele>100</ele></trkpt><trkpt lat="43.001" lon="5.001"><ele>200</ele></trkpt></trkseg></trk></gpx>`
+
 func TestUpsertAndGet(t *testing.T) {
 	db := newTestDB(t)
-	err := Save(db, sampleRide())
+	sample := sampleRide(t)
+	err := Save(db, sample)
 	require.NoError(t, err)
 
 	got, ok, err := GetByExternalID(db, "strava:1234")
 	require.NoError(t, err)
 	require.True(t, ok)
 	assert.Equal(t, "Morning Ride", got.Name)
-	assert.Equal(t, sampleRide().StartDate, got.StartDate)
+	assert.Equal(t, sample.StartDate, got.StartDate)
 	assert.Equal(t, 42_195.0, got.DistanceM)
-	assert.Equal(t, "rides/activity_1234.gpx", got.GPXPath)
+	assert.Equal(t, sample.GPXPath, got.GPXPath)
+	score, found := got.CotacolScore()
+	require.True(t, found)
+	assert.Greater(t, score, 0.0)
+	assert.Equal(t, ride.CotacolAlgorithmVersion, got.CotacolAlgorithmVersion())
 
 	_, ok, err = GetByExternalID(db, "strava:9999")
 	require.NoError(t, err)
 	assert.False(t, ok)
 }
 
+func TestSaveComputesCotacol(t *testing.T) {
+	db := newTestDB(t)
+	require.NoError(t, Save(db, sampleRide(t)))
+
+	got, ok, err := GetByExternalID(db, "strava:1234")
+	require.NoError(t, err)
+	require.True(t, ok)
+	score, found := got.CotacolScore()
+	require.True(t, found)
+	assert.Greater(t, score, 0.0)
+	assert.Equal(t, ride.CotacolAlgorithmVersion, got.CotacolAlgorithmVersion())
+}
+
+func TestBackfill(t *testing.T) {
+	db := newTestDB(t)
+	require.NoError(t, Save(db, sampleRide(t)))
+
+	count, err := Backfill(db)
+	require.NoError(t, err)
+	assert.Equal(t, 1, count)
+	got, ok, err := GetByExternalID(db, "strava:1234")
+	require.NoError(t, err)
+	require.True(t, ok)
+	_, found := got.CotacolScore()
+	assert.True(t, found)
+	assert.Equal(t, ride.CotacolAlgorithmVersion, got.CotacolAlgorithmVersion())
+
+	count, err = Backfill(db)
+	require.NoError(t, err)
+	assert.Equal(t, 1, count)
+}
+
 func TestUpsertUpdatesExisting(t *testing.T) {
 	db := newTestDB(t)
-	ride := sampleRide()
+	ride := sampleRide(t)
 	require.NoError(t, Save(db, ride))
 	ride.Name = "Renamed Ride"
 	ride.DistanceM = 50_000
@@ -87,9 +134,9 @@ func TestUpsertUpdatesExisting(t *testing.T) {
 
 func TestList(t *testing.T) {
 	db := newTestDB(t)
-	first := sampleRide()
+	first := sampleRide(t)
 	first.StartDate = time.Date(2026, 7, 1, 7, 0, 0, 0, time.UTC)
-	second := sampleRide()
+	second := sampleRide(t)
 	second.ExternalID = "strava:5678"
 	second.Name = "Evening Ride"
 	require.NoError(t, Save(db, first))