def test_get_db_group_for_default(self):
        router = ShardedPerTenantRouter()
        router._write_mode = True
        db_group = router._get_db_group(TShardedModel)

        self.assertEqual(db_group, 'test1')
    def test_get_db_group_for_read(self):
        router = ShardedPerTenantRouter()
        router._write_mode = False
        db_group = router._get_db_group(TShardedModel)

        self.assertEqual(db_group, 'test2')