Add test cases for D1