39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
|
#-*- coding: utf-8; -*-
|
||
|
|
||
|
from unittest.mock import patch
|
||
|
|
||
|
from wuttjamaican.testing import DataTestCase
|
||
|
|
||
|
from wuttasync.importing import wutta as mod
|
||
|
|
||
|
|
||
|
class TestToWuttaHandler(DataTestCase):
|
||
|
|
||
|
def make_handler(self, **kwargs):
|
||
|
return mod.ToWuttaHandler(self.config, **kwargs)
|
||
|
|
||
|
def test_get_target_title(self):
|
||
|
handler = self.make_handler()
|
||
|
|
||
|
# uses app title by default
|
||
|
self.config.setdefault('wutta.app_title', "What About This")
|
||
|
self.assertEqual(handler.get_target_title(), 'What About This')
|
||
|
|
||
|
# or generic default if present
|
||
|
handler.generic_target_title = "WHATABOUTTHIS"
|
||
|
self.assertEqual(handler.get_target_title(), 'WHATABOUTTHIS')
|
||
|
|
||
|
# but prefer specific title if present
|
||
|
handler.target_title = "what_about_this"
|
||
|
self.assertEqual(handler.get_target_title(), 'what_about_this')
|
||
|
|
||
|
def test_make_target_session(self):
|
||
|
handler = self.make_handler()
|
||
|
|
||
|
# makes "new" (mocked in our case) app session
|
||
|
with patch.object(self.app, 'make_session') as make_session:
|
||
|
make_session.return_value = self.session
|
||
|
session = handler.make_target_session()
|
||
|
make_session.assert_called_once_with()
|
||
|
self.assertIs(session, self.session)
|