/*
 * 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.jackrabbit.oak.jcr;

import java.security.Principal;
import java.util.Iterator;
import java.util.Properties;

import javax.jcr.Credentials;
import javax.jcr.GuestCredentials;
import javax.jcr.Repository;
import javax.jcr.RepositoryException;
import javax.jcr.Session;
import javax.jcr.UnsupportedRepositoryOperationException;

import org.apache.jackrabbit.api.JackrabbitSession;
import org.apache.jackrabbit.api.security.principal.PrincipalManager;
import org.apache.jackrabbit.mk.core.MicroKernelImpl;
import org.apache.jackrabbit.test.NotExecutableException;
import org.apache.jackrabbit.test.RepositoryStub;

public class OakRepositoryStubBase extends RepositoryStub {

    private final Repository repository;

    /**
     * Constructor as required by the JCR TCK.
     * 
     * @param settings repository settings
     */
    public OakRepositoryStubBase(Properties settings) {
        super(settings);

        String dir = "target/mk-tck-" + System.currentTimeMillis();
        Jcr jcr = new Jcr(new MicroKernelImpl(dir));
        preCreateRepository(jcr);
        repository = jcr.createRepository();
    }

    protected void preCreateRepository(Jcr jcr) {
    }

    /**
     * Returns the configured repository instance.
     * 
     * @return the configured repository instance.
     */
    @Override
    public synchronized Repository getRepository() {
        return repository;
    }

    @Override
    public Credentials getReadOnlyCredentials() {
        return new GuestCredentials();
    }

    @Override
    public Principal getKnownPrincipal(Session session) throws RepositoryException {
        if (session instanceof JackrabbitSession) {
            Iterator<Principal> principals = ((JackrabbitSession) session).getPrincipalManager().getPrincipals(PrincipalManager.SEARCH_TYPE_NOT_GROUP);
            if (principals.hasNext()) {
                return principals.next();
            }
        }

        throw new UnsupportedRepositoryOperationException();
    }

    private static final Principal UNKNOWN_PRINCIPAL = new Principal() {
        @Override
        public String getName() {
            return "an_unknown_user";
        }
    };

    @Override
    public Principal getUnknownPrincipal(Session session) throws RepositoryException, NotExecutableException {
        return UNKNOWN_PRINCIPAL;
    }

}

@@ -836,16 +836,65 @@
 k.jcr;%0A%0A
+import java.io.File;%0Aimport java.io.IOException;%0A
 import j
@@ -1346,31 +1346,109 @@
 bit.
-mk.core.MicroKernelImpl
+oak.plugins.segment.SegmentNodeStore;%0Aimport org.apache.jackrabbit.oak.plugins.segment.file.FileStore
 ;%0Aim
@@ -1615,16 +1615,254 @@
 Stub %7B%0A%0A
+    private static final int MAX_FILE_SIZE = 64 * 1024 * 1024;%0A    private static final int CACHE_SIZE = 32 * 1024 * 1024;%0A    private static final boolean MMAP =%0A            System.getProperty(%22sun.arch.data.model%22, %2232%22).equals(%2264%22);%0A%0A
     priv
@@ -2062,16 +2062,55 @@
 ettings)
+%0A            throws RepositoryException
  %7B%0A     
@@ -2142,29 +2142,57 @@
     
-S
 tr
-ing dir = 
+y %7B%0A            File dir = new File(
 %22target
-/
+%22, %22
 mk-t
@@ -2224,16 +2224,17 @@
 Millis()
+)
 ;%0A      
@@ -2235,16 +2235,20 @@
         
+ 
+   
 Jcr jcr 
@@ -2265,31 +2265,105 @@
 new 
-MicroKernelImpl(dir));%0A
+SegmentNodeStore(%0A                    new FileStore(dir, MAX_FILE_SIZE, CACHE_SIZE, MMAP)));%0A    
     
@@ -2400,16 +2400,20 @@
         
+    
 reposito
@@ -2437,24 +2437,114 @@
 pository();%0A
+        %7D catch (IOException e) %7B%0A            throw new RepositoryException(e);%0A        %7D%0A
     %7D%0A%0A    p

Change OakRepositoryStubBase to use NodeStore setup