Add unit tests for getting tickets via uniq prefix

1 files changed, 16 insertions, 0 deletions

diff --git a/distixlib/ticket_store_tests.py b/distixlib/ticket_store_tests.py
index 1be4511..9e47479 100644
--- a/distixlib/ticket_store_tests.py
+++ b/distixlib/ticket_store_tests.py
@@ -141,6 +141,22 @@ class TicketStoreTests(unittest.TestCase):
             [id(ticket) for ticket in tickets],
             [id(self.ticket)])
 
+    def test_gets_ticket_by_unique_id_prefix(self):
+        self.ticket.set_ticket_id('foo-id')
+        self.ticket_store.add_ticket(self.ticket)
+        self.assertEqual(self.ticket_store.get_ticket('foo'), self.ticket)
+
+    def test_fails_getging_ticket_by_id_prefix_if_not_unique(self):
+        self.ticket.set_ticket_id('foo-id')
+        self.ticket_store.add_ticket(self.ticket)
+
+        ticket2 = distixlib.Ticket()
+        ticket2.set_ticket_id('foo-id2')
+        self.ticket_store.add_ticket(ticket2)
+
+        with self.assertRaises(distixlib.TicketNotInStoreError):
+            self.ticket_store.get_ticket('foo')
+
     def test_saving_makes_store_clean(self):
         self.ticket.set_ticket_id('foo-id')
         self.ticket_store.add_ticket(self.ticket)