Add Index.Packs() and IDSet.Equals()

repository/index.go

@@ -205,6 +205,19 @@ func (idx *Index) Each(done chan struct{}) <-chan PackedBlob {
 	return ch
 }
 
+// Packs returns all packs in this index
+func (idx *Index) Packs() backend.IDSet {
+	idx.m.Lock()
+	defer idx.m.Unlock()
+
+	packs := backend.NewIDSet()
+	for _, entry := range idx.pack {
+		packs.Insert(entry.packID)
+	}
+
+	return packs
+}
+
 // Count returns the number of blobs of type t in the index.
 func (idx *Index) Count(t pack.BlobType) (n uint) {
 	debug.Log("Index.Count", "counting blobs of type %v", t)

repository/index_test.go

@@ -325,3 +325,18 @@ func TestConvertIndex(t *testing.T) {
 		}
 	})
 }
+
+func TestIndexPacks(t *testing.T) {
+	idx := repository.NewIndex()
+	packs := backend.NewIDSet()
+
+	for i := 0; i < 20; i++ {
+		packID := randomID()
+		idx.Store(pack.Data, randomID(), packID, 0, 23)
+
+		packs.Insert(packID)
+	}
+
+	idxPacks := idx.Packs()
+	Assert(t, packs.Equals(idxPacks), "packs in index do not match packs added to index")
+}