def test_set_current_unit_updates_activ_unit(self): selector = OrgUnitSelector( {CURRENT_ORG_UNIT_KEY: 'clientb'}, [self.unit_a, self.unit_b, self.unit_c], [self.unit_a, self.unit_b, self.unit_c]) selector.set_current_unit('clienta') self.assertEquals(self.unit_a, selector.get_current_unit())