decode_test.go 20 KB

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