# -----------------------------------------------------------------------------

# Copyright (c) 2014--, The Qiita Development Team.

#

# Distributed under the terms of the BSD 3-clause License.

#

# The full license is in the file LICENSE, distributed with this software.

# -----------------------------------------------------------------------------

from unittest import TestCase, main

from qiita_core.util import qiita_test_checker

import qiita_db as qdb

@qiita_test_checker()

class TestOntology(TestCase):

    def setUp(self):

        self.ontology = qdb.ontology.Ontology(999999999)

    def _remove_term(self, term):

        sql = "DELETE FROM qiita.term WHERE ontology_id = %s AND term = %s"

        qdb.sql_connection.perform_as_transaction(

            sql, [self.ontology.id, term])

    def testConvertToID(self):

        self.assertEqual(qdb.util.convert_to_id('ENA', 'ontology'), 999999999)

    def testConvertFromID(self):

        self.assertEqual(

            qdb.util.convert_from_id(999999999, 'ontology'), 'ENA')

    def testShortNameProperty(self):

        self.assertEqual(self.ontology.shortname, 'ENA')

    def testTerms(self):

        obs = self.ontology.terms

        self.assertEqual(

            obs, ['WGS', 'Metagenomics', 'AMPLICON', 'RNA-Seq', 'Other'])

    def test_user_defined_terms(self):

        obs = self.ontology.user_defined_terms

        self.assertEqual(obs, [])

    def test_term_type(self):

        obs = self.ontology.term_type('RNA-Seq')

        self.assertEqual('ontology', obs)

        obs = self.ontology.term_type('Sasquatch')

        self.assertEqual('not_ontology', obs)

        self.ontology.add_user_defined_term('Test Term')

        obs = self.ontology.term_type('Test Term')

        self.assertEqual('user_defined', obs)

        self._remove_term('Test Term')

    def test_add_user_defined_term(self):

        self.assertFalse('Test Term' in self.ontology.user_defined_terms)

        pre = len(self.ontology.user_defined_terms)

        self.ontology.add_user_defined_term('Test Term')

        post = len(self.ontology.user_defined_terms)

        self.assertTrue('Test Term' in self.ontology.user_defined_terms)

        self.assertEqual(post-pre, 1)

        # Clean up the previously added term to avoid test failures

        self._remove_term('Test Term')

    def testContains(self):

        self.assertTrue('Metagenomics' in self.ontology)

        self.assertFalse('NotATerm' in self.ontology)

if __name__ == '__main__':

    main()