61 lines
2 KiB
Python
61 lines
2 KiB
Python
# -*- coding: utf-8; -*-
|
|
|
|
import socket
|
|
|
|
from unittest.mock import patch
|
|
|
|
from wuttjamaican.testing import DataTestCase
|
|
|
|
from wutta_continuum import conf as mod
|
|
|
|
|
|
class TestWuttaContinuumConfigExtension(DataTestCase):
|
|
|
|
def make_extension(self):
|
|
return mod.WuttaContinuumConfigExtension()
|
|
|
|
def test_startup(self):
|
|
ext = self.make_extension()
|
|
|
|
with patch.object(mod, "make_versioned") as make_versioned:
|
|
with patch.object(mod, "configure_mappers") as configure_mappers:
|
|
|
|
# nothing happens by default
|
|
ext.startup(self.config)
|
|
make_versioned.assert_not_called()
|
|
configure_mappers.assert_not_called()
|
|
|
|
# but will if we enable it in config
|
|
self.config.setdefault("wutta_continuum.enable_versioning", "true")
|
|
ext.startup(self.config)
|
|
make_versioned.assert_called_once()
|
|
configure_mappers.assert_called_once_with()
|
|
|
|
|
|
class TestWuttaContinuumPlugin(DataTestCase):
|
|
|
|
def make_plugin(self):
|
|
return mod.WuttaContinuumPlugin()
|
|
|
|
def test_remote_addr(self):
|
|
plugin = self.make_plugin()
|
|
with patch.object(socket, "gethostbyname", return_value="127.0.0.1"):
|
|
self.assertEqual(plugin.get_remote_addr(None, self.session), "127.0.0.1")
|
|
|
|
def test_user_id(self):
|
|
plugin = self.make_plugin()
|
|
self.assertIsNone(plugin.get_user_id(None, self.session))
|
|
|
|
def test_transaction_args(self):
|
|
plugin = self.make_plugin()
|
|
with patch.object(socket, "gethostbyname", return_value="127.0.0.1"):
|
|
self.assertEqual(
|
|
plugin.transaction_args(None, self.session),
|
|
{"remote_addr": "127.0.0.1"},
|
|
)
|
|
|
|
with patch.object(plugin, "get_user_id", return_value="some-random-uuid"):
|
|
self.assertEqual(
|
|
plugin.transaction_args(None, self.session),
|
|
{"remote_addr": "127.0.0.1", "user_id": "some-random-uuid"},
|
|
)
|