Added some DB integration tests