from __future__ import print_function

from __future__ import unicode_literals

from __future__ import division

from __future__ import absolute_import

from future import standard_library

standard_library.install_aliases()

from builtins import *

import unittest

import os

import sys

sys.path.append('../')

import datetime

try:

    import json

except:

    import simplejson as json

from glob import glob

from whois.parser import WhoisEntry, cast_date

class TestParser(unittest.TestCase):

    def test_com_expiration(self):

        data = """

        Status: ok

        Updated Date: 14-apr-2008

        Creation Date: 14-apr-2008

        Expiration Date: 14-apr-2009

        >>> Last update of whois database: Sun, 31 Aug 2008 00:18:23 UTC <<<

        """

        w = WhoisEntry.load('urlowl.com', data)

        expires = w.expiration_date.strftime('%Y-%m-%d')

        self.assertEqual(expires, '2009-04-14')

    def test_cast_date(self):

        dates = ['14-apr-2008', '2008-04-14']

        for d in dates:

            r = cast_date(d).strftime('%Y-%m-%d')

            self.assertEqual(r, '2008-04-14')

    def test_com_allsamples(self):

        """

        Iterate over all of the sample/whois/*.com files, read the data,

        parse it, and compare to the expected values in sample/expected/.

        Only keys defined in keys_to_test will be tested.

        To generate fresh expected value dumps, see NOTE below.

        """

        keys_to_test = ['domain_name', 'expiration_date', 'updated_date',

                        'creation_date', 'status']

        fail = 0

        total = 0

            # Parse whois data

            domain = os.path.basename(path)

            with open(path) as whois_fp:

                data = whois_fp.read()

            w = WhoisEntry.load(domain, data)

            results = {key: w.get(key) for key in keys_to_test}

            # NOTE: Toggle condition below to write expected results from the

            # parse results This will overwrite the existing expected results.

            # Only do this if you've manually confirmed that the parser is

            # generating correct values at its current state.

            if False:

                def date2str4json(obj):

                    if isinstance(obj, datetime.datetime):

                        return str(obj)

                    raise TypeError(

                            '{} is not JSON serializable'.format(repr(obj)))

                with open(outfile_name, 'w') as outfil:

                    expected_results = json.dump(results, outfil,

                                                       default=date2str4json)

                continue

            # Load expected result

                expected_results = json.load(infil)

            # Compare each key

            for key in results:

                total += 1

                result = results.get(key)

                if isinstance(result, datetime.datetime):

                    result = str(result)

                expected = expected_results.get(key)

                if expected != result:

                    print("%s \t(%s):\t %s != %s" % (domain, key, result, expected))

                    fail += 1

        if fail:

            self.fail("%d/%d sample whois attributes were not parsed properly!"

                      % (fail, total))

    def test_ca_parse(self):

        data = """

        Domain name:           testdomain.ca

        Domain status:         registered

        Creation date:         2000/11/20

        Expiry date:           2020/03/08

        Updated date:          2016/04/29

        DNSSEC:                Unsigned

        Registrar:

            Name:              Webnames.ca Inc.

            Number:            70

        Registrant:

            Name:              Test Industries

        Administrative contact:

            Name:              Test Person1

            Postal address:    Test Address

                               Test City, TestVille

            Phone:             +1.1235434123x123

            Fax:               +1.123434123

            Email:             testperson1@testcompany.ca

        Technical contact:

            Name:              Test Persion2

            Postal address:    Other TestAddress

                               TestTown OCAS Canada

            Phone:             +1.09876545123

            Fax:               +1.12312993873

            Email:             testpersion2@testcompany.ca

        Name servers:

            ns1.testserver1.net

            ns2.testserver2.net

        """

        expected_results = {

            "updated_date": "2016-04-29 00:00:00",

            "registrant_name": [

                "Webnames.ca Inc.",

                "Test Industries",

                "Test Person1",

                "Test Persion2"

            ],

            "fax": [

                "+1.123434123",

                "+1.12312993873"

            ],

            "dnssec": "Unsigned",

            "registrant_number": "70",

            "expiration_date": "2020-03-08 00:00:00",

            "domain_name": "testdomain.ca",

            "creation_date": "2000-11-20 00:00:00",

            "phone": [

                "+1.1235434123x123",

                "+1.09876545123"

            ],

            "domain_status": "registered",

            "emails": [

                "testperson1@testcompany.ca",

                "testpersion2@testcompany.ca"

            ]

        }

        self._parse_and_compare('testcompany.ca', data, expected_results)

    def test_il_parse(self):

        data = """

            query:        python.org.il

            reg-name:     python

            domain:       python.org.il

            descr:        Arik Baratz

            descr:        PO Box 7775 PMB 8452

            descr:        San Francisco, CA

            descr:        94120

            descr:        USA

            phone:        +1 650 6441973

            e-mail:       hostmaster AT arik.baratz.org

            admin-c:      LD-AB16063-IL

            tech-c:       LD-AB16063-IL

            zone-c:       LD-AB16063-IL

            nserver:      dns1.zoneedit.com

            nserver:      dns2.zoneedit.com

            nserver:      dns3.zoneedit.com

            validity:     10-05-2018

            DNSSEC:       unsigned

            status:       Transfer Locked

            changed:      domain-registrar AT isoc.org.il 20050524 (Assigned)

            changed:      domain-registrar AT isoc.org.il 20070520 (Transferred)

            changed:      domain-registrar AT isoc.org.il 20070520 (Changed)

            changed:      domain-registrar AT isoc.org.il 20070520 (Changed)

            changed:      domain-registrar AT isoc.org.il 20070807 (Changed)

            changed:      domain-registrar AT isoc.org.il 20071025 (Changed)

            changed:      domain-registrar AT isoc.org.il 20071025 (Changed)

            changed:      domain-registrar AT isoc.org.il 20081221 (Changed)

            changed:      domain-registrar AT isoc.org.il 20081221 (Changed)

            changed:      domain-registrar AT isoc.org.il 20160301 (Changed)

            changed:      domain-registrar AT isoc.org.il 20160301 (Changed)

            person:       Arik Baratz

            address:      PO Box 7775 PMB 8452

            address:      San Francisco, CA

            address:      94120

            address:      USA

            phone:        +1 650 9635533

            e-mail:       hostmaster AT arik.baratz.org

            nic-hdl:      LD-AB16063-IL

            changed:      Managing Registrar 20070514

            changed:      Managing Registrar 20081002

            changed:      Managing Registrar 20081221

            changed:      Managing Registrar 20081221

            changed:      Managing Registrar 20090502

            registrar name: LiveDns Ltd

            registrar info: http://domains.livedns.co.il

        """

        expected_results = {

            "updated_date": None,

            "registrant_name": "Arik Baratz",

            "fax": None,

            "dnssec": "unsigned",

            "expiration_date": "2018-05-10 00:00:00",

            "domain_name": "python.org.il",

            "creation_date": None,

            "phone": ['+1 650 6441973', '+1 650 9635533'],

            "status": "Transfer Locked",

            "emails": "hostmaster@arik.baratz.org",

            "name_servers": ["dns1.zoneedit.com", "dns2.zoneedit.com", "dns3.zoneedit.com"],

            "registrar": "LiveDns Ltd",

            "referral_url": "http://domains.livedns.co.il"

        }

        self._parse_and_compare('python.org.il', data, expected_results)

    def _parse_and_compare(self, domain_name, data, expected_results):

        results = WhoisEntry.load(domain_name, data)

        fail = 0

        total = 0

        # Compare each key

        for key in expected_results:

            total += 1

            result = results.get(key)

            if isinstance(result, datetime.datetime):

                result = str(result)

            expected = expected_results.get(key)

            if expected != result:

                print("%s \t(%s):\t %s != %s" % (domain_name, key, result, expected))

                fail += 1

        if fail:

            self.fail("%d/%d sample whois attributes were not parsed properly!"

                      % (fail, total))

if __name__ == '__main__':

    unittest.main()