add more tests

command/acl/token/update/token_update_test.go

@@ -52,7 +52,19 @@ func TestTokenUpdateCommand(t *testing.T) {
 
 	// create a token
 	token, _, err := client.ACL().TokenCreate(
-		&api.ACLToken{Description: "test"},
+		&api.ACLToken{Description: "test",
+			NodeIdentities: []*api.ACLNodeIdentity{
+				{
+					NodeName:   "first-node",
+					Datacenter: "middleearth-southwest",
+				},
+			},
+			ServiceIdentities: []*api.ACLServiceIdentity{
+				{
+					ServiceName: "fake-service",
+				},
+			},
+		},
 		&api.WriteOptions{Token: "root"},
 	)
 	require.NoError(t, err)
@@ -135,6 +147,22 @@ func TestTokenUpdateCommand(t *testing.T) {
 		require.Len(t, token.Policies, 1)
 	})
 
+	// update with service-identity
+	t.Run("service-identity", func(t *testing.T) {
+		require.Len(t, token.ServiceIdentities, 1)
+
+		token := run(t, []string{
+			"-http-addr=" + a.HTTPAddr(),
+			"-accessor-id=" + token.AccessorID,
+			"-token=root",
+			"-service-identity=service:datapalace",
+			"-description=test token",
+		})
+
+		require.Len(t, token.ServiceIdentities, 1)
+		require.Equal(t, "service", token.ServiceIdentities[0].ServiceName)
+	})
+
 	// update with no description shouldn't delete the current description
 	t.Run("merge-description", func(t *testing.T) {
 		token := run(t, []string{
@@ -186,6 +214,11 @@ func TestTokenUpdateCommandWithAppend(t *testing.T) {
 					Datacenter: "middleearth-southwest",
 				},
 			},
+			ServiceIdentities: []*api.ACLServiceIdentity{
+				{
+					ServiceName: "fake-service",
+				},
+			},
 		},
 		&api.WriteOptions{Token: "root"},
 	)
@@ -260,6 +293,22 @@ func TestTokenUpdateCommandWithAppend(t *testing.T) {
 		require.Equal(t, "foo", token.NodeIdentities[1].NodeName)
 		require.Equal(t, "bar", token.NodeIdentities[1].Datacenter)
 	})
+
+	// update with append-service-identity
+	t.Run("append-service-identity", func(t *testing.T) {
+		require.Len(t, token.ServiceIdentities, 1)
+
+		token := run(t, []string{
+			"-http-addr=" + a.HTTPAddr(),
+			"-accessor-id=" + token.AccessorID,
+			"-token=root",
+			"-append-service-identity=service:datapalace",
+			"-description=test token",
+		})
+
+		require.Len(t, token.ServiceIdentities, 2)
+		require.Equal(t, "service", token.ServiceIdentities[1].ServiceName)
+	})
 }
 
 func TestTokenUpdateCommand_JSON(t *testing.T) {