Revert "Tests: Refactor the main module test"

This reverts commit b4a97f3fefb54e9f93aa7343668aaed06df41e28
with some modifications.

Let's revert my style :-)