get locode: support the most widely used format.

The format of "XX YYY" as well as "XXYYY" now supported for Get calls.

Signed-off-by: Ekaterina Pavlova <[email protected]>

pkg/locodedb/calls.go

@@ -24,7 +24,14 @@ func Get(locodeStr string) (Record, error) {
 		return Record{}, err
 	}
 
-	key, err := NewKey(locodeStr[:2], locodeStr[2:])
+	country := locodeStr[:2]
+	location := locodeStr[2:]
+
+	if len(locodeStr) == 6 {
+		location = locodeStr[3:]
+	}
+
+	key, err := NewKey(country, location)
 	if err != nil {
 		return Record{}, err
 	}

pkg/locodedb/calls_test.go

@@ -19,7 +19,7 @@ func TestGet(t *testing.T) {
 	})
 
 	t.Run("locode", func(t *testing.T) {
-		rec, err := locodedb.Get("RUMOW")
+		rec, err := locodedb.Get("RU MOW")
 		require.NoError(t, err)
 
 		require.Equal(t, rec.Country, "Russia")