ungana

Client application that creates customized .ics files for ticket booking and event reservations
Info | Log | Files | Refs | README

test_wala_store.py (1205B)

      1 import unittest
      2 from unittest.mock import patch, MagicMock
      3 from ungana.store.wala_store import UnganaWalaStore
      4 
      5 
      6 class TestUnganaWalaStore(unittest.TestCase):
      7 
      8     @patch("ungana.store.wala_store.requests.put")
      9     def test_wala_store_put(self, mock_put):
     10         mock_put.return_value = MagicMock(status_code=200, text="digest123")
     11 
     12         store = UnganaWalaStore()
     13         url = "https://wala.ungana.com/store"
     14         key = "ignored_key"
     15         data = b"foobar"
     16 
     17         result_url = store.put(url, key, data)
     18         self.assertEqual(result_url, "https://wala.ungana.com/store/digest123")
     19         mock_put.assert_called_once_with(url, data=data, timeout=10)
     20 
     21     @patch("ungana.store.wala_store.requests.get")
     22     def test_wala_store_get(self, mock_get):
     23         mock_get.return_value = MagicMock(status_code=200, content=b"ABCDEF")
     24 
     25         store = UnganaWalaStore()
     26         base_url = "https://wala.ungana.com/store"
     27         key = "digest123"
     28 
     29         result = store.get(base_url, key)
     30         self.assertEqual(result, b"ABCDEF")
     31         mock_get.assert_called_once_with(
     32             "https://wala.ungana.com/store/digest123", timeout=10
     33         )
     34 
     35 
     36 if __name__ == "__main__":
     37     unittest.main()