Use MockDataStore for handler_test.go