Changeset 12278 for main/waeup.ikoba/branches

Timestamp:

21 Dec 2014, 08:45:24 (10 years ago)

Author:

uli

Message:

Update tests.

File:

• main/waeup.ikoba/branches/uli-payments/src/waeup/ikoba/payments/tests/test_payment.py (modified) (1 diff)

main/waeup.ikoba/branches/uli-payments/src/waeup/ikoba/payments/tests/test_payment.py

@@ -19 +19 @@
 import re
 import unittest
+from zope.component import getUtilitiesFor, getSiteManager
+from zope.interface import implements
 from zope.interface.verify import verifyClass, verifyObject
 from waeup.ikoba.payments.interfaces import (
     IPayment, STATE_UNPAID, STATE_PAID, STATE_FAILED,
+    IPaymentGatewayService,
     )
-from waeup.ikoba.payments.payment import Payment
+from waeup.ikoba.payments.payment import Payment, get_payment_providers
+
+
+class HelperTests(unittest.TestCase):
+
+    def tearDown(self):
+        # unregister any IPaymentGatewayServices
+        sm = getSiteManager(None)
+        for name, util in getUtilitiesFor(
+            IPaymentGatewayService):
+            sm.unregisterUtility(util, name=name)
+
+    def test_get_payment_providers_no_providers(self):
+        # we can get a dict of all payment providers
+        result = get_payment_providers()
+        assert isinstance(result, dict)
+        assert result == {}
+
+    def test_get_payment_providers(self):
+        # we get any payment providers registered
+        sm = getSiteManager(None)
+
+        class FakeUtil(object):
+            implements(IPaymentGatewayService)
+
+        fake_util = FakeUtil()
+        sm.registerUtility(fake_util, name=u'some_name')
+        result = get_payment_providers()
+        assert isinstance(result, dict)
+        assert result.keys() == ['some_name', ]
+        assert result['some_name'] is fake_util