Add capacity examples

ClientLibrary/Snippets/Microsoft.TeamServices.Samples.Client/Work/TeamSettingsSample.cs

@@ -82,5 +82,97 @@ public TeamSetting UpdateTeamSettings()
 
             return result;
         }
+
+        [ClientSampleMethod]
+        public TeamMemberCapacityIdentityRef GetTeamMemberCapacity()
+        {
+            VssConnection connection = Context.Connection;
+            WorkHttpClient workClient = connection.GetClient<WorkHttpClient>();
+
+            Guid projectId = ClientSampleHelpers.FindAnyProject(this.Context).Id;
+            Guid teamId = ClientSampleHelpers.FindAnyTeam(this.Context, projectId).Id;
+            Guid userId = ClientSampleHelpers.GetCurrentUserId(this.Context);
+
+            TeamContext teamContext = new TeamContext(projectId, teamId);
+            List<TeamSettingsIteration> result = workClient.GetTeamIterationsAsync(teamContext, "current").Result;
+            Guid iterationId = result[0].Id;
+
+            TeamMemberCapacityIdentityRef capacity = workClient.GetCapacityWithIdentityRefAsync(teamContext, iterationId, userId).Result;
+
+            return capacity;
+        }
+
+        [ClientSampleMethod]
+        public List<TeamMemberCapacityIdentityRef> GetTeamCapacity()
+        {
+            VssConnection connection = Context.Connection;
+            WorkHttpClient workClient = connection.GetClient<WorkHttpClient>();
+
+            Guid projectId = ClientSampleHelpers.FindAnyProject(this.Context).Id;
+            Guid teamId = ClientSampleHelpers.FindAnyTeam(this.Context, projectId).Id;
+
+            TeamContext teamContext = new TeamContext(projectId, teamId);
+            List<TeamSettingsIteration> result = workClient.GetTeamIterationsAsync(teamContext, "current").Result;
+            Guid iterationId = result[0].Id;
+
+            List<TeamMemberCapacityIdentityRef> capacity = workClient.GetCapacitiesWithIdentityRefAsync(teamContext, iterationId).Result;
+
+            return capacity;
+        }
+
+        [ClientSampleMethod]
+        public TeamMemberCapacityIdentityRef UpdateTeamMemberCapacity()
+        {
+            VssConnection connection = Context.Connection;
+            WorkHttpClient workClient = connection.GetClient<WorkHttpClient>();
+
+            Guid projectId = ClientSampleHelpers.FindAnyProject(this.Context).Id;
+            Guid teamId = ClientSampleHelpers.FindAnyTeam(this.Context, projectId).Id;
+            Guid userId = ClientSampleHelpers.GetCurrentUserId(this.Context);
+
+            TeamContext teamContext = new TeamContext(projectId, teamId);
+            List<TeamSettingsIteration> result = workClient.GetTeamIterationsAsync(teamContext, "current").Result;
+            Guid iterationId = result[0].Id;
+
+            TeamMemberCapacityIdentityRef capacity = workClient.GetCapacityWithIdentityRefAsync(teamContext, iterationId, userId).Result;
+            CapacityPatch capacityPatch = new CapacityPatch()
+            {
+                Activities = capacity.Activities.Select(a => { return new Activity { Name = a.Name, CapacityPerDay = a.CapacityPerDay + 1 }; }),
+                DaysOff = capacity.DaysOff
+            };
+
+            TeamMemberCapacityIdentityRef updatedCapacity = workClient.UpdateCapacityWithIdentityRefAsync(capacityPatch, teamContext, iterationId, userId).Result;
+
+            return updatedCapacity;
+        }
+
+        [ClientSampleMethod]
+        public List<TeamMemberCapacityIdentityRef> ReplaceTeamMemberCapacities()
+        {
+            VssConnection connection = Context.Connection;
+            WorkHttpClient workClient = connection.GetClient<WorkHttpClient>();
+
+            Guid projectId = ClientSampleHelpers.FindAnyProject(this.Context).Id;
+            Guid teamId = ClientSampleHelpers.FindAnyTeam(this.Context, projectId).Id;
+
+            TeamContext teamContext = new TeamContext(projectId, teamId);
+            List<TeamSettingsIteration> result = workClient.GetTeamIterationsAsync(teamContext, "current").Result;
+            Guid iterationId = result[0].Id;
+
+            List<TeamMemberCapacityIdentityRef> capacity = workClient.GetCapacitiesWithIdentityRefAsync(teamContext, iterationId).Result;
+            var updatedCapacity = capacity.Select(teamMemberCapacity =>
+            {
+                return new TeamMemberCapacityIdentityRef()
+                {
+                    TeamMember = teamMemberCapacity.TeamMember,
+                    Activities = teamMemberCapacity.Activities.Select(a => { return new Activity() { Name = a.Name, CapacityPerDay = a.CapacityPerDay }; }),
+                    DaysOff = teamMemberCapacity.DaysOff
+                };
+            });
+
+            var updatedCapacityResult = workClient.ReplaceCapacitiesWithIdentityRefAsync(updatedCapacity, teamContext, iterationId).Result;
+
+            return updatedCapacityResult;
+        }
     }
 }