Add initial integration tests in test directory