refactor: Add schedule_parser.go and schedule_parser_test.go

The code changes introduce a new `schedule_parser.go` file and its corresponding test file `schedule_parser_test.go`. These files contain the implementation and tests for the `FindWaypoints` function, which extracts waypoints from a given message. This addition enhances the functionality of the code by providing a way to parse and retrieve specific information from messages.
This commit is contained in:
windyboy
2024-07-29 15:40:30 +08:00
parent 2a512d419a
commit bb4faf4d13
2 changed files with 57 additions and 0 deletions
+31
View File
@@ -0,0 +1,31 @@
package parsers
import "regexp"
const (
// DepartureCode = "dep"
// DepartureTime = "dep_time"
// ArrivalCode = "arr"
AirportCode = "airport"
WaypointPattern = `(?P<arr_time>\d{4}(\(\d{2}\w{3}\))?)(?P<airport>\w{3})\/?(?P<dep_time>\d{4}(\(\d{2}\w{3}\))?)`
)
var (
WaypointExpression = regexp.MustCompile(WaypointPattern)
)
func FindWaypoints(message string) map[string]string {
matches := WaypointExpression.FindStringSubmatch(message)
if matches == nil {
return nil
}
result := make(map[string]string)
for i, name := range WaypointExpression.SubexpNames() {
if i != 0 && name != "" {
result[name] = matches[i]
}
}
return result
}
+26
View File
@@ -0,0 +1,26 @@
package parsers
import (
. "github.com/onsi/ginkgo/v2"
. "github.com/onsi/gomega"
)
var _ = Describe("Schedule Parser", func() {
Describe("FindWaypoints", func() {
It("should return the correct waypoints based on the message", func() {
message := "1845(11JUN)TSN/2100"
waypoints := FindWaypoints(message)
Expect(waypoints).NotTo(BeNil())
Expect(waypoints[ArrivalTime]).To(Equal("1845(11JUN)"))
Expect(waypoints[AirportCode]).To(Equal("TSN"))
Expect(waypoints[DepartureTime]).To(Equal("2100"))
})
It("should return nil if no waypoints are found", func() {
message := "1845TSN"
waypoints := FindWaypoints(message)
Expect(waypoints).To(BeNil())
})
})
})