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')