decode_test.go 22 KB

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