[TEST] Add unit test for existing code

[kashifkhan0771]

Feature Description

Add unit test for all Public methods.

---

Use Case

N/A

---

Proposed Solution

Use go built-in library for adding unit tests

---

Pseudo Code

N/A