Implement Unit-Tests

### Is your feature request related to a problem? If so, please describe.

It's hard to implement new things and fix some bugs, with Unit Tests these problems might be easy to work on

### What feature or improvement would you like to see?

Use JUnit and Mockito to tests

### Additional context

_No response_