"""

Part of the bsie test suite.
A copy of the license is provided with the project.
Author: Matthias Baumgartner, 2022
"""
# standard imports
import os
import tempfile
import unittest

# external imports
import yaml

# objects to test
from bsie.apps._loader import load_pipeline


## code ##

class TestLoader(unittest.TestCase):
    def test_load_pipeline(self):
        # config file can be empty
        config = {
            'ReaderBuilder': {},
            'ExtractorBuilder': []
            }
        # create config file
        _, path = tempfile.mkstemp(prefix='bsie-test-', suffix='.yaml')
        with open(path, 'wt') as cfile:
            yaml.dump(config, cfile)
        # pipeline contains only default predicates
        pipeline = load_pipeline(path)
        self.assertSetEqual({pred.uri for pred in pipeline.schema.predicates()}, {
            'http://bsfs.ai/schema/Predicate',
            })

        # pipeline is built according to configured extractors
        config = {
            'ReaderBuilder': {},
            'ExtractorBuilder': [
                {'bsie.extractor.preview.Preview': {
                    'max_sides': [50],
                    }},
                {'bsie.extractor.generic.path.Path': {}},
                {'bsie.extractor.generic.constant.Constant': {
                    'schema': '''
                        bse:author rdfs:subClassOf bsfs:Predicate ;
                            rdfs:domain bsfs:Entity ;
                            rdfs:range xsd:string ;
                            bsfs:unique "true"^^xsd:boolean .
                        ''',
                    'tuples': [['http://bsfs.ai/schema/Entity#author', 'Me, myself, and I']],
                  }},
                {'bsie.extractor.image.colors_spatial.ColorsSpatial': {
                    'width': 2,
                    'height': 2,
                    'exp': 2,
                    }},
                ]
            }
        # create config file
        _, path = tempfile.mkstemp(prefix='bsie-test-', suffix='.yaml')
        with open(path, 'wt') as cfile:
            yaml.dump(config, cfile)
        # pipeline contains all defined predicates
        pipeline = load_pipeline(path)
        self.assertSetEqual({pred.uri for pred in pipeline.schema.predicates()}, {
            'http://bsfs.ai/schema/Entity#author',
            'http://bsfs.ai/schema/Predicate',
            'http://bsfs.ai/schema/Entity#filename',
            'http://bsfs.ai/schema/Entity/colors_spatial#0658f2234a054e1dd59a14462c89f7733e019160419c796356aa831498bd0a04',
            'http://bsfs.ai/schema/Entity#preview',
            'http://bsfs.ai/schema/Preview#width',
            'http://bsfs.ai/schema/Preview#height',
            'http://bsfs.ai/schema/Preview#asset',
            })

        # config file must exist
        self.assertRaises(OSError, load_pipeline, 'invalid.yaml')


## main ##

if __name__ == '__main__':
    unittest.main()

## EOF ##