def test_parse_ad_bad_args(self):
token = mock.Mock()
token.split_contents = mock.Mock(return_value=['ad', 'SLOT'])
with self.assertRaises(template.TemplateSyntaxError) as exc:
tags.parse_ad(None, token)
self.assertEqual(str(exc.exception), u'usage: {% ad SLOT BREAKPOINT [BREAKPOINT ...] %}')
def test_parse_ad(self):
token = mock.Mock()
token.split_contents = mock.Mock(return_value=['ad', 'SLOT', 'BREAKPOINT'])
node = tags.parse_ad(None, token)
self.assertEqual(node.slot, 'SLOT')
self.assertListEqual(node.breakpoints, ['BREAKPOINT'])
