From 476ffaea8cf3a7f830a2262298f03a991ef61eae Mon Sep 17 00:00:00 2001
From: Dao Cong Tien <tiendc@vn.fujitsu.com>
Date: Wed, 8 Jan 2020 15:43:57 +0700
Subject: [PATCH] Add unit test for List() of pkg/storage/driver/memory.go

Signed-off-by: Dao Cong Tien <tiendc@vn.fujitsu.com>
---
 pkg/storage/driver/memory_test.go | 40 +++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/pkg/storage/driver/memory_test.go b/pkg/storage/driver/memory_test.go
index c74df9432..e9d709c1f 100644
--- a/pkg/storage/driver/memory_test.go
+++ b/pkg/storage/driver/memory_test.go
@@ -81,6 +81,46 @@ func TestMemoryGet(t *testing.T) {
 	}
 }
 
+func TestMemoryList(t *testing.T) {
+	ts := tsFixtureMemory(t)
+
+	// list all deployed releases
+	dpl, err := ts.List(func(rel *rspb.Release) bool {
+		return rel.Info.Status == rspb.StatusDeployed
+	})
+	// check
+	if err != nil {
+		t.Errorf("Failed to list deployed releases: %s", err)
+	}
+	if len(dpl) != 2 {
+		t.Errorf("Expected 2 deployed, got %d", len(dpl))
+	}
+
+	// list all superseded releases
+	ssd, err := ts.List(func(rel *rspb.Release) bool {
+		return rel.Info.Status == rspb.StatusSuperseded
+	})
+	// check
+	if err != nil {
+		t.Errorf("Failed to list superseded releases: %s", err)
+	}
+	if len(ssd) != 6 {
+		t.Errorf("Expected 6 superseded, got %d", len(ssd))
+	}
+
+	// list all deleted releases
+	del, err := ts.List(func(rel *rspb.Release) bool {
+		return rel.Info.Status == rspb.StatusUninstalled
+	})
+	// check
+	if err != nil {
+		t.Errorf("Failed to list deleted releases: %s", err)
+	}
+	if len(del) != 0 {
+		t.Errorf("Expected 0 deleted, got %d", len(del))
+	}
+}
+
 func TestMemoryQuery(t *testing.T) {
 	var tests = []struct {
 		desc string