package com.mysema.rdfbean.virtuoso;

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

import java.util.Collections;

import org.junit.Before;
import org.junit.Test;

import com.mysema.rdfbean.model.BID;
import com.mysema.rdfbean.model.RDF;
import com.mysema.rdfbean.model.RDFBeanTransaction;
import com.mysema.rdfbean.model.RDFS;
import com.mysema.rdfbean.model.RepositoryException;
import com.mysema.rdfbean.model.STMT;

public class VirtuosoTransactionTest extends AbstractConnectionTest {

    private RDFBeanTransaction tx;

    @Override
    @Before
    public void setUp() {
        super.setUp();
        tx = connection.beginTransaction(false, RDFBeanTransaction.TIMEOUT, RDFBeanTransaction.ISOLATION);
    }

    @Test
    public void Commit() {
        STMT stmt = new STMT(new BID(), RDF.type, RDFS.Class);
        toBeRemoved = Collections.singleton(stmt);
        connection.update(null, Collections.singleton(stmt));
        tx.commit();

        assertExists(stmt.getSubject(), null, null, null);
    }

    @Test
    public void IsActive() {
        assertTrue(tx.isActive());
    }

    @Test
    public void IsRollbackOnly() {
        assertFalse(tx.isRollbackOnly());
        tx.setRollbackOnly();
        assertTrue(tx.isRollbackOnly());
    }

    @Test
    public void Prepare() {
        tx.prepare();
    }

    @Test
    public void Rollback() {
        STMT stmt = new STMT(new BID(), RDF.type, RDFS.Class);
        toBeRemoved = Collections.singleton(stmt);
        connection.update(null, Collections.singleton(stmt));
        tx.rollback();

        assertNotExists(stmt.getSubject(), null, null, null);
    }

    @Test(expected = RepositoryException.class)
    public void SetRollbackOnly() {
        tx.setRollbackOnly();
        tx.commit();
    }

}