# -*- 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))

[((762, 783), 'luigi.DictParameter', 'luigi.DictParameter', ([], {}), '()\n', (781, 783), False, 'import luigi\n'), ((843, 912), 'collections.OrderedDict', 'collections.OrderedDict', (["[('username', 'me'), ('password', 'secret')]"], {}), "([('username', 'me'), ('password', 'secret')])\n", (866, 912), False, 'import collections\n'), ((980, 1015), 'json.dumps', 'json.dumps', (['DictParameterTest._dict'], {}), '(DictParameterTest._dict)\n', (990, 1015), False, 'import json\n'), ((952, 973), 'luigi.DictParameter', 'luigi.DictParameter', ([], {}), '()\n', (971, 973), False, 'import luigi\n'), ((1113, 1134), 'luigi.DictParameter', 'luigi.DictParameter', ([], {}), '()\n', (1132, 1134), False, 'import luigi\n'), ((2253, 2274), 'luigi.DictParameter', 'luigi.DictParameter', ([], {}), '()\n', (2272, 2274), False, 'import luigi\n'), ((2317, 2338), 'luigi.DictParameter', 'luigi.DictParameter', ([], {}), '()\n', (2336, 2338), False, 'import luigi\n'), ((1443, 1464), 'luigi.DictParameter', 'luigi.DictParameter', ([], {}), '()\n', (1462, 1464), False, 'import luigi\n'), ((1524, 1545), 'luigi.DictParameter', 'luigi.DictParameter', ([], {}), '()\n', (1543, 1545), False, 'import luigi\n'), ((2061, 2082), 'luigi.DictParameter', 'luigi.DictParameter', ([], {}), '()\n', (2080, 2082), False, 'import luigi\n'), ((2191, 2212), 'luigi.DictParameter', 'luigi.DictParameter', ([], {}), '()\n', (2210, 2212), False, 'import luigi\n')]