### Question:
CriteoAdUnitController { @Nullable static AdUnit getAdUnit(@Nullable String adUnitId) { if (TextUtils.isEmpty(adUnitId)) { return null; } return adUnitMap.get(adUnitId); }    }### Answer:
@Test public void getAdUnit_negativeTests() { AdUnit adUnit1 = CriteoAdUnitController.getAdUnit(null); AdUnit adUnit2 = CriteoAdUnitController.getAdUnit(""); AdUnit adUnit3 = CriteoAdUnitController.getAdUnit("ad_unit_id"); assertNull(adUnit1); assertNull(adUnit2); assertNull(adUnit3); }
@Test public void getAdUnit() { AdUnit adUnit1 = mock(AdUnit.class); AdUnit adUnit2 = mock(AdUnit.class); CriteoAdUnitController.getAdUnitMap().put("ad_unit_1", adUnit1); CriteoAdUnitController.getAdUnitMap().put("ad_unit_2", adUnit2); assertEquals(adUnit1, CriteoAdUnitController.getAdUnit("ad_unit_1")); assertEquals(adUnit2, CriteoAdUnitController.getAdUnit("ad_unit_2")); }