tests: Add tests for all classes