Modify Integration test to DbGenerator