# -*- coding: utf-8 -*-
from __future__ import absolute_import, print_function, unicode_literals
import re
import unittest
import yamlish
import yaml

OUT = [
  "---",
  "bill-to:",
  "  address:",
  "    city: 'Royal Oak'",
  "    lines: \"458 Walkman Dr.\\nSuite #292\\n\"",
  "    postal: 48046",
  "    state: MI",
  "  family: Dumars",
  "  given: Chris",
  "comments: \"Late afternoon is best. Backup contact is Nancy Billsmer \@ 338-4338\\n\"",
  "date: 2001-01-23",
  "invoice: 34843",
  "product:",
  "  -",
  "    description: Basketball",
  "    price: 450.00",
  "    quantity: 4",
  "    sku: BL394D",
  "  -",
  "    description: 'Super Hoop'",
  "    price: 2392.00",
  "    quantity: 1",
  "    sku: BL4438H",
  "tax: 251.42",
  "total: 4443.52",
  "...",
]

IN = {
  'bill-to': {
    'given': 'Chris',
    'address': {
      'city': 'Royal Oak',
      'postal': '48046',
      'lines': "458 Walkman Dr.\nSuite #292\n",
      'state': 'MI'
    },
    'family': 'Dumars'
  },
  'invoice': '34843',
  'date': '2001-01-23',
  'tax': '251.42',
  'product': [
    {
      'sku': 'BL394D',
      'quantity': '4',
      'price': '450.00',
      'description': 'Basketball'
    },
    {
      'sku': 'BL4438H',
      'quantity': '1',
      'price': '2392.00',
      'description': 'Super Hoop'
    }
  ],
  'comments': "Late afternoon is best. Backup contact is Nancy Billsmer @ 338-4338\n",
  'total': '4443.52'
}

buf1 = []
buf2 = []
buf3 = ""

destination = [
  {
    "name": 'Array reference',
    "destination": buf1,
    "normalise": (lambda : buf1),
  },
#  {
#    "name": 'Closure',
#    "destination": sub { push @buf2, shift },
#    "normalise": sub { return \@buf2 },
#  },
  {
    "name": 'Scalar',
    "destination": buf3,
    "normalise": (lambda : re.split(r" \n ", buf3))
  }
]

class TestOuptut(unittest.TestCase):
    def test_output(self):
        for dest in destination:
            name = dest['name']

            yaml.dump(IN, dest)
            got = dest['normalise']()

            self.assertEqual(got, OUT, """%s: Result matches
              expected = %s
              
              observed = %s
              """ % (name, OUT, got))