private Set<String> iterate() throws Exception { Iterator<String> cur = blobStore.getAllChunkIds(0); Set<String> existing = Sets.newHashSet(); while (cur.hasNext()) { existing.add(cur.next()); } return existing; }
@Test public void testGetAllChunkIds() throws Exception { File rootFolder = folder.newFolder(); dataStore = getBlobStore(rootFolder); int number = 10; Set<String> added = newHashSet(); for (int i = 0; i < number; i++) { String rec = dataStore.writeBlob(randomStream(i, 16516)); added.add(rec); } Set<String> retrieved = newHashSet(dataStore.getAllChunkIds(0)); assertEquals(added, retrieved); }
@Test public void testGetAllChunks() throws Exception{ DataIdentifier d10 = new DataIdentifier("d-10"); DataIdentifier d20 = new DataIdentifier("d-20"); DataIdentifier d30 = new DataIdentifier("d-30"); List<DataIdentifier> dis = ImmutableList.of(d10, d20, d30); List<DataRecord> recs = Lists.newArrayList( Iterables.transform(dis, new Function<DataIdentifier, DataRecord>() { @Nullable @Override public DataRecord apply(@Nullable DataIdentifier input) { return new TimeDataRecord(input); } })); OakFileDataStore mockedDS = mock(OakFileDataStore.class); when(mockedDS.getAllRecords()).thenReturn(recs.iterator()); when(mockedDS.getRecord(new DataIdentifier("d-10"))).thenReturn(new TimeDataRecord(d10)); when(mockedDS.getRecord(new DataIdentifier("d-20"))).thenReturn(new TimeDataRecord(d20)); when(mockedDS.getRecord(new DataIdentifier("d-30"))).thenReturn(new TimeDataRecord(d30)); DataStoreBlobStore ds = new DataStoreBlobStore(mockedDS); Iterator<String> chunks = ds.getAllChunkIds(25); Set<String> expected = Sets.newHashSet("d-10","d-20"); assertEquals(expected, Sets.newHashSet(chunks)); }