Moved test behaviour classes to src/tests folder