#!/usr/bin/env python

from __future__ import print_function

import unittest

from corebio.db  import *
from test_corebio import *

class DBTest(unittest.TestCase):

    def test_dbxref(self):
        ref = Dbxref('/db_xref="GDB:39999"')
        self.assertEqual(ref.database, 'GDB')
        self.assertEqual(ref.identifier, '39999')

        ref = Dbxref("GDB:39999")
        self.assertEqual(ref.database, 'GDB')
        self.assertEqual(ref.identifier, '39999')

        ref = Dbxref('GDB','39999')
        self.assertEqual(ref.database, 'GDB')
        self.assertEqual(ref.identifier, '39999')

        s = repr(ref)
        s2 = str(ref)

    def test_datasource(self) :
        ds = DataSource(
            abbrev='abbrev',
            alt_abbrev=('a','b') ,
            url = 'abc',
            resource_url = 'abc%s',
            parser = None,
            description = 'blah',
            name = 'somename'
            )

    def test_databases(self):
        db = default_registry['pdb']


if __name__ == '__main__':
    print("## Know databases.")
    print()
    print(default_registry)
    print()
    print()

    print("## Running Non-unit tests.")
    print()

    tests = [
        ('embl', 'ab050095'),
        ('pdb', '1hlb'),
        ('swissprot', 'p50105'),
    ]

    for t in tests:
        ref = Dbxref(t[0], t[1])
        print(ref, ref.data_url())

        data = ref.data_stream()
        print(data.readline(), end='')
        print(data.readline(), end='')
        print(data.readline(), end='')
        print(data.readline(), end='')
        print()

    print()
    print()
    print('# Running Unittests')
    unittest.main()