pkg/storage/records: add unit test for Get

Signed-off-by: Zhou Hao <zhouhao@cn.fujitsu.com>
(cherry picked from commit 6b1eebd23a)
release-3.1
Zhou Hao 6 years ago committed by Matt Farina
parent afe7058540
commit 1303ae560e
No known key found for this signature in database
GPG Key ID: 9436E80BFBA46909

@ -17,6 +17,7 @@ limitations under the License.
package driver // import "helm.sh/helm/v3/pkg/storage/driver"
import (
"reflect"
"testing"
rspb "helm.sh/helm/v3/pkg/release"
@ -110,3 +111,34 @@ func TestRecordsRemoveAt(t *testing.T) {
t.Fatalf("Expected length of rs to be 1, got %d", len(rs))
}
}
func TestRecordsGet(t *testing.T) {
rs := records([]*record{
newRecord("rls-a.v1", releaseStub("rls-a", 1, "default", rspb.StatusSuperseded)),
newRecord("rls-a.v2", releaseStub("rls-a", 2, "default", rspb.StatusDeployed)),
})
var tests = []struct {
desc string
key string
rec *record
}{
{
"get valid key",
"rls-a.v1",
newRecord("rls-a.v1", releaseStub("rls-a", 1, "default", rspb.StatusSuperseded)),
},
{
"get invalid key",
"rls-a.v3",
nil,
},
}
for _, tt := range tests {
got := rs.Get(tt.key)
if !reflect.DeepEqual(tt.rec, got) {
t.Fatalf("Expected %v, got %v", tt.rec, got)
}
}
}

Loading…
Cancel
Save