def test_get_absolute_url(self):
expect(PoliceUnit(unit_name='011').v2_to).to.eq('/unit/011/')
def test_v2_to(self):
unit = '011'
to = '/unit/011/'
expect(PoliceUnit(unit_name=unit).v2_to).to.eq(to)
def test_v1_url(self):
unit = 'unit'
url = 'domain/url-mediator/session-builder?unit=unit'
expect(PoliceUnit(unit_name=unit).v1_url).to.eq(url)
def test_str(self):
self.assertEqual(str(PoliceUnit(unit_name='lorem')), 'lorem')