Added a unit test for the get_log_index method.

pull/61/head
Christian T. Jacobs 2017-04-23 20:04:38 +01:00
rodzic 3288a1f47d
commit 6b069e280e
2 zmienionych plików z 8 dodań i 1 usunięć

Wyświetl plik

@ -940,7 +940,7 @@ class Logbook:
""" Given the name of a log, return its index in the list of Log objects.
:arg str name: The name of the log. If None, use the name of the currently-selected log.
:returns: The index of the named log in the list of Log objects.
:returns: The index of the named log in the list of Log objects. Returns None is the log cannot be found.
:rtype: int
"""
if(name is None):
@ -954,6 +954,7 @@ class Logbook:
# If a page of the logbook (and therefore a Log object) gets deleted,
# then the page_index may not correspond to the index of the log in the self.logs list.
# Therefore, we have to search for the tab with the same name as the log.
log_index = None
for i in range(0, len(self.logs)):
if(self.logs[i].name == name):
log_index = i

Wyświetl plik

@ -89,5 +89,11 @@ class TestLogbook(unittest.TestCase):
present = self.logbook.filter_by_callsign(model, iter, data=None)
assert(not present) # "HELLOWORLD" is not present in "TEST123"
def test_get_log_index(self):
""" Check that a log's index can be resolved using the log's name. """
assert(self.logbook.get_log_index(name="test") == 0)
assert(self.logbook.get_log_index(name="test2") == 1)
assert(self.logbook.get_log_index(name="helloworld") is None)
if(__name__ == '__main__'):
unittest.main()