decode_test.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989
  1. package yaml_test
  2. import (
  3. "errors"
  4. . "gopkg.in/check.v1"
  5. "gopkg.in/yaml.v2"
  6. "math"
  7. "net"
  8. "reflect"
  9. "strings"
  10. "time"
  11. )
  12. var unmarshalIntTest = 123
  13. var unmarshalTests = []struct {
  14. data string
  15. value interface{}
  16. }{
  17. {
  18. "",
  19. &struct{}{},
  20. }, {
  21. "{}", &struct{}{},
  22. }, {
  23. "v: hi",
  24. map[string]string{"v": "hi"},
  25. }, {
  26. "v: hi", map[string]interface{}{"v": "hi"},
  27. }, {
  28. "v: true",
  29. map[string]string{"v": "true"},
  30. }, {
  31. "v: true",
  32. map[string]interface{}{"v": true},
  33. }, {
  34. "v: 10",
  35. map[string]interface{}{"v": 10},
  36. }, {
  37. "v: 0b10",
  38. map[string]interface{}{"v": 2},
  39. }, {
  40. "v: 0xA",
  41. map[string]interface{}{"v": 10},
  42. }, {
  43. "v: 4294967296",
  44. map[string]int64{"v": 4294967296},
  45. }, {
  46. "v: 0.1",
  47. map[string]interface{}{"v": 0.1},
  48. }, {
  49. "v: .1",
  50. map[string]interface{}{"v": 0.1},
  51. }, {
  52. "v: .Inf",
  53. map[string]interface{}{"v": math.Inf(+1)},
  54. }, {
  55. "v: -.Inf",
  56. map[string]interface{}{"v": math.Inf(-1)},
  57. }, {
  58. "v: -10",
  59. map[string]interface{}{"v": -10},
  60. }, {
  61. "v: -.1",
  62. map[string]interface{}{"v": -0.1},
  63. },
  64. // Simple values.
  65. {
  66. "123",
  67. &unmarshalIntTest,
  68. },
  69. // Floats from spec
  70. {
  71. "canonical: 6.8523e+5",
  72. map[string]interface{}{"canonical": 6.8523e+5},
  73. }, {
  74. "expo: 685.230_15e+03",
  75. map[string]interface{}{"expo": 685.23015e+03},
  76. }, {
  77. "fixed: 685_230.15",
  78. map[string]interface{}{"fixed": 685230.15},
  79. }, {
  80. "neginf: -.inf",
  81. map[string]interface{}{"neginf": math.Inf(-1)},
  82. }, {
  83. "fixed: 685_230.15",
  84. map[string]float64{"fixed": 685230.15},
  85. },
  86. //{"sexa: 190:20:30.15", map[string]interface{}{"sexa": 0}}, // Unsupported
  87. //{"notanum: .NaN", map[string]interface{}{"notanum": math.NaN()}}, // Equality of NaN fails.
  88. // Bools from spec
  89. {
  90. "canonical: y",
  91. map[string]interface{}{"canonical": true},
  92. }, {
  93. "answer: NO",
  94. map[string]interface{}{"answer": false},
  95. }, {
  96. "logical: True",
  97. map[string]interface{}{"logical": true},
  98. }, {
  99. "option: on",
  100. map[string]interface{}{"option": true},
  101. }, {
  102. "option: on",
  103. map[string]bool{"option": true},
  104. },
  105. // Ints from spec
  106. {
  107. "canonical: 685230",
  108. map[string]interface{}{"canonical": 685230},
  109. }, {
  110. "decimal: +685_230",
  111. map[string]interface{}{"decimal": 685230},
  112. }, {
  113. "octal: 02472256",
  114. map[string]interface{}{"octal": 685230},
  115. }, {
  116. "hexa: 0x_0A_74_AE",
  117. map[string]interface{}{"hexa": 685230},
  118. }, {
  119. "bin: 0b1010_0111_0100_1010_1110",
  120. map[string]interface{}{"bin": 685230},
  121. }, {
  122. "bin: -0b101010",
  123. map[string]interface{}{"bin": -42},
  124. }, {
  125. "decimal: +685_230",
  126. map[string]int{"decimal": 685230},
  127. },
  128. //{"sexa: 190:20:30", map[string]interface{}{"sexa": 0}}, // Unsupported
  129. // Nulls from spec
  130. {
  131. "empty:",
  132. map[string]interface{}{"empty": nil},
  133. }, {
  134. "canonical: ~",
  135. map[string]interface{}{"canonical": nil},
  136. }, {
  137. "english: null",
  138. map[string]interface{}{"english": nil},
  139. }, {
  140. "~: null key",
  141. map[interface{}]string{nil: "null key"},
  142. }, {
  143. "empty:",
  144. map[string]*bool{"empty": nil},
  145. },
  146. // Flow sequence
  147. {
  148. "seq: [A,B]",
  149. map[string]interface{}{"seq": []interface{}{"A", "B"}},
  150. }, {
  151. "seq: [A,B,C,]",
  152. map[string][]string{"seq": []string{"A", "B", "C"}},
  153. }, {
  154. "seq: [A,1,C]",
  155. map[string][]string{"seq": []string{"A", "1", "C"}},
  156. }, {
  157. "seq: [A,1,C]",
  158. map[string][]int{"seq": []int{1}},
  159. }, {
  160. "seq: [A,1,C]",
  161. map[string]interface{}{"seq": []interface{}{"A", 1, "C"}},
  162. },
  163. // Block sequence
  164. {
  165. "seq:\n - A\n - B",
  166. map[string]interface{}{"seq": []interface{}{"A", "B"}},
  167. }, {
  168. "seq:\n - A\n - B\n - C",
  169. map[string][]string{"seq": []string{"A", "B", "C"}},
  170. }, {
  171. "seq:\n - A\n - 1\n - C",
  172. map[string][]string{"seq": []string{"A", "1", "C"}},
  173. }, {
  174. "seq:\n - A\n - 1\n - C",
  175. map[string][]int{"seq": []int{1}},
  176. }, {
  177. "seq:\n - A\n - 1\n - C",
  178. map[string]interface{}{"seq": []interface{}{"A", 1, "C"}},
  179. },
  180. // Literal block scalar
  181. {
  182. "scalar: | # Comment\n\n literal\n\n \ttext\n\n",
  183. map[string]string{"scalar": "\nliteral\n\n\ttext\n"},
  184. },
  185. // Folded block scalar
  186. {
  187. "scalar: > # Comment\n\n folded\n line\n \n next\n line\n * one\n * two\n\n last\n line\n\n",
  188. map[string]string{"scalar": "\nfolded line\nnext line\n * one\n * two\n\nlast line\n"},
  189. },
  190. // Map inside interface with no type hints.
  191. {
  192. "a: {b: c}",
  193. map[interface{}]interface{}{"a": map[interface{}]interface{}{"b": "c"}},
  194. },
  195. // Structs and type conversions.
  196. {
  197. "hello: world",
  198. &struct{ Hello string }{"world"},
  199. }, {
  200. "a: {b: c}",
  201. &struct{ A struct{ B string } }{struct{ B string }{"c"}},
  202. }, {
  203. "a: {b: c}",
  204. &struct{ A *struct{ B string } }{&struct{ B string }{"c"}},
  205. }, {
  206. "a: {b: c}",
  207. &struct{ A map[string]string }{map[string]string{"b": "c"}},
  208. }, {
  209. "a: {b: c}",
  210. &struct{ A *map[string]string }{&map[string]string{"b": "c"}},
  211. }, {
  212. "a:",
  213. &struct{ A map[string]string }{},
  214. }, {
  215. "a: 1",
  216. &struct{ A int }{1},
  217. }, {
  218. "a: 1",
  219. &struct{ A float64 }{1},
  220. }, {
  221. "a: 1.0",
  222. &struct{ A int }{1},
  223. }, {
  224. "a: 1.0",
  225. &struct{ A uint }{1},
  226. }, {
  227. "a: [1, 2]",
  228. &struct{ A []int }{[]int{1, 2}},
  229. }, {
  230. "a: 1",
  231. &struct{ B int }{0},
  232. }, {
  233. "a: 1",
  234. &struct {
  235. B int "a"
  236. }{1},
  237. }, {
  238. "a: y",
  239. &struct{ A bool }{true},
  240. },
  241. // Some cross type conversions
  242. {
  243. "v: 42",
  244. map[string]uint{"v": 42},
  245. }, {
  246. "v: -42",
  247. map[string]uint{},
  248. }, {
  249. "v: 4294967296",
  250. map[string]uint64{"v": 4294967296},
  251. }, {
  252. "v: -4294967296",
  253. map[string]uint64{},
  254. },
  255. // int
  256. {
  257. "int_max: 2147483647",
  258. map[string]int{"int_max": math.MaxInt32},
  259. },
  260. {
  261. "int_min: -2147483648",
  262. map[string]int{"int_min": math.MinInt32},
  263. },
  264. {
  265. "int_overflow: 9223372036854775808", // math.MaxInt64 + 1
  266. map[string]int{},
  267. },
  268. // int64
  269. {
  270. "int64_max: 9223372036854775807",
  271. map[string]int64{"int64_max": math.MaxInt64},
  272. },
  273. {
  274. "int64_max_base2: 0b111111111111111111111111111111111111111111111111111111111111111",
  275. map[string]int64{"int64_max_base2": math.MaxInt64},
  276. },
  277. {
  278. "int64_min: -9223372036854775808",
  279. map[string]int64{"int64_min": math.MinInt64},
  280. },
  281. {
  282. "int64_neg_base2: -0b111111111111111111111111111111111111111111111111111111111111111",
  283. map[string]int64{"int64_neg_base2": -math.MaxInt64},
  284. },
  285. {
  286. "int64_overflow: 9223372036854775808", // math.MaxInt64 + 1
  287. map[string]int64{},
  288. },
  289. // uint
  290. {
  291. "uint_min: 0",
  292. map[string]uint{"uint_min": 0},
  293. },
  294. {
  295. "uint_max: 4294967295",
  296. map[string]uint{"uint_max": math.MaxUint32},
  297. },
  298. {
  299. "uint_underflow: -1",
  300. map[string]uint{},
  301. },
  302. // uint64
  303. {
  304. "uint64_min: 0",
  305. map[string]uint{"uint64_min": 0},
  306. },
  307. {
  308. "uint64_max: 18446744073709551615",
  309. map[string]uint64{"uint64_max": math.MaxUint64},
  310. },
  311. {
  312. "uint64_max_base2: 0b1111111111111111111111111111111111111111111111111111111111111111",
  313. map[string]uint64{"uint64_max_base2": math.MaxUint64},
  314. },
  315. {
  316. "uint64_maxint64: 9223372036854775807",
  317. map[string]uint64{"uint64_maxint64": math.MaxInt64},
  318. },
  319. {
  320. "uint64_underflow: -1",
  321. map[string]uint64{},
  322. },
  323. // float32
  324. {
  325. "float32_max: 3.40282346638528859811704183484516925440e+38",
  326. map[string]float32{"float32_max": math.MaxFloat32},
  327. },
  328. {
  329. "float32_nonzero: 1.401298464324817070923729583289916131280e-45",
  330. map[string]float32{"float32_nonzero": math.SmallestNonzeroFloat32},
  331. },
  332. {
  333. "float32_maxuint64: 18446744073709551615",
  334. map[string]float32{"float32_maxuint64": float32(math.MaxUint64)},
  335. },
  336. {
  337. "float32_maxuint64+1: 18446744073709551616",
  338. map[string]float32{"float32_maxuint64+1": float32(math.MaxUint64 + 1)},
  339. },
  340. // float64
  341. {
  342. "float64_max: 1.797693134862315708145274237317043567981e+308",
  343. map[string]float64{"float64_max": math.MaxFloat64},
  344. },
  345. {
  346. "float64_nonzero: 4.940656458412465441765687928682213723651e-324",
  347. map[string]float64{"float64_nonzero": math.SmallestNonzeroFloat64},
  348. },
  349. {
  350. "float64_maxuint64: 18446744073709551615",
  351. map[string]float64{"float64_maxuint64": float64(math.MaxUint64)},
  352. },
  353. {
  354. "float64_maxuint64+1: 18446744073709551616",
  355. map[string]float64{"float64_maxuint64+1": float64(math.MaxUint64 + 1)},
  356. },
  357. // Overflow cases.
  358. {
  359. "v: 4294967297",
  360. map[string]int32{},
  361. }, {
  362. "v: 128",
  363. map[string]int8{},
  364. },
  365. // Quoted values.
  366. {
  367. "'1': '\"2\"'",
  368. map[interface{}]interface{}{"1": "\"2\""},
  369. }, {
  370. "v:\n- A\n- 'B\n\n C'\n",
  371. map[string][]string{"v": []string{"A", "B\nC"}},
  372. },
  373. // Explicit tags.
  374. {
  375. "v: !!float '1.1'",
  376. map[string]interface{}{"v": 1.1},
  377. }, {
  378. "v: !!null ''",
  379. map[string]interface{}{"v": nil},
  380. }, {
  381. "%TAG !y! tag:yaml.org,2002:\n---\nv: !y!int '1'",
  382. map[string]interface{}{"v": 1},
  383. },
  384. // Anchors and aliases.
  385. {
  386. "a: &x 1\nb: &y 2\nc: *x\nd: *y\n",
  387. &struct{ A, B, C, D int }{1, 2, 1, 2},
  388. }, {
  389. "a: &a {c: 1}\nb: *a",
  390. &struct {
  391. A, B struct {
  392. C int
  393. }
  394. }{struct{ C int }{1}, struct{ C int }{1}},
  395. }, {
  396. "a: &a [1, 2]\nb: *a",
  397. &struct{ B []int }{[]int{1, 2}},
  398. }, {
  399. "b: *a\na: &a {c: 1}",
  400. &struct {
  401. A, B struct {
  402. C int
  403. }
  404. }{struct{ C int }{1}, struct{ C int }{1}},
  405. },
  406. // Bug #1133337
  407. {
  408. "foo: ''",
  409. map[string]*string{"foo": new(string)},
  410. }, {
  411. "foo: null",
  412. map[string]string{"foo": ""},
  413. }, {
  414. "foo: null",
  415. map[string]interface{}{"foo": nil},
  416. },
  417. // Ignored field
  418. {
  419. "a: 1\nb: 2\n",
  420. &struct {
  421. A int
  422. B int "-"
  423. }{1, 0},
  424. },
  425. // Bug #1191981
  426. {
  427. "" +
  428. "%YAML 1.1\n" +
  429. "--- !!str\n" +
  430. `"Generic line break (no glyph)\n\` + "\n" +
  431. ` Generic line break (glyphed)\n\` + "\n" +
  432. ` Line separator\u2028\` + "\n" +
  433. ` Paragraph separator\u2029"` + "\n",
  434. "" +
  435. "Generic line break (no glyph)\n" +
  436. "Generic line break (glyphed)\n" +
  437. "Line separator\u2028Paragraph separator\u2029",
  438. },
  439. // Struct inlining
  440. {
  441. "a: 1\nb: 2\nc: 3\n",
  442. &struct {
  443. A int
  444. C inlineB `yaml:",inline"`
  445. }{1, inlineB{2, inlineC{3}}},
  446. },
  447. // Map inlining
  448. {
  449. "a: 1\nb: 2\nc: 3\n",
  450. &struct {
  451. A int
  452. C map[string]int `yaml:",inline"`
  453. }{1, map[string]int{"b": 2, "c": 3}},
  454. },
  455. // bug 1243827
  456. {
  457. "a: -b_c",
  458. map[string]interface{}{"a": "-b_c"},
  459. },
  460. {
  461. "a: +b_c",
  462. map[string]interface{}{"a": "+b_c"},
  463. },
  464. {
  465. "a: 50cent_of_dollar",
  466. map[string]interface{}{"a": "50cent_of_dollar"},
  467. },
  468. // Duration
  469. {
  470. "a: 3s",
  471. map[string]time.Duration{"a": 3 * time.Second},
  472. },
  473. // Issue #24.
  474. {
  475. "a: <foo>",
  476. map[string]string{"a": "<foo>"},
  477. },
  478. // Base 60 floats are obsolete and unsupported.
  479. {
  480. "a: 1:1\n",
  481. map[string]string{"a": "1:1"},
  482. },
  483. // Binary data.
  484. {
  485. "a: !!binary gIGC\n",
  486. map[string]string{"a": "\x80\x81\x82"},
  487. }, {
  488. "a: !!binary |\n " + strings.Repeat("kJCQ", 17) + "kJ\n CQ\n",
  489. map[string]string{"a": strings.Repeat("\x90", 54)},
  490. }, {
  491. "a: !!binary |\n " + strings.Repeat("A", 70) + "\n ==\n",
  492. map[string]string{"a": strings.Repeat("\x00", 52)},
  493. },
  494. // Ordered maps.
  495. {
  496. "{b: 2, a: 1, d: 4, c: 3, sub: {e: 5}}",
  497. &yaml.MapSlice{{"b", 2}, {"a", 1}, {"d", 4}, {"c", 3}, {"sub", yaml.MapSlice{{"e", 5}}}},
  498. },
  499. // Issue #39.
  500. {
  501. "a:\n b:\n c: d\n",
  502. map[string]struct{ B interface{} }{"a": {map[interface{}]interface{}{"c": "d"}}},
  503. },
  504. // Custom map type.
  505. {
  506. "a: {b: c}",
  507. M{"a": M{"b": "c"}},
  508. },
  509. // Support encoding.TextUnmarshaler.
  510. {
  511. "a: 1.2.3.4\n",
  512. map[string]net.IP{"a": net.IPv4(1, 2, 3, 4)},
  513. },
  514. {
  515. "a: 2015-02-24T18:19:39Z\n",
  516. map[string]time.Time{"a": time.Unix(1424801979, 0).In(time.UTC)},
  517. },
  518. // Encode empty lists as zero-length slices.
  519. {
  520. "a: []",
  521. &struct{ A []int }{[]int{}},
  522. },
  523. // UTF-16-LE
  524. {
  525. "\xff\xfe\xf1\x00o\x00\xf1\x00o\x00:\x00 \x00v\x00e\x00r\x00y\x00 \x00y\x00e\x00s\x00\n\x00",
  526. M{"ñoño": "very yes"},
  527. },
  528. // UTF-16-LE with surrogate.
  529. {
  530. "\xff\xfe\xf1\x00o\x00\xf1\x00o\x00:\x00 \x00v\x00e\x00r\x00y\x00 \x00y\x00e\x00s\x00 \x00=\xd8\xd4\xdf\n\x00",
  531. M{"ñoño": "very yes 🟔"},
  532. },
  533. // UTF-16-BE
  534. {
  535. "\xfe\xff\x00\xf1\x00o\x00\xf1\x00o\x00:\x00 \x00v\x00e\x00r\x00y\x00 \x00y\x00e\x00s\x00\n",
  536. M{"ñoño": "very yes"},
  537. },
  538. // UTF-16-BE with surrogate.
  539. {
  540. "\xfe\xff\x00\xf1\x00o\x00\xf1\x00o\x00:\x00 \x00v\x00e\x00r\x00y\x00 \x00y\x00e\x00s\x00 \xd8=\xdf\xd4\x00\n",
  541. M{"ñoño": "very yes 🟔"},
  542. },
  543. }
  544. type M map[interface{}]interface{}
  545. type inlineB struct {
  546. B int
  547. inlineC `yaml:",inline"`
  548. }
  549. type inlineC struct {
  550. C int
  551. }
  552. func (s *S) TestUnmarshal(c *C) {
  553. for _, item := range unmarshalTests {
  554. t := reflect.ValueOf(item.value).Type()
  555. var value interface{}
  556. switch t.Kind() {
  557. case reflect.Map:
  558. value = reflect.MakeMap(t).Interface()
  559. case reflect.String:
  560. value = reflect.New(t).Interface()
  561. case reflect.Ptr:
  562. value = reflect.New(t.Elem()).Interface()
  563. default:
  564. c.Fatalf("missing case for %s", t)
  565. }
  566. err := yaml.Unmarshal([]byte(item.data), value)
  567. if _, ok := err.(*yaml.TypeError); !ok {
  568. c.Assert(err, IsNil)
  569. }
  570. if t.Kind() == reflect.String {
  571. c.Assert(*value.(*string), Equals, item.value)
  572. } else {
  573. c.Assert(value, DeepEquals, item.value)
  574. }
  575. }
  576. }
  577. func (s *S) TestUnmarshalNaN(c *C) {
  578. value := map[string]interface{}{}
  579. err := yaml.Unmarshal([]byte("notanum: .NaN"), &value)
  580. c.Assert(err, IsNil)
  581. c.Assert(math.IsNaN(value["notanum"].(float64)), Equals, true)
  582. }
  583. var unmarshalErrorTests = []struct {
  584. data, error string
  585. }{
  586. {"v: !!float 'error'", "yaml: cannot decode !!str `error` as a !!float"},
  587. {"v: [A,", "yaml: line 1: did not find expected node content"},
  588. {"v:\n- [A,", "yaml: line 2: did not find expected node content"},
  589. {"a: *b\n", "yaml: unknown anchor 'b' referenced"},
  590. {"a: &a\n b: *a\n", "yaml: anchor 'a' value contains itself"},
  591. {"value: -", "yaml: block sequence entries are not allowed in this context"},
  592. {"a: !!binary ==", "yaml: !!binary value contains invalid base64 data"},
  593. {"{[.]}", `yaml: invalid map key: \[\]interface \{\}\{"\."\}`},
  594. {"{{.}}", `yaml: invalid map key: map\[interface\ \{\}\]interface \{\}\{".":interface \{\}\(nil\)\}`},
  595. }
  596. func (s *S) TestUnmarshalErrors(c *C) {
  597. for _, item := range unmarshalErrorTests {
  598. var value interface{}
  599. err := yaml.Unmarshal([]byte(item.data), &value)
  600. c.Assert(err, ErrorMatches, item.error, Commentf("Partial unmarshal: %#v", value))
  601. }
  602. }
  603. var unmarshalerTests = []struct {
  604. data, tag string
  605. value interface{}
  606. }{
  607. {"_: {hi: there}", "!!map", map[interface{}]interface{}{"hi": "there"}},
  608. {"_: [1,A]", "!!seq", []interface{}{1, "A"}},
  609. {"_: 10", "!!int", 10},
  610. {"_: null", "!!null", nil},
  611. {`_: BAR!`, "!!str", "BAR!"},
  612. {`_: "BAR!"`, "!!str", "BAR!"},
  613. {"_: !!foo 'BAR!'", "!!foo", "BAR!"},
  614. {`_: ""`, "!!str", ""},
  615. }
  616. var unmarshalerResult = map[int]error{}
  617. type unmarshalerType struct {
  618. value interface{}
  619. }
  620. func (o *unmarshalerType) UnmarshalYAML(unmarshal func(v interface{}) error) error {
  621. if err := unmarshal(&o.value); err != nil {
  622. return err
  623. }
  624. if i, ok := o.value.(int); ok {
  625. if result, ok := unmarshalerResult[i]; ok {
  626. return result
  627. }
  628. }
  629. return nil
  630. }
  631. type unmarshalerPointer struct {
  632. Field *unmarshalerType "_"
  633. }
  634. type unmarshalerValue struct {
  635. Field unmarshalerType "_"
  636. }
  637. func (s *S) TestUnmarshalerPointerField(c *C) {
  638. for _, item := range unmarshalerTests {
  639. obj := &unmarshalerPointer{}
  640. err := yaml.Unmarshal([]byte(item.data), obj)
  641. c.Assert(err, IsNil)
  642. if item.value == nil {
  643. c.Assert(obj.Field, IsNil)
  644. } else {
  645. c.Assert(obj.Field, NotNil, Commentf("Pointer not initialized (%#v)", item.value))
  646. c.Assert(obj.Field.value, DeepEquals, item.value)
  647. }
  648. }
  649. }
  650. func (s *S) TestUnmarshalerValueField(c *C) {
  651. for _, item := range unmarshalerTests {
  652. obj := &unmarshalerValue{}
  653. err := yaml.Unmarshal([]byte(item.data), obj)
  654. c.Assert(err, IsNil)
  655. c.Assert(obj.Field, NotNil, Commentf("Pointer not initialized (%#v)", item.value))
  656. c.Assert(obj.Field.value, DeepEquals, item.value)
  657. }
  658. }
  659. func (s *S) TestUnmarshalerWholeDocument(c *C) {
  660. obj := &unmarshalerType{}
  661. err := yaml.Unmarshal([]byte(unmarshalerTests[0].data), obj)
  662. c.Assert(err, IsNil)
  663. value, ok := obj.value.(map[interface{}]interface{})
  664. c.Assert(ok, Equals, true, Commentf("value: %#v", obj.value))
  665. c.Assert(value["_"], DeepEquals, unmarshalerTests[0].value)
  666. }
  667. func (s *S) TestUnmarshalerTypeError(c *C) {
  668. unmarshalerResult[2] = &yaml.TypeError{[]string{"foo"}}
  669. unmarshalerResult[4] = &yaml.TypeError{[]string{"bar"}}
  670. defer func() {
  671. delete(unmarshalerResult, 2)
  672. delete(unmarshalerResult, 4)
  673. }()
  674. type T struct {
  675. Before int
  676. After int
  677. M map[string]*unmarshalerType
  678. }
  679. var v T
  680. data := `{before: A, m: {abc: 1, def: 2, ghi: 3, jkl: 4}, after: B}`
  681. err := yaml.Unmarshal([]byte(data), &v)
  682. c.Assert(err, ErrorMatches, ""+
  683. "yaml: unmarshal errors:\n"+
  684. " line 1: cannot unmarshal !!str `A` into int\n"+
  685. " foo\n"+
  686. " bar\n"+
  687. " line 1: cannot unmarshal !!str `B` into int")
  688. c.Assert(v.M["abc"], NotNil)
  689. c.Assert(v.M["def"], IsNil)
  690. c.Assert(v.M["ghi"], NotNil)
  691. c.Assert(v.M["jkl"], IsNil)
  692. c.Assert(v.M["abc"].value, Equals, 1)
  693. c.Assert(v.M["ghi"].value, Equals, 3)
  694. }
  695. type proxyTypeError struct{}
  696. func (v *proxyTypeError) UnmarshalYAML(unmarshal func(interface{}) error) error {
  697. var s string
  698. var a int32
  699. var b int64
  700. if err := unmarshal(&s); err != nil {
  701. panic(err)
  702. }
  703. if s == "a" {
  704. if err := unmarshal(&b); err == nil {
  705. panic("should have failed")
  706. }
  707. return unmarshal(&a)
  708. }
  709. if err := unmarshal(&a); err == nil {
  710. panic("should have failed")
  711. }
  712. return unmarshal(&b)
  713. }
  714. func (s *S) TestUnmarshalerTypeErrorProxying(c *C) {
  715. type T struct {
  716. Before int
  717. After int
  718. M map[string]*proxyTypeError
  719. }
  720. var v T
  721. data := `{before: A, m: {abc: a, def: b}, after: B}`
  722. err := yaml.Unmarshal([]byte(data), &v)
  723. c.Assert(err, ErrorMatches, ""+
  724. "yaml: unmarshal errors:\n"+
  725. " line 1: cannot unmarshal !!str `A` into int\n"+
  726. " line 1: cannot unmarshal !!str `a` into int32\n"+
  727. " line 1: cannot unmarshal !!str `b` into int64\n"+
  728. " line 1: cannot unmarshal !!str `B` into int")
  729. }
  730. type failingUnmarshaler struct{}
  731. var failingErr = errors.New("failingErr")
  732. func (ft *failingUnmarshaler) UnmarshalYAML(unmarshal func(interface{}) error) error {
  733. return failingErr
  734. }
  735. func (s *S) TestUnmarshalerError(c *C) {
  736. err := yaml.Unmarshal([]byte("a: b"), &failingUnmarshaler{})
  737. c.Assert(err, Equals, failingErr)
  738. }
  739. type sliceUnmarshaler []int
  740. func (su *sliceUnmarshaler) UnmarshalYAML(unmarshal func(interface{}) error) error {
  741. var slice []int
  742. err := unmarshal(&slice)
  743. if err == nil {
  744. *su = slice
  745. return nil
  746. }
  747. var intVal int
  748. err = unmarshal(&intVal)
  749. if err == nil {
  750. *su = []int{intVal}
  751. return nil
  752. }
  753. return err
  754. }
  755. func (s *S) TestUnmarshalerRetry(c *C) {
  756. var su sliceUnmarshaler
  757. err := yaml.Unmarshal([]byte("[1, 2, 3]"), &su)
  758. c.Assert(err, IsNil)
  759. c.Assert(su, DeepEquals, sliceUnmarshaler([]int{1, 2, 3}))
  760. err = yaml.Unmarshal([]byte("1"), &su)
  761. c.Assert(err, IsNil)
  762. c.Assert(su, DeepEquals, sliceUnmarshaler([]int{1}))
  763. }
  764. // From http://yaml.org/type/merge.html
  765. var mergeTests = `
  766. anchors:
  767. list:
  768. - &CENTER { "x": 1, "y": 2 }
  769. - &LEFT { "x": 0, "y": 2 }
  770. - &BIG { "r": 10 }
  771. - &SMALL { "r": 1 }
  772. # All the following maps are equal:
  773. plain:
  774. # Explicit keys
  775. "x": 1
  776. "y": 2
  777. "r": 10
  778. label: center/big
  779. mergeOne:
  780. # Merge one map
  781. << : *CENTER
  782. "r": 10
  783. label: center/big
  784. mergeMultiple:
  785. # Merge multiple maps
  786. << : [ *CENTER, *BIG ]
  787. label: center/big
  788. override:
  789. # Override
  790. << : [ *BIG, *LEFT, *SMALL ]
  791. "x": 1
  792. label: center/big
  793. shortTag:
  794. # Explicit short merge tag
  795. !!merge "<<" : [ *CENTER, *BIG ]
  796. label: center/big
  797. longTag:
  798. # Explicit merge long tag
  799. !<tag:yaml.org,2002:merge> "<<" : [ *CENTER, *BIG ]
  800. label: center/big
  801. inlineMap:
  802. # Inlined map
  803. << : {"x": 1, "y": 2, "r": 10}
  804. label: center/big
  805. inlineSequenceMap:
  806. # Inlined map in sequence
  807. << : [ *CENTER, {"r": 10} ]
  808. label: center/big
  809. `
  810. func (s *S) TestMerge(c *C) {
  811. var want = map[interface{}]interface{}{
  812. "x": 1,
  813. "y": 2,
  814. "r": 10,
  815. "label": "center/big",
  816. }
  817. var m map[interface{}]interface{}
  818. err := yaml.Unmarshal([]byte(mergeTests), &m)
  819. c.Assert(err, IsNil)
  820. for name, test := range m {
  821. if name == "anchors" {
  822. continue
  823. }
  824. c.Assert(test, DeepEquals, want, Commentf("test %q failed", name))
  825. }
  826. }
  827. func (s *S) TestMergeStruct(c *C) {
  828. type Data struct {
  829. X, Y, R int
  830. Label string
  831. }
  832. want := Data{1, 2, 10, "center/big"}
  833. var m map[string]Data
  834. err := yaml.Unmarshal([]byte(mergeTests), &m)
  835. c.Assert(err, IsNil)
  836. for name, test := range m {
  837. if name == "anchors" {
  838. continue
  839. }
  840. c.Assert(test, Equals, want, Commentf("test %q failed", name))
  841. }
  842. }
  843. var unmarshalNullTests = []func() interface{}{
  844. func() interface{} { var v interface{}; v = "v"; return &v },
  845. func() interface{} { var s = "s"; return &s },
  846. func() interface{} { var s = "s"; sptr := &s; return &sptr },
  847. func() interface{} { var i = 1; return &i },
  848. func() interface{} { var i = 1; iptr := &i; return &iptr },
  849. func() interface{} { m := map[string]int{"s": 1}; return &m },
  850. func() interface{} { m := map[string]int{"s": 1}; return m },
  851. }
  852. func (s *S) TestUnmarshalNull(c *C) {
  853. for _, test := range unmarshalNullTests {
  854. item := test()
  855. zero := reflect.Zero(reflect.TypeOf(item).Elem()).Interface()
  856. err := yaml.Unmarshal([]byte("null"), item)
  857. c.Assert(err, IsNil)
  858. if reflect.TypeOf(item).Kind() == reflect.Map {
  859. c.Assert(reflect.ValueOf(item).Interface(), DeepEquals, reflect.MakeMap(reflect.TypeOf(item)).Interface())
  860. } else {
  861. c.Assert(reflect.ValueOf(item).Elem().Interface(), DeepEquals, zero)
  862. }
  863. }
  864. }
  865. func (s *S) TestUnmarshalSliceOnPreset(c *C) {
  866. // Issue #48.
  867. v := struct{ A []int }{[]int{1}}
  868. yaml.Unmarshal([]byte("a: [2]"), &v)
  869. c.Assert(v.A, DeepEquals, []int{2})
  870. }
  871. //var data []byte
  872. //func init() {
  873. // var err error
  874. // data, err = ioutil.ReadFile("/tmp/file.yaml")
  875. // if err != nil {
  876. // panic(err)
  877. // }
  878. //}
  879. //
  880. //func (s *S) BenchmarkUnmarshal(c *C) {
  881. // var err error
  882. // for i := 0; i < c.N; i++ {
  883. // var v map[string]interface{}
  884. // err = yaml.Unmarshal(data, &v)
  885. // }
  886. // if err != nil {
  887. // panic(err)
  888. // }
  889. //}
  890. //
  891. //func (s *S) BenchmarkMarshal(c *C) {
  892. // var v map[string]interface{}
  893. // yaml.Unmarshal(data, &v)
  894. // c.ResetTimer()
  895. // for i := 0; i < c.N; i++ {
  896. // yaml.Marshal(&v)
  897. // }
  898. //}