From 5b03e660486c30222583311dabbc5b170019b5b4 Mon Sep 17 00:00:00 2001 From: Itai Shirav Date: Tue, 7 Feb 2017 16:36:41 +0200 Subject: [PATCH] Send readonly=1 when database is created in readonly mode --- src/infi/clickhouse_orm/database.py | 2 ++ tests/test_database.py | 14 +++++++++++++- 2 files changed, 15 insertions(+), 1 deletion(-) diff --git a/src/infi/clickhouse_orm/database.py b/src/infi/clickhouse_orm/database.py index 7b4b398..43ca36e 100644 --- a/src/infi/clickhouse_orm/database.py +++ b/src/infi/clickhouse_orm/database.py @@ -146,6 +146,8 @@ class Database(object): params['user'] = self.username if self.password: params['password'] = self.password + if self.readonly: + params['readonly'] = '1' return params def _substitute(self, query, model_class=None): diff --git a/tests/test_database.py b/tests/test_database.py index 30c25d3..1e62472 100644 --- a/tests/test_database.py +++ b/tests/test_database.py @@ -2,7 +2,7 @@ import unittest -from infi.clickhouse_orm.database import Database +from infi.clickhouse_orm.database import Database, DatabaseException from infi.clickhouse_orm.models import Model from infi.clickhouse_orm.fields import * from infi.clickhouse_orm.engines import * @@ -117,6 +117,18 @@ class DatabaseTestCase(unittest.TestCase): p = list(self.database.select("SELECT * from $table", Person))[0] self.assertEquals(p.first_name, s) + def test_readonly(self): + orig_database = self.database + self.database = Database(orig_database.db_name, readonly=True) + with self.assertRaises(DatabaseException): + self._insert_and_check(self._sample_data(), len(data)) + self.assertEquals(self.database.count(Person), 0) + with self.assertRaises(DatabaseException): + self.database.drop_table(Person) + with self.assertRaises(DatabaseException): + self.database.drop_database() + self.database = orig_database + def _sample_data(self): for entry in data: yield Person(**entry)