[FEATURE] Add the actual tests.