From d1864a4cdc09e430b6d085969de88a5acc7918d9 Mon Sep 17 00:00:00 2001
From: Philip O'Toole <philip@influxdb.com>
Date: Fri, 5 Jun 2015 12:28:20 -0700
Subject: [PATCH] Unit test expired and deleted shard groups

---
 meta/data_test.go | 45 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)

diff --git a/meta/data_test.go b/meta/data_test.go
index 4998b0a32c..0e6da6640b 100644
--- a/meta/data_test.go
+++ b/meta/data_test.go
@@ -297,6 +297,51 @@ func TestData_CreateShardGroup(t *testing.T) {
 	}
 }
 
+// Ensure that a shard group is correctly detected as expired.
+func TestData_ShardGroupExpiredDeleted(t *testing.T) {
+	var data meta.Data
+	if err := data.CreateNode("node0"); err != nil {
+		t.Fatal(err)
+	} else if err = data.CreateNode("node1"); err != nil {
+		t.Fatal(err)
+	} else if err = data.CreateDatabase("db0"); err != nil {
+		t.Fatal(err)
+	} else if err = data.CreateRetentionPolicy("db0", &meta.RetentionPolicyInfo{Name: "rp0", Duration: 1 * time.Hour}); err != nil {
+		t.Fatal(err)
+	}
+
+	// Create shard groups.
+	if err := data.CreateShardGroup("db0", "rp0", time.Date(2000, time.January, 1, 0, 0, 0, 0, time.UTC)); err != nil {
+		t.Fatal(err)
+	}
+	if err := data.CreateShardGroup("db0", "rp0", time.Date(2001, time.January, 1, 0, 0, 0, 0, time.UTC)); err != nil {
+		t.Fatal(err)
+	}
+
+	// Check expiration.
+	rp, _ := data.RetentionPolicy("db0", "rp0")
+	groups := rp.ExpiredShardGroups(time.Date(2001, time.January, 1, 0, 0, 0, 0, time.UTC))
+	if len(groups) != 1 {
+		t.Fatalf("wrong number of expired shard groups returned, got %d, exp 1", len(groups))
+	}
+	if groups[0].StartTime != time.Date(2000, time.January, 1, 0, 0, 0, 0, time.UTC) {
+		t.Fatal("wrong shard group marked as expired")
+	}
+
+	// Check deletion.
+	if err := data.DeleteShardGroup("db0", "rp0", groups[0].ID); err != nil {
+		t.Fatal(err)
+	}
+	groups = rp.DeletedShardGroups()
+	if len(groups) != 1 {
+		t.Fatalf("wrong number of deleted shard groups returned, got %d, exp 1", len(groups))
+	}
+	if groups[0].StartTime != time.Date(2000, time.January, 1, 0, 0, 0, 0, time.UTC) {
+		t.Fatal("wrong shard group marked as expired")
+	}
+
+}
+
 // Test shard group selection.
 func TestShardGroup_Overlaps(t *testing.T) {
 	// Make a shard group 1 hour in duration