Expose context and DataStoreService for testing.