decode_test.go 21 KB

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