From 97ba8771bc3df14f975831e024730de16a4c83f2 Mon Sep 17 00:00:00 2001
From: Matt Jaffee <matthew.jaffee@gmail.com>
Date: Fri, 15 Mar 2019 14:13:49 -0500
Subject: [PATCH] add test for only opening owned shards

---
 server_internal_test.go | 58 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 58 insertions(+)

diff --git a/server_internal_test.go b/server_internal_test.go
index a8af8186c..86688ebbe 100644
--- a/server_internal_test.go
+++ b/server_internal_test.go
@@ -16,9 +16,12 @@ package pilosa
 
 import (
 	"io/ioutil"
+	"os"
 	"runtime"
 	"testing"
 	"time"
+
+	"github.com/pilosa/pilosa/roaring"
 )
 
 // Ensure the file handle count is working
@@ -60,3 +63,58 @@ func TestMonitorAntiEntropyZero(t *testing.T) {
 		t.Fatalf("monitorAntiEntropy should have returned immediately with duration 0")
 	}
 }
+
+func TestOnlyOpenOwnedFiles(t *testing.T) {
+	path, err := ioutil.TempDir("", "pilosa")
+	if err != nil {
+		t.Fatalf("getting temp dir: %v", err)
+	}
+	defer func() {
+		err := os.RemoveAll(path)
+		if err != nil {
+			t.Logf("cleaning up temp dir: %v", err)
+		}
+	}()
+	bm := roaring.NewFileBitmap(1, 2, 3)
+	err = os.MkdirAll(path+"/i/f/views/standard/fragments", os.ModeDir|os.ModePerm)
+	if err != nil {
+		t.Fatalf("mkdirall: %v", err)
+	}
+	one, err := os.Create(path + "/i/f/views/standard/fragments/1")
+	if err != nil {
+		t.Fatalf("creating one: %v", err)
+	}
+	two, err := os.Create(path + "/i/f/views/standard/fragments/2")
+	if err != nil {
+		t.Fatalf("creating two: %v", err)
+	}
+	_, err = bm.WriteTo(one)
+	if err != nil {
+		t.Fatalf("writing to one: %v", err)
+	}
+	_, err = bm.WriteTo(two)
+	if err != nil {
+		t.Fatalf("writing to two: %v", err)
+	}
+
+	h := NewHolder()
+	h.Path = path
+	h.shardValidatorFunc = func(index string, shard uint64) bool {
+		return shard == 1
+	}
+
+	err = h.Open()
+	if err != nil {
+		t.Fatalf("opening holder: %v", err)
+	}
+
+	view := h.Index("i").Field("f").view("standard")
+
+	if len(view.fragments) != 1 {
+		t.Errorf("should have one fragment, but have: %d", len(view.fragments))
+	}
+
+	if _, ok := view.fragments[1]; !ok {
+		t.Errorf("should have fragment 1, but fragments: %#v", view.fragments)
+	}
+}