Files
go-caatsm/internal/adapter/parser/pattern_test.go
T

44 lines
1.2 KiB
Go
Raw Normal View History

package parser
2024-07-19 20:02:18 +08:00
import (
. "github.com/onsi/ginkgo/v2"
. "github.com/onsi/gomega"
)
var _ = Describe("Pattern Parser", func() {
Describe("FindPatterns", func() {
It("should return the correct BodyConfig based on the message body", func() {
message := "(ARR-AB123-SSR1234-KJFK-KLAX)"
bodyConfig := FindPatterns(message)
2024-07-19 20:02:18 +08:00
Expect(bodyConfig).NotTo(BeNil())
// Expect(bodyConfig.Name).To(Equal("ARR"))
2024-07-19 20:02:18 +08:00
})
It("should return nil if no pattern matches", func() {
message := "(XYZ-123)"
bodyConfig := FindPatterns(message)
2024-07-19 20:02:18 +08:00
Expect(bodyConfig).To(BeNil())
})
})
Describe("ParseBody", func() {
It("should parse the message body and extract data based on patterns", func() {
2024-07-23 16:45:03 +08:00
message := "(ARR-AB123/A1234-KJFK-KLAX1234)"
parsedData := ParseBody(message)
2024-07-19 20:02:18 +08:00
Expect(parsedData).NotTo(BeNil())
2024-07-21 00:25:16 +08:00
Expect(parsedData["category"]).To(Equal("ARR"))
2024-07-19 20:02:18 +08:00
Expect(parsedData["number"]).To(Equal("AB123"))
2024-07-23 16:45:03 +08:00
Expect(parsedData["ssr"]).To(Equal("A1234"))
2024-07-29 15:40:20 +08:00
Expect(parsedData[DepartureCode]).To(Equal("KJFK"))
Expect(parsedData[ArrivalCode]).To(Equal("KLAX"))
2024-07-19 20:02:18 +08:00
})
It("should return nil if no patterns match", func() {
message := "(XYZ-123)"
parsedData := ParseBody(message)
2024-07-19 20:02:18 +08:00
Expect(parsedData).To(BeNil())
})
})
})