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())
