From 9b8f179e154fd8a542af6b4597b8e418ff12d9c9 Mon Sep 17 00:00:00 2001
From: Pavel Zavora <pavel.zavora@bonitoo.io>
Date: Wed, 16 Mar 2022 07:40:37 +0100
Subject: [PATCH] feat(server): update organizations_test with reader role

---
 kv/organizations_test.go | 44 +++++++++++++++++++++++++++++++++++++---
 1 file changed, 41 insertions(+), 3 deletions(-)

diff --git a/kv/organizations_test.go b/kv/organizations_test.go
index a65e2b52a..443ff0a56 100644
--- a/kv/organizations_test.go
+++ b/kv/organizations_test.go
@@ -273,7 +273,7 @@ func TestOrganizationsStore_Update(t *testing.T) {
 			addFirst: true,
 		},
 		{
-			name:   "Update organization default role",
+			name:   "Update organization default role to viewer",
 			fields: fields{},
 			args: args{
 				ctx: context.Background(),
@@ -291,7 +291,25 @@ func TestOrganizationsStore_Update(t *testing.T) {
 			addFirst: true,
 		},
 		{
-			name:   "Update organization name and default role",
+			name:   "Update organization default role to reader",
+			fields: fields{},
+			args: args{
+				ctx: context.Background(),
+				initial: &chronograf.Organization{
+					Name: "The Good Place",
+				},
+				updates: &chronograf.Organization{
+					DefaultRole: roles.ReaderRoleName,
+				},
+			},
+			want: &chronograf.Organization{
+				Name:        "The Good Place",
+				DefaultRole: roles.ReaderRoleName,
+			},
+			addFirst: true,
+		},
+		{
+			name:   "Update organization name and default role to viewer",
 			fields: fields{},
 			args: args{
 				ctx: context.Background(),
@@ -310,6 +328,26 @@ func TestOrganizationsStore_Update(t *testing.T) {
 			},
 			addFirst: true,
 		},
+		{
+			name:   "Update organization name and default role to reader",
+			fields: fields{},
+			args: args{
+				ctx: context.Background(),
+				initial: &chronograf.Organization{
+					Name:        "The Good Place",
+					DefaultRole: roles.AdminRoleName,
+				},
+				updates: &chronograf.Organization{
+					Name:        "The Bad Place",
+					DefaultRole: roles.ReaderRoleName,
+				},
+			},
+			want: &chronograf.Organization{
+				Name:        "The Bad Place",
+				DefaultRole: roles.ReaderRoleName,
+			},
+			addFirst: true,
+		},
 		{
 			name:   "Update organization name, role",
 			fields: fields{},
@@ -405,7 +443,7 @@ func TestOrganizationsStore_Update(t *testing.T) {
 		}
 
 		if tt.addFirst {
-			tt.args.initial, err = s.Add(tt.args.ctx, tt.args.initial)
+			tt.args.initial, _ = s.Add(tt.args.ctx, tt.args.initial)
 		}
 
 		if tt.args.updates.Name != "" {