def test_test_to_negotiate_uri(self): ''' Check test_to_negotiate_uri works as expected ''' # # It should always return the first URI: it suck but that is # how it works as of now. # lst = RunnerLst() lst.update({'foo': ['a', 'b', 'c']}) self.assertEqual(lst.test_to_negotiate_uri('foo'), 'a') self.assertEqual(lst.test_to_negotiate_uri('foo'), 'a') self.assertEqual(lst.test_to_negotiate_uri('foo'), 'a')
def test_when_empty(self): ''' Check behavior is correct when empty ''' # It should return None in both cases lst = RunnerLst() self.assertEqual(lst.test_to_negotiate_uri('foo'), None) self.assertEqual(lst.get_next_test(), None)