def test_pretty_key(self): visitor = ShardingByLangVisitor(Mock()) self.assertEqual(visitor.pretty_key('1'), 'Lang #1')
def test_key(self): visitor = ShardingByLangVisitor(Mock()) self.assertEqual(visitor.build_key(Mock(lang_id='1')), '1')