adding condition checks in count.js and edge cases in the test

Author: TzeMingHo

Sprint-3/2-practice-tdd/count.js

@@ -1,5 +1,15 @@
 function countChar(stringOfCharacters, findCharacter) {
-  const arrayOfCharacters = stringOfCharacters.split("");
+  let arrayOfCharacters = [];
+  if (typeof stringOfCharacters === "string") {
+    arrayOfCharacters = stringOfCharacters.split("");
+  } else if (Array.isArray(stringOfCharacters)) {
+    arrayOfCharacters = stringOfCharacters;
+  } else if (typeof stringOfCharacters === "number") {
+    arrayOfCharacters = stringOfCharacters.toString().split("");
+  } else if (typeof stringOfCharacters === "object") {
+    return 0;
+  }
+
   let count = 0;
   for (let index = 0; index < arrayOfCharacters.length; index++) {
     if (arrayOfCharacters[index] === findCharacter) {

Sprint-3/2-practice-tdd/count.test.js

@@ -14,7 +14,7 @@ test("should count multiple occurrences of a character", () => {
   const str = "aaaaa";
   const char = "a";
   const count = countChar(str, char);
-  expect(count).toEqual(5);
+  expect(parseInt(count)).toEqual(5);
 });
 
 // Scenario: No Occurrences
@@ -27,5 +27,29 @@ test("should return 0 when character does not exist in the string", () => {
   const str = "abcdefg";
   const char = "h";
   const count = countChar(str, char);
-  expect(count).toEqual(0);
+  expect(parseInt(count)).toEqual(0);
+});
+
+// test for empty string
+test("should return 0 when string is empty", () => {
+  const str = "";
+  const char = "a";
+  const count = countChar(str, char);
+  expect(parseInt(count)).toEqual(0);
+});
+
+// test for str is an array
+test("should return 0 when str is an array", () => {
+  const str = ["a", "b", "c"];
+  const char = "a";
+  const count = countChar(str, char);
+  expect(parseInt(count)).toEqual(1);
+});
+
+// test for str is a number
+test("should return 0 when str is a number", () => {
+  const str = 12345;
+  const char = "3";
+  const count = countChar(str, char);
+  expect(parseInt(count)).toEqual(1);
 });