Add test for DbContext