Fixing tests
Adding tests for integration class