decode_test.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902
  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. // Bug #1133337
  400. {
  401. "foo: ''",
  402. map[string]*string{"foo": new(string)},
  403. }, {
  404. "foo: null",
  405. map[string]string{"foo": ""},
  406. }, {
  407. "foo: null",
  408. map[string]interface{}{"foo": nil},
  409. },
  410. // Ignored field
  411. {
  412. "a: 1\nb: 2\n",
  413. &struct {
  414. A int
  415. B int "-"
  416. }{1, 0},
  417. },
  418. // Bug #1191981
  419. {
  420. "" +
  421. "%YAML 1.1\n" +
  422. "--- !!str\n" +
  423. `"Generic line break (no glyph)\n\` + "\n" +
  424. ` Generic line break (glyphed)\n\` + "\n" +
  425. ` Line separator\u2028\` + "\n" +
  426. ` Paragraph separator\u2029"` + "\n",
  427. "" +
  428. "Generic line break (no glyph)\n" +
  429. "Generic line break (glyphed)\n" +
  430. "Line separator\u2028Paragraph separator\u2029",
  431. },
  432. // Struct inlining
  433. {
  434. "a: 1\nb: 2\nc: 3\n",
  435. &struct {
  436. A int
  437. C inlineB `yaml:",inline"`
  438. }{1, inlineB{2, inlineC{3}}},
  439. },
  440. // bug 1243827
  441. {
  442. "a: -b_c",
  443. map[string]interface{}{"a": "-b_c"},
  444. },
  445. {
  446. "a: +b_c",
  447. map[string]interface{}{"a": "+b_c"},
  448. },
  449. {
  450. "a: 50cent_of_dollar",
  451. map[string]interface{}{"a": "50cent_of_dollar"},
  452. },
  453. // Duration
  454. {
  455. "a: 3s",
  456. map[string]time.Duration{"a": 3 * time.Second},
  457. },
  458. // Issue #24.
  459. {
  460. "a: <foo>",
  461. map[string]string{"a": "<foo>"},
  462. },
  463. // Base 60 floats are obsolete and unsupported.
  464. {
  465. "a: 1:1\n",
  466. map[string]string{"a": "1:1"},
  467. },
  468. // Binary data.
  469. {
  470. "a: !!binary gIGC\n",
  471. map[string]string{"a": "\x80\x81\x82"},
  472. }, {
  473. "a: !!binary |\n " + strings.Repeat("kJCQ", 17) + "kJ\n CQ\n",
  474. map[string]string{"a": strings.Repeat("\x90", 54)},
  475. }, {
  476. "a: !!binary |\n " + strings.Repeat("A", 70) + "\n ==\n",
  477. map[string]string{"a": strings.Repeat("\x00", 52)},
  478. },
  479. // Ordered maps.
  480. {
  481. "{b: 2, a: 1, d: 4, c: 3, sub: {e: 5}}",
  482. &yaml.MapSlice{{"b", 2}, {"a", 1}, {"d", 4}, {"c", 3}, {"sub", yaml.MapSlice{{"e", 5}}}},
  483. },
  484. // Issue #39.
  485. {
  486. "a:\n b:\n c: d\n",
  487. map[string]struct{ B interface{} }{"a": {map[interface{}]interface{}{"c": "d"}}},
  488. },
  489. // Custom map type.
  490. {
  491. "a: {b: c}",
  492. M{"a": M{"b": "c"}},
  493. },
  494. // Support encoding.TextUnmarshaler.
  495. {
  496. "a: 1.2.3.4\n",
  497. map[string]net.IP{"a": net.IPv4(1, 2, 3, 4)},
  498. },
  499. }
  500. type M map[interface{}]interface{}
  501. type inlineB struct {
  502. B int
  503. inlineC `yaml:",inline"`
  504. }
  505. type inlineC struct {
  506. C int
  507. }
  508. func (s *S) TestUnmarshal(c *C) {
  509. for _, item := range unmarshalTests {
  510. t := reflect.ValueOf(item.value).Type()
  511. var value interface{}
  512. switch t.Kind() {
  513. case reflect.Map:
  514. value = reflect.MakeMap(t).Interface()
  515. case reflect.String:
  516. value = reflect.New(t).Interface()
  517. case reflect.Ptr:
  518. value = reflect.New(t.Elem()).Interface()
  519. default:
  520. c.Fatalf("missing case for %s", t)
  521. }
  522. err := yaml.Unmarshal([]byte(item.data), value)
  523. if _, ok := err.(*yaml.TypeError); !ok {
  524. c.Assert(err, IsNil)
  525. }
  526. if t.Kind() == reflect.String {
  527. c.Assert(*value.(*string), Equals, item.value)
  528. } else {
  529. c.Assert(value, DeepEquals, item.value)
  530. }
  531. }
  532. }
  533. func (s *S) TestUnmarshalNaN(c *C) {
  534. value := map[string]interface{}{}
  535. err := yaml.Unmarshal([]byte("notanum: .NaN"), &value)
  536. c.Assert(err, IsNil)
  537. c.Assert(math.IsNaN(value["notanum"].(float64)), Equals, true)
  538. }
  539. var unmarshalErrorTests = []struct {
  540. data, error string
  541. }{
  542. {"v: !!float 'error'", "yaml: cannot decode !!str `error` as a !!float"},
  543. {"v: [A,", "yaml: line 1: did not find expected node content"},
  544. {"v:\n- [A,", "yaml: line 2: did not find expected node content"},
  545. {"a: *b\n", "yaml: unknown anchor 'b' referenced"},
  546. {"a: &a\n b: *a\n", "yaml: anchor 'a' value contains itself"},
  547. {"value: -", "yaml: block sequence entries are not allowed in this context"},
  548. {"a: !!binary ==", "yaml: !!binary value contains invalid base64 data"},
  549. {"{[.]}", `yaml: invalid map key: \[\]interface \{\}\{"\."\}`},
  550. {"{{.}}", `yaml: invalid map key: map\[interface\ \{\}\]interface \{\}\{".":interface \{\}\(nil\)\}`},
  551. }
  552. func (s *S) TestUnmarshalErrors(c *C) {
  553. for _, item := range unmarshalErrorTests {
  554. var value interface{}
  555. err := yaml.Unmarshal([]byte(item.data), &value)
  556. c.Assert(err, ErrorMatches, item.error, Commentf("Partial unmarshal: %#v", value))
  557. }
  558. }
  559. var unmarshalerTests = []struct {
  560. data, tag string
  561. value interface{}
  562. }{
  563. {"_: {hi: there}", "!!map", map[interface{}]interface{}{"hi": "there"}},
  564. {"_: [1,A]", "!!seq", []interface{}{1, "A"}},
  565. {"_: 10", "!!int", 10},
  566. {"_: null", "!!null", nil},
  567. {`_: BAR!`, "!!str", "BAR!"},
  568. {`_: "BAR!"`, "!!str", "BAR!"},
  569. {"_: !!foo 'BAR!'", "!!foo", "BAR!"},
  570. }
  571. var unmarshalerResult = map[int]error{}
  572. type unmarshalerType struct {
  573. value interface{}
  574. }
  575. func (o *unmarshalerType) UnmarshalYAML(unmarshal func(v interface{}) error) error {
  576. if err := unmarshal(&o.value); err != nil {
  577. return err
  578. }
  579. if i, ok := o.value.(int); ok {
  580. if result, ok := unmarshalerResult[i]; ok {
  581. return result
  582. }
  583. }
  584. return nil
  585. }
  586. type unmarshalerPointer struct {
  587. Field *unmarshalerType "_"
  588. }
  589. type unmarshalerValue struct {
  590. Field unmarshalerType "_"
  591. }
  592. func (s *S) TestUnmarshalerPointerField(c *C) {
  593. for _, item := range unmarshalerTests {
  594. obj := &unmarshalerPointer{}
  595. err := yaml.Unmarshal([]byte(item.data), obj)
  596. c.Assert(err, IsNil)
  597. if item.value == nil {
  598. c.Assert(obj.Field, IsNil)
  599. } else {
  600. c.Assert(obj.Field, NotNil, Commentf("Pointer not initialized (%#v)", item.value))
  601. c.Assert(obj.Field.value, DeepEquals, item.value)
  602. }
  603. }
  604. }
  605. func (s *S) TestUnmarshalerValueField(c *C) {
  606. for _, item := range unmarshalerTests {
  607. obj := &unmarshalerValue{}
  608. err := yaml.Unmarshal([]byte(item.data), obj)
  609. c.Assert(err, IsNil)
  610. c.Assert(obj.Field, NotNil, Commentf("Pointer not initialized (%#v)", item.value))
  611. c.Assert(obj.Field.value, DeepEquals, item.value)
  612. }
  613. }
  614. func (s *S) TestUnmarshalerWholeDocument(c *C) {
  615. obj := &unmarshalerType{}
  616. err := yaml.Unmarshal([]byte(unmarshalerTests[0].data), obj)
  617. c.Assert(err, IsNil)
  618. value, ok := obj.value.(map[interface{}]interface{})
  619. c.Assert(ok, Equals, true, Commentf("value: %#v", obj.value))
  620. c.Assert(value["_"], DeepEquals, unmarshalerTests[0].value)
  621. }
  622. func (s *S) TestUnmarshalerTypeError(c *C) {
  623. unmarshalerResult[2] = &yaml.TypeError{[]string{"foo"}}
  624. unmarshalerResult[4] = &yaml.TypeError{[]string{"bar"}}
  625. defer func() {
  626. delete(unmarshalerResult, 2)
  627. delete(unmarshalerResult, 4)
  628. }()
  629. type T struct {
  630. Before int
  631. After int
  632. M map[string]*unmarshalerType
  633. }
  634. var v T
  635. data := `{before: A, m: {abc: 1, def: 2, ghi: 3, jkl: 4}, after: B}`
  636. err := yaml.Unmarshal([]byte(data), &v)
  637. c.Assert(err, ErrorMatches, ""+
  638. "yaml: unmarshal errors:\n"+
  639. " line 1: cannot unmarshal !!str `A` into int\n"+
  640. " foo\n"+
  641. " bar\n"+
  642. " line 1: cannot unmarshal !!str `B` into int")
  643. c.Assert(v.M["abc"], NotNil)
  644. c.Assert(v.M["def"], IsNil)
  645. c.Assert(v.M["ghi"], NotNil)
  646. c.Assert(v.M["jkl"], IsNil)
  647. c.Assert(v.M["abc"].value, Equals, 1)
  648. c.Assert(v.M["ghi"].value, Equals, 3)
  649. }
  650. type proxyTypeError struct{}
  651. func (v *proxyTypeError) UnmarshalYAML(unmarshal func(interface{}) error) error {
  652. var s string
  653. var a int32
  654. var b int64
  655. if err := unmarshal(&s); err != nil {
  656. panic(err)
  657. }
  658. if s == "a" {
  659. if err := unmarshal(&b); err == nil {
  660. panic("should have failed")
  661. }
  662. return unmarshal(&a)
  663. }
  664. if err := unmarshal(&a); err == nil {
  665. panic("should have failed")
  666. }
  667. return unmarshal(&b)
  668. }
  669. func (s *S) TestUnmarshalerTypeErrorProxying(c *C) {
  670. type T struct {
  671. Before int
  672. After int
  673. M map[string]*proxyTypeError
  674. }
  675. var v T
  676. data := `{before: A, m: {abc: a, def: b}, after: B}`
  677. err := yaml.Unmarshal([]byte(data), &v)
  678. c.Assert(err, ErrorMatches, ""+
  679. "yaml: unmarshal errors:\n"+
  680. " line 1: cannot unmarshal !!str `A` into int\n"+
  681. " line 1: cannot unmarshal !!str `a` into int32\n"+
  682. " line 1: cannot unmarshal !!str `b` into int64\n"+
  683. " line 1: cannot unmarshal !!str `B` into int")
  684. }
  685. type failingUnmarshaler struct{}
  686. var failingErr = errors.New("failingErr")
  687. func (ft *failingUnmarshaler) UnmarshalYAML(unmarshal func(interface{}) error) error {
  688. return failingErr
  689. }
  690. func (s *S) TestUnmarshalerError(c *C) {
  691. err := yaml.Unmarshal([]byte("a: b"), &failingUnmarshaler{})
  692. c.Assert(err, Equals, failingErr)
  693. }
  694. // From http://yaml.org/type/merge.html
  695. var mergeTests = `
  696. anchors:
  697. list:
  698. - &CENTER { "x": 1, "y": 2 }
  699. - &LEFT { "x": 0, "y": 2 }
  700. - &BIG { "r": 10 }
  701. - &SMALL { "r": 1 }
  702. # All the following maps are equal:
  703. plain:
  704. # Explicit keys
  705. "x": 1
  706. "y": 2
  707. "r": 10
  708. label: center/big
  709. mergeOne:
  710. # Merge one map
  711. << : *CENTER
  712. "r": 10
  713. label: center/big
  714. mergeMultiple:
  715. # Merge multiple maps
  716. << : [ *CENTER, *BIG ]
  717. label: center/big
  718. override:
  719. # Override
  720. << : [ *BIG, *LEFT, *SMALL ]
  721. "x": 1
  722. label: center/big
  723. shortTag:
  724. # Explicit short merge tag
  725. !!merge "<<" : [ *CENTER, *BIG ]
  726. label: center/big
  727. longTag:
  728. # Explicit merge long tag
  729. !<tag:yaml.org,2002:merge> "<<" : [ *CENTER, *BIG ]
  730. label: center/big
  731. inlineMap:
  732. # Inlined map
  733. << : {"x": 1, "y": 2, "r": 10}
  734. label: center/big
  735. inlineSequenceMap:
  736. # Inlined map in sequence
  737. << : [ *CENTER, {"r": 10} ]
  738. label: center/big
  739. `
  740. func (s *S) TestMerge(c *C) {
  741. var want = map[interface{}]interface{}{
  742. "x": 1,
  743. "y": 2,
  744. "r": 10,
  745. "label": "center/big",
  746. }
  747. var m map[interface{}]interface{}
  748. err := yaml.Unmarshal([]byte(mergeTests), &m)
  749. c.Assert(err, IsNil)
  750. for name, test := range m {
  751. if name == "anchors" {
  752. continue
  753. }
  754. c.Assert(test, DeepEquals, want, Commentf("test %q failed", name))
  755. }
  756. }
  757. func (s *S) TestMergeStruct(c *C) {
  758. type Data struct {
  759. X, Y, R int
  760. Label string
  761. }
  762. want := Data{1, 2, 10, "center/big"}
  763. var m map[string]Data
  764. err := yaml.Unmarshal([]byte(mergeTests), &m)
  765. c.Assert(err, IsNil)
  766. for name, test := range m {
  767. if name == "anchors" {
  768. continue
  769. }
  770. c.Assert(test, Equals, want, Commentf("test %q failed", name))
  771. }
  772. }
  773. var unmarshalNullTests = []func() interface{}{
  774. func() interface{} { var v interface{}; v = "v"; return &v },
  775. func() interface{} { var s = "s"; return &s },
  776. func() interface{} { var s = "s"; sptr := &s; return &sptr },
  777. func() interface{} { var i = 1; return &i },
  778. func() interface{} { var i = 1; iptr := &i; return &iptr },
  779. func() interface{} { m := map[string]int{"s": 1}; return &m },
  780. func() interface{} { m := map[string]int{"s": 1}; return m },
  781. }
  782. func (s *S) TestUnmarshalNull(c *C) {
  783. for _, test := range unmarshalNullTests {
  784. item := test()
  785. zero := reflect.Zero(reflect.TypeOf(item).Elem()).Interface()
  786. err := yaml.Unmarshal([]byte("null"), item)
  787. c.Assert(err, IsNil)
  788. if reflect.TypeOf(item).Kind() == reflect.Map {
  789. c.Assert(reflect.ValueOf(item).Interface(), DeepEquals, reflect.MakeMap(reflect.TypeOf(item)).Interface())
  790. } else {
  791. c.Assert(reflect.ValueOf(item).Elem().Interface(), DeepEquals, zero)
  792. }
  793. }
  794. }
  795. //var data []byte
  796. //func init() {
  797. // var err error
  798. // data, err = ioutil.ReadFile("/tmp/file.yaml")
  799. // if err != nil {
  800. // panic(err)
  801. // }
  802. //}
  803. //
  804. //func (s *S) BenchmarkUnmarshal(c *C) {
  805. // var err error
  806. // for i := 0; i < c.N; i++ {
  807. // var v map[string]interface{}
  808. // err = yaml.Unmarshal(data, &v)
  809. // }
  810. // if err != nil {
  811. // panic(err)
  812. // }
  813. //}
  814. //
  815. //func (s *S) BenchmarkMarshal(c *C) {
  816. // var v map[string]interface{}
  817. // yaml.Unmarshal(data, &v)
  818. // c.ResetTimer()
  819. // for i := 0; i < c.N; i++ {
  820. // yaml.Marshal(&v)
  821. // }
  822. //}