# -*- coding: utf-8 -*-
#
# Copyright 2012-2015 Spotify AB
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

from helpers import unittest, in_parse

import luigi
import luigi.interface
import json
import collections


class DictParameterTask(luigi.Task):
    param = luigi.DictParameter()


class DictParameterTest(unittest.TestCase):

    _dict = collections.OrderedDict([('username', 'me'), ('password', 'secret')])

    def test_parse(self):
        d = luigi.DictParameter().parse(json.dumps(DictParameterTest._dict))
        self.assertEqual(d, DictParameterTest._dict)

    def test_serialize(self):
        d = luigi.DictParameter().serialize(DictParameterTest._dict)
        self.assertEqual(d, '{"username": "me", "password": "secret"}')

    def test_parse_and_serialize(self):
        inputs = ['{"username": "me", "password": "secret"}', '{"password": "secret", "username": "me"}']
        for json_input in inputs:
            _dict = luigi.DictParameter().parse(json_input)
            self.assertEqual(json_input, luigi.DictParameter().serialize(_dict))

    def test_parse_interface(self):
        in_parse(["DictParameterTask", "--param", '{"username": "me", "password": "secret"}'],
                 lambda task: self.assertEqual(task.param, DictParameterTest._dict))

    def test_serialize_task(self):
        t = DictParameterTask(DictParameterTest._dict)
        self.assertEqual(str(t), 'DictParameterTask(param={"username": "me", "password": "secret"})')

    def test_parse_invalid_input(self):
        self.assertRaises(ValueError, lambda: luigi.DictParameter().parse('{"invalid"}'))

    def test_hash_normalize(self):
        self.assertRaises(TypeError, lambda: hash(luigi.DictParameter().parse('{"a": {"b": []}}')))
        a = luigi.DictParameter().normalize({"a": [{"b": []}]})
        b = luigi.DictParameter().normalize({"a": [{"b": []}]})
        self.assertEqual(hash(a), hash(b))

[((27, 12, 27, 33), 'luigi.DictParameter', 'luigi.DictParameter', ({}, {}), '()', False, 'import luigi\n'), ((32, 12, 32, 81), 'collections.OrderedDict', 'collections.OrderedDict', ({(32, 36, 32, 80): "[('username', 'me'), ('password', 'secret')]"}, {}), "([('username', 'me'), ('password', 'secret')])", False, 'import collections\n'), ((35, 40, 35, 75), 'json.dumps', 'json.dumps', ({(35, 51, 35, 74): 'DictParameterTest._dict'}, {}), '(DictParameterTest._dict)', False, 'import json\n'), ((35, 12, 35, 33), 'luigi.DictParameter', 'luigi.DictParameter', ({}, {}), '()', False, 'import luigi\n'), ((39, 12, 39, 33), 'luigi.DictParameter', 'luigi.DictParameter', ({}, {}), '()', False, 'import luigi\n'), ((61, 12, 61, 33), 'luigi.DictParameter', 'luigi.DictParameter', ({}, {}), '()', False, 'import luigi\n'), ((62, 12, 62, 33), 'luigi.DictParameter', 'luigi.DictParameter', ({}, {}), '()', False, 'import luigi\n'), ((45, 20, 45, 41), 'luigi.DictParameter', 'luigi.DictParameter', ({}, {}), '()', False, 'import luigi\n'), ((46, 41, 46, 62), 'luigi.DictParameter', 'luigi.DictParameter', ({}, {}), '()', False, 'import luigi\n'), ((57, 46, 57, 67), 'luigi.DictParameter', 'luigi.DictParameter', ({}, {}), '()', False, 'import luigi\n'), ((60, 50, 60, 71), 'luigi.DictParameter', 'luigi.DictParameter', ({}, {}), '()', False, 'import luigi\n')]