private NeoStores CreateNeoStores() { NeoStores neoStores = mock(typeof(NeoStores)); LabelTokenStore labelTokenStore = mock(typeof(LabelTokenStore)); RelationshipTypeTokenStore typeTokenStore = mock(typeof(RelationshipTypeTokenStore)); PropertyKeyTokenStore propertyKeyTokenStore = mock(typeof(PropertyKeyTokenStore)); when(labelTokenStore.Tokens).thenReturn(LabelTokens); when(typeTokenStore.Tokens).thenReturn(TypeTokes); when(propertyKeyTokenStore.Tokens).thenReturn(PropertyTokens); when(neoStores.LabelTokenStore).thenReturn(labelTokenStore); when(neoStores.RelationshipTypeTokenStore).thenReturn(typeTokenStore); when(neoStores.PropertyKeyTokenStore).thenReturn(propertyKeyTokenStore); return(neoStores); }
private GraphDatabaseAPI DatabaseWithManyPropertyKeys(int propertyKeyCount) { PageCache pageCache = _pageCacheRule.getPageCache(_fileSystemRule.get()); StoreFactory storeFactory = new StoreFactory(_testDirectory.databaseLayout(), Config.defaults(), new DefaultIdGeneratorFactory(_fileSystemRule.get()), pageCache, _fileSystemRule.get(), NullLogProvider.Instance, EmptyVersionContextSupplier.EMPTY); NeoStores neoStores = storeFactory.OpenAllNeoStores(true); PropertyKeyTokenStore store = neoStores.PropertyKeyTokenStore; for (int i = 0; i < propertyKeyCount; i++) { PropertyKeyTokenRecord record = new PropertyKeyTokenRecord(( int )store.nextId()); record.InUse = true; ICollection <DynamicRecord> nameRecords = store.AllocateNameRecords(PropertyStore.encodeString(Key(i))); record.AddNameRecords(nameRecords); record.NameId = ( int )Iterables.first(nameRecords).Id; store.UpdateRecord(record); } neoStores.Close(); return(Database()); }