package org.cagrid.gme.service.impl;

import org.cagrid.gme.model.XMLSchema;
import org.cagrid.gme.model.XMLSchemaDocument;
import org.cagrid.gme.service.impl.dao.XMLSchemaInformationDao;
import org.cagrid.gme.service.impl.domain.XMLSchemaInformation;
import org.cagrid.gme.service.impl.testutils.GMETestCaseBase;
import org.junit.Test;

import javax.annotation.Resource;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.Collection;
import java.util.HashSet;
import java.util.Set;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;


public class XMLSchemaInformationDaoTestCase extends GMETestCaseBase {

    @Resource
    protected XMLSchemaInformationDao xmlSchemaInformationDao;

    @Test
    public void testWithoutDependencies() throws URISyntaxException {

        XMLSchemaInformation schemaInfo = new XMLSchemaInformation();
        XMLSchema schema = new XMLSchema();
        schemaInfo.setSchema(schema);

        URI namespace = new URI("http://foo");
        schema.setTargetNamespace(namespace);

        XMLSchemaDocument doc = new XMLSchemaDocument();
        doc.setSystemID("foo");
        doc.setSchemaText("my schema");

        XMLSchemaDocument doc2 = new XMLSchemaDocument();
        doc2.setSystemID("foo2");
        doc2.setSchemaText("my schema");

        schema.setRootDocument(doc);
        Set<XMLSchemaDocument> docSet = new HashSet<XMLSchemaDocument>();
        docSet.add(doc2);
        schema.getAdditionalDocuments().addAll(docSet);

        this.xmlSchemaInformationDao.save(schemaInfo);

        Collection<URI> allNamespaces = this.xmlSchemaInformationDao.getAllNamespaces();
        assertEquals(1, allNamespaces.size());
        assertTrue(allNamespaces.contains(namespace));

        XMLSchema schemaForURI = this.xmlSchemaInformationDao.getXMLSchemaByTargetNamespace(namespace);
        assertEquals(schema, schemaForURI);

        Collection<XMLSchemaInformation> dependingSchemas = this.xmlSchemaInformationDao
            .getDependingXMLSchemaInformation(namespace);
        assertEquals(0, dependingSchemas.size());
    }
}