@Test
  public void testGetStore() {
    KeyValueStore store = mock(KeyValueStore.class);
    when(keyValueStoreProvider.apply("myStore")).thenReturn(store);
    assertEquals(store, taskContext.getStore("myStore"));
  }
Given that there is a store corresponding to the storeName, getStore should return the store.
samza-core/src/test/java/org/apache/samza/context/TestTaskContextImpl.java