/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.jena.tdb2.solver;

import static org.junit.Assert.assertEquals ;

import java.util.Iterator ;

import org.apache.jena.atlas.lib.tuple.Tuple ;
import org.apache.jena.system.Txn;
import org.apache.jena.graph.NodeFactory ;
import org.apache.jena.sparql.core.Quad ;
import org.apache.jena.sparql.sse.SSE ;
import org.apache.jena.tdb2.junit.TL;
import org.apache.jena.tdb2.solver.stats.StatsCollectorNodeId;
import org.apache.jena.tdb2.solver.stats.StatsResults;
import org.apache.jena.tdb2.store.DatasetGraphTDB;
import org.apache.jena.tdb2.store.NodeId;
import org.apache.jena.tdb2.store.nodetable.NodeTable;
import org.apache.jena.tdb2.store.nodetupletable.NodeTupleTable;
import org.apache.jena.tdb2.sys.TDBInternal;
import org.junit.Test ;

public class TestStats
{
    static DatasetGraphTDB dsg      = TDBInternal.getDatasetGraphTDB(TL.createTestDatasetGraphMem()) ;
    static NodeTupleTable quads     = dsg.getQuadTable().getNodeTupleTable() ;
    static NodeTupleTable triples   = dsg.getTripleTable().getNodeTupleTable() ;
    static NodeTable nt             = quads.getNodeTable() ;
    
    static Quad q1 = SSE.parseQuad("(<g1> <s> <p> 1)") ; 
    static Quad q2 = SSE.parseQuad("(<g2> <s> <p> 2)") ; 
    static Quad q3 = SSE.parseQuad("(<g2> <s> <p> 9)") ; 
    static Quad q4 = SSE.parseQuad("(_    <s> <p> 1)") ; 
    static {
        Txn.executeWrite(dsg, ()->{
            dsg.add(q1) ;
            dsg.add(q2) ;
            dsg.add(q3) ;
            dsg.add(q4) ;
        });
    }
    
    private StatsResults statsForGraph(NodeId gid) {
        // StatsCollectorNodeId writes nodes for rdf:type (this is not good).
        return Txn.calculateWrite(dsg, ()-> {
            Iterator<Tuple<NodeId>> iter = quads.find(gid, null, null, null) ;

            StatsCollectorNodeId stats = new StatsCollectorNodeId(nt) ;
            for ( ; iter.hasNext() ; ) {
                Tuple<NodeId> t = iter.next() ;
                stats.record(t.get(0), t.get(1), t.get(2), t.get(3)) ;
            }

            return stats.results() ;
        }) ;
    }

    private StatsResults statsForDftGraph() {
        // StatsCollectorNodeId writes nodes for rdf:type (this is not good).
        return Txn.calculateWrite(dsg, ()-> {
            Iterator<Tuple<NodeId>> iter = triples.findAll() ;

            StatsCollectorNodeId stats = new StatsCollectorNodeId(nt) ;
            for ( ; iter.hasNext() ; ) {
                Tuple<NodeId> t = iter.next() ;
                stats.record(null, t.get(0), t.get(1), t.get(2)) ;
            }

            return stats.results() ;
        }) ;
    }

    @Test public void stats_01() {
        StatsResults r = statsForDftGraph() ;
        assertEquals(1, r.getCount()) ; 
        assertEquals(1, r.getPredicates().keySet().size()) ;
    }

    @Test public void stats_02() { 
        NodeId gid = nt.getNodeIdForNode(NodeFactory.createURI("g1")) ;
        StatsResults r = statsForGraph(gid) ;
        assertEquals(1, r.getCount()) ; 
        assertEquals(1, r.getPredicates().keySet().size()) ;
    }
    
    @Test public void stats_03() { 
        NodeId gid = nt.getNodeIdForNode(NodeFactory.createURI("g2")) ;
        StatsResults r = statsForGraph(gid) ;
        assertEquals(2, r.getCount()) ; 
        assertEquals(1, r.getPredicates().keySet().size()) ;
    }
    
    @Test public void stats_04() { 
        StatsResults r = statsForGraph(null) ;
        assertEquals(3, r.getCount()) ; 
        assertEquals(1, r.getPredicates().keySet().size()) ;
    }

}


