import json
import unittest

from databuilder.models.table_elasticsearch_document import TableESDocument


class TestTableElasticsearchDocument(unittest.TestCase):

    def test_to_json(self):
        # type: () -> None
        """
        Test string generated from to_json method
        """
        test_obj = TableESDocument(database='test_database',
                                   cluster='test_cluster',
                                   schema_name='test_schema',
                                   name='test_table',
                                   key='test_table_key',
                                   last_updated_epoch=123456789,
                                   description='test_table_description',
                                   column_names=['test_col1', 'test_col2'],
                                   column_descriptions=['test_description1', 'test_description2'],
                                   total_usage=100,
                                   unique_usage=10,
                                   tags=['test'])

        expected_document_dict = {"database": "test_database",
                                  "cluster": "test_cluster",
                                  "schema_name": "test_schema",
                                  "name": "test_table",
                                  "key": "test_table_key",
                                  "last_updated_epoch": 123456789,
                                  "description": "test_table_description",
                                  "column_names": ["test_col1", "test_col2"],
                                  "column_descriptions": ["test_description1", "test_description2"],
                                  "total_usage": 100,
                                  "unique_usage": 10,
                                  "tags": ["test"]
                                  }

        result = test_obj.to_json()
        results = result.split("\n")

        # verify two new line characters in result
        self.assertEqual(len(results), 2, "Result from to_json() function doesn't have a newline!")
        self.assertDictEqual(json.loads(results[0]), expected_document_dict)