def setUp(self):
self.render_context = RenderContext()
def setUp(self):
self.render_context = RenderContext()
class RenderContextTests(TestCase):
def setUp(self):
self.render_context = RenderContext()
def test_render_script(self):
script = MagicMock()
ok_region = MagicMock()
ok_region.render.return_value = 'a'
ko_region = MagicMock()
ko_region.render.return_value = None
script.regions = [
ok_region,
ko_region,
ok_region,
]
result = self.render_context.render_script(script=script)
self.assertEqual('a\na', result)
def test_render_script_already_rendered(self):
script = MagicMock()
ok_region = MagicMock()
ok_region.render.return_value = 'a'
ko_region = MagicMock()
ko_region.render.return_value = None
script.regions = [
ok_region,
ko_region,
ok_region,
]
# Render the script a first time.
self.render_context.render_script(script=script)
result = self.render_context.render_script(script=script)
self.assertEqual('a\na', result)
def test_render_script_already_rendered_pragma_once(self):
script = MagicMock()
script.multiple_inclusion = False
ok_region = MagicMock()
ok_region.render.return_value = 'a'
ko_region = MagicMock()
ko_region.render.return_value = None
script.regions = [
ok_region,
ko_region,
ok_region,
]
# Render the script a first time.
self.render_context.render_script(script=script)
result = self.render_context.render_script(script=script)
self.assertIsNone(result)
def test_render_key(self):
result = self.render_context.render_key(name='mykey')
self.assertEqual('local mykey = KEYS[1]', result)
def test_render_arg_as_int(self):
result = self.render_context.render_arg(
name='myarg',
type_=int,
)
self.assertEqual('local myarg = tonumber(ARGV[1])', result)
def test_render_arg_as_bool(self):
result = self.render_context.render_arg(
name='myarg',
type_=bool,
)
self.assertEqual('local myarg = tonumber(ARGV[1]) ~= 0', result)
def test_render_arg_as_string(self):
result = self.render_context.render_arg(
name='myarg',
type_=str,
)
self.assertEqual('local myarg = ARGV[1]', result)
def test_render_arg_as_list(self):
result = self.render_context.render_arg(
name='myarg',
type_=list,
)
self.assertEqual('local myarg = cjson.decode(ARGV[1])', result)
def test_render_arg_as_dict(self):
result = self.render_context.render_arg(
name='myarg',
type_=dict,
)
self.assertEqual('local myarg = cjson.decode(ARGV[1])', result)
def test_render_return(self):
result = self.render_context.render_return(
type_=str,
)
self.assertEqual('-- Expected return type is: %r' % str, result)
def test_render_pragma(self):
result = self.render_context.render_pragma(
value='once',
)
self.assertEqual('-- File can only be included once.', result)
def test_render_pragma_unknown(self):
with self.assertRaises(AssertionError):
self.render_context.render_pragma(
value='unknown',
)
def test_render_text(self):
result = self.render_context.render_text(
text="foo foo foo",
)
self.assertEqual('foo foo foo', result)
class RenderContextTests(TestCase):
def setUp(self):
self.render_context = RenderContext()
def test_render_script(self):
script = MagicMock()
ok_region = MagicMock()
ok_region.render.return_value = 'a'
ko_region = MagicMock()
ko_region.render.return_value = None
script.regions = [
ok_region,
ko_region,
ok_region,
]
result = self.render_context.render_script(script=script)
self.assertEqual('a\na', result)
def test_render_script_already_rendered(self):
script = MagicMock()
ok_region = MagicMock()
ok_region.render.return_value = 'a'
ko_region = MagicMock()
ko_region.render.return_value = None
script.regions = [
ok_region,
ko_region,
ok_region,
]
# Render the script a first time.
self.render_context.render_script(script=script)
result = self.render_context.render_script(script=script)
self.assertEqual('a\na', result)
def test_render_script_already_rendered_pragma_once(self):
script = MagicMock()
script.multiple_inclusion = False
ok_region = MagicMock()
ok_region.render.return_value = 'a'
ko_region = MagicMock()
ko_region.render.return_value = None
script.regions = [
ok_region,
ko_region,
ok_region,
]
# Render the script a first time.
self.render_context.render_script(script=script)
result = self.render_context.render_script(script=script)
self.assertIsNone(result)
def test_render_key(self):
result = self.render_context.render_key(name='mykey')
self.assertEqual('local mykey = KEYS[1]', result)
def test_render_arg_as_int(self):
result = self.render_context.render_arg(
name='myarg',
type_=int,
)
self.assertEqual('local myarg = tonumber(ARGV[1])', result)
def test_render_arg_as_bool(self):
result = self.render_context.render_arg(
name='myarg',
type_=bool,
)
self.assertEqual('local myarg = tonumber(ARGV[1]) ~= 0', result)
def test_render_arg_as_string(self):
result = self.render_context.render_arg(
name='myarg',
type_=str,
)
self.assertEqual('local myarg = ARGV[1]', result)
def test_render_arg_as_list(self):
result = self.render_context.render_arg(
name='myarg',
type_=list,
)
self.assertEqual('local myarg = cjson.decode(ARGV[1])', result)
def test_render_arg_as_dict(self):
result = self.render_context.render_arg(
name='myarg',
type_=dict,
)
self.assertEqual('local myarg = cjson.decode(ARGV[1])', result)
def test_render_return(self):
result = self.render_context.render_return(type_=str, )
self.assertEqual('-- Expected return type is: %r' % str, result)
def test_render_pragma(self):
result = self.render_context.render_pragma(value='once', )
self.assertEqual('-- File can only be included once.', result)
def test_render_pragma_unknown(self):
with self.assertRaises(AssertionError):
self.render_context.render_pragma(value='unknown', )
def test_render_text(self):
result = self.render_context.render_text(text="foo foo foo", )
self.assertEqual('foo foo foo', result)