Data layer test code