Testing the Pomodoro Functionality

Now that we have both implementations for the Pomodoro business logic and the in-memory repository, let’s write some tests for the business logic.

For brevity, we’ll only add tests for the business logic, which will indirectly test the repository when it’s used. For a real production application, we recommend that we write unit tests for the repository implementation as well.

Updating the inmemory_test.go file

Some of these tests require access to the repository. Because we can have different implementations of the repository, first let’s create a helper function getRepo() to get an instance of the repository. We can implement different versions of this function for the different repository implementations without changing the test code.

We open the inmemory_test.go file in our editor to write the specific function for this repository implementation. We add the package definition and import section. We’ll use the testing package for the testing-related functions, and we’ll also use pomodoro and repository, so we can use the repository:

Finally, we define the function getRepo(), which returns the repository instance and a cleanup function. The in-memory repository doesn’t require a cleanup ...