Merge pull request #32 from levimatheri/user-workflow

Add get-by-id and delete user

Author: hiranya911

FirebaseAdmin/FirebaseAdmin.Tests/Auth/FirebaseUserManagerTest.cs

@@ -71,6 +71,44 @@ public void TooLargeClaimsPayload()
             Assert.Throws<ArgumentException>(() => new UserRecord("user1") { CustomClaims = customClaims });
         }
 
+        [Fact]
+        public async Task GetUserById()
+        {
+            var handler = new MockMessageHandler()
+            {
+                Response = new UserRecord("user1"),
+            };
+            var factory = new MockHttpClientFactory(handler);
+            var userManager = new FirebaseUserManager(
+                new FirebaseUserManagerArgs
+                {
+                    Credential = MockCredential,
+                    ProjectId = MockProjectId,
+                    ClientFactory = factory,
+                });
+            var userRecord = await userManager.GetUserById("user1");
+            Assert.Equal("user1", userRecord.Uid);
+        }
+
+        [Fact]
+        public async Task GetUserByIdUserNotFound()
+        {
+            var handler = new MockMessageHandler()
+            {
+                StatusCode = HttpStatusCode.NotFound,
+            };
+            var factory = new MockHttpClientFactory(handler);
+            var userManager = new FirebaseUserManager(
+                new FirebaseUserManagerArgs
+                {
+                    Credential = MockCredential,
+                    ProjectId = MockProjectId,
+                    ClientFactory = factory,
+                });
+            await Assert.ThrowsAsync<FirebaseException>(
+                async () => await userManager.GetUserById("user1"));
+        }
+
         [Fact]
         public async Task UpdateUser()
         {
@@ -165,5 +203,45 @@ public async Task UpdateUserHttpError()
             await Assert.ThrowsAsync<FirebaseException>(
                 async () => await userManager.UpdateUserAsync(new UserRecord("user1") { CustomClaims = customClaims }));
         }
+
+        [Fact]
+        public async Task DeleteUser()
+        {
+            var handler = new MockMessageHandler()
+            {
+                Response = new Dictionary<string, string>()
+                {
+                    { "kind", "identitytoolkit#DeleteAccountResponse" },
+                },
+            };
+            var factory = new MockHttpClientFactory(handler);
+            var userManager = new FirebaseUserManager(
+                new FirebaseUserManagerArgs
+                {
+                    Credential = MockCredential,
+                    ProjectId = MockProjectId,
+                    ClientFactory = factory,
+                });
+            await userManager.DeleteUser("user1");
+        }
+
+        [Fact]
+        public async Task DeleteUserNotFound()
+        {
+            var handler = new MockMessageHandler()
+            {
+                StatusCode = HttpStatusCode.NotFound,
+            };
+            var factory = new MockHttpClientFactory(handler);
+            var userManager = new FirebaseUserManager(
+                new FirebaseUserManagerArgs
+                {
+                    Credential = MockCredential,
+                    ProjectId = MockProjectId,
+                    ClientFactory = factory,
+                });
+            await Assert.ThrowsAsync<FirebaseException>(
+               async () => await userManager.DeleteUser("user1"));
+        }
     }
 }

FirebaseAdmin/FirebaseAdmin/Auth/FirebaseUserManager.cs

@@ -13,6 +13,7 @@
 // limitations under the License.
 
 using System;
+using System.Collections.Generic;
 using System.Net.Http;
 using System.Threading;
 using System.Threading.Tasks;
@@ -60,6 +61,36 @@ public static FirebaseUserManager Create(FirebaseApp app)
             return new FirebaseUserManager(args);
         }
 
+        /// <summary>
+        /// Gets the user data corresponding to the given user ID.
+        /// </summary>
+        /// <param name="uid">A user ID string.</param>
+        /// <param name="cancellationToken">A cancellation token to monitor the asynchronous
+        /// operation.</param>
+        /// <returns>A record of user with the queried id if one exists.</returns>
+        public async Task<UserRecord> GetUserById(
+            string uid, CancellationToken cancellationToken = default(CancellationToken))
+        {
+            if (string.IsNullOrEmpty(uid))
+            {
+                throw new ArgumentException("User ID cannot be null or empty.");
+            }
+
+            const string getUserPath = "accounts:lookup";
+            var payload = new Dictionary<string, object>()
+            {
+                { "localId", uid },
+            };
+            var response = await this.PostAndDeserializeAsync<JObject>(
+                getUserPath, payload, cancellationToken).ConfigureAwait(false);
+            if (response == null || uid != (string)response["localId"])
+            {
+                throw new FirebaseException($"Failed to get user: {uid}");
+            }
+
+            return new UserRecord((string)response["localId"]);
+        }
+
         /// <summary>
         /// Update an existing user.
         /// </summary>
@@ -79,6 +110,33 @@ public async Task UpdateUserAsync(
             }
         }
 
+        /// <summary>
+        /// Delete user data corresponding to the given user ID.
+        /// </summary>
+        /// <param name="uid">A user ID string.</param>
+        /// <param name="cancellationToken">A cancellation token to monitor the asynchronous
+        /// operation.</param>
+        public async Task DeleteUser(
+            string uid, CancellationToken cancellationToken = default(CancellationToken))
+        {
+            if (string.IsNullOrEmpty(uid))
+            {
+                throw new ArgumentException("User id cannot be null or empty.");
+            }
+
+            const string getUserPath = "accounts:delete";
+            var payload = new Dictionary<string, object>()
+            {
+                { "localId", uid },
+            };
+            var response = await this.PostAndDeserializeAsync<JObject>(
+                getUserPath, payload, cancellationToken).ConfigureAwait(false);
+            if (response == null || (string)response["kind"] == null)
+            {
+                throw new FirebaseException($"Failed to delete user: {uid}");
+            }
+        }
+
         public void Dispose()
         {
             this.httpClient.Dispose();