package main

import (
	"os"
	"testing"
	"time"
)

func TestFindRoute(t *testing.T) {
	os.Chdir("../..")
	tp := NewTripPlanner()

	err := tp.LoadData("data")
	if err != nil {
		t.Fatalf("Failed to load data: %v", err)
	}

	err = tp.Preprocess()
	if err != nil {
		t.Fatalf("Failed to preprocess: %v", err)
	}

	route := tp.FindRoute("740000030", "740000001", time.Now())
	if route == nil {
		t.Fatal("No route found from Falun Centralstation to Stockholm Centralstation")
	}

	if len(route.Legs) < 1 {
		t.Fatal("Route has no legs")
	}

	stops := []string{}
	if len(route.Legs) > 0 {
		stops = append(stops, route.Legs[0].From)
		for _, leg := range route.Legs {
			stops = append(stops, leg.To)
		}
	}

	if len(stops) < 2 {
		t.Fatal("Route path is too short")
	}

	if stops[0] != "740000030" {
		t.Errorf("Route does not start at Falun Centralstation (740000030), starts at %s", stops[0])
	}

	if stops[len(stops)-1] != "740000001" {
		t.Errorf("Route does not end at Stockholm Centralstation (740000001), ends at %s", stops[len(stops)-1])
	}

	// Additional check: ensure all stops in path exist
	for _, stopID := range stops {
		if s, exists := tp.stops[stopID]; !exists {
			t.Errorf("Stop %s in path does not exist", stopID)
		} else {
			t.Logf("stop: %s", s.StopName)
		}
	}
}

func TestFindRouteToMalmo(t *testing.T) {
	tp := NewTripPlanner()

	err := tp.LoadData("data")
	if err != nil {
		t.Fatalf("Failed to load data: %v", err)
	}

	err = tp.Preprocess()
	if err != nil {
		t.Fatalf("Failed to preprocess: %v", err)
	}

	route := tp.FindRoute("740000030", "740000003", time.Now())
	if route == nil {
		t.Fatal("No route found from Falun Centralstation to Malmö Centralstation")
	}

	if len(route.Legs) < 1 {
		t.Fatal("Route has no legs")
	}

	stops := []string{}
	if len(route.Legs) > 0 {
		stops = append(stops, route.Legs[0].From)
		for _, leg := range route.Legs {
			stops = append(stops, leg.To)
		}
	}

	if len(stops) < 2 {
		t.Fatal("Route path is too short")
	}

	if stops[0] != "740000030" {
		t.Errorf("Route does not start at Falun Centralstation (740000030), starts at %s", stops[0])
	}

	if stops[len(stops)-1] != "740000003" {
		t.Errorf("Route does not end at Malmö Centralstation (740000003), ends at %s", stops[len(stops)-1])
	}

	// Additional check: ensure all stops in path exist
	for _, stopID := range stops {
		if _, exists := tp.stops[stopID]; !exists {
			t.Errorf("Stop %s in path does not exist", stopID)
		}
	}
}