Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
2dot5
ClickHouse
提交
5a73a986
C
ClickHouse
项目概览
2dot5
/
ClickHouse
通知
3
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
C
ClickHouse
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
5a73a986
编写于
2月 22, 2019
作者:
A
alesapin
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Remove pymongo
上级
2bb54b0b
变更
2
显示空白变更内容
内联
并排
Showing
2 changed file
with
37 addition
and
35 deletion
+37
-35
dbms/tests/integration/test_external_dictionaries/external_sources.py
...ntegration/test_external_dictionaries/external_sources.py
+35
-35
dbms/tests/integration/test_external_dictionaries/test.py
dbms/tests/integration/test_external_dictionaries/test.py
+2
-0
未找到文件。
dbms/tests/integration/test_external_dictionaries/external_sources.py
浏览文件 @
5a73a986
# -*- coding: utf-8 -*-
import
warnings
import
pymysql.cursors
import
pymongo
#
import pymongo
import
os
...
...
@@ -123,40 +123,40 @@ class SourceMySQL(ExternalSource):
self
.
execute_mysql_query
(
query
)
class
SourceMongo
(
ExternalSource
):
def
get_source_str
(
self
,
table_name
):
return
'''
<mongodb>
<host>{host}</host>
<port>{port}</port>
<user>{user}</user>
<password>{password}</password>
<db>test</db>
<collection>{tbl}</collection>
</mongodb>
'''
.
format
(
host
=
self
.
docker_hostname
,
port
=
self
.
docker_port
,
user
=
self
.
user
,
password
=
self
.
password
,
tbl
=
table_name
,
)
def
prepare
(
self
,
structure
,
table_name
,
cluster
):
connection_str
=
'mongodb://{user}:{password}@{host}:{port}'
.
format
(
host
=
self
.
internal_hostname
,
port
=
self
.
internal_port
,
user
=
self
.
user
,
password
=
self
.
password
)
self
.
connection
=
pymongo
.
MongoClient
(
connection_str
)
self
.
connection
.
create
self
.
structure
=
structure
self
.
db
=
self
.
connection
[
'test'
]
self
.
prepared
=
True
def
load_data
(
self
,
data
,
table_name
):
tbl
=
self
.
db
[
table_name
]
to_insert
=
[
dict
(
row
.
data
)
for
row
in
data
]
result
=
tbl
.
insert_many
(
to_insert
)
#
class SourceMongo(ExternalSource):
#
#
def get_source_str(self, table_name):
#
return '''
#
<mongodb>
#
<host>{host}</host>
#
<port>{port}</port>
#
<user>{user}</user>
#
<password>{password}</password>
#
<db>test</db>
#
<collection>{tbl}</collection>
#
</mongodb>
#
'''.format(
#
host=self.docker_hostname,
#
port=self.docker_port,
#
user=self.user,
#
password=self.password,
#
tbl=table_name,
#
)
#
#
def prepare(self, structure, table_name, cluster):
#
connection_str = 'mongodb://{user}:{password}@{host}:{port}'.format(
#
host=self.internal_hostname, port=self.internal_port,
#
user=self.user, password=self.password)
#
self.connection = pymongo.MongoClient(connection_str)
#
self.connection.create
#
self.structure = structure
#
self.db = self.connection['test']
#
self.prepared = True
#
#
def load_data(self, data, table_name):
#
tbl = self.db[table_name]
#
to_insert = [dict(row.data) for row in data]
#
result = tbl.insert_many(to_insert)
class
SourceClickHouse
(
ExternalSource
):
...
...
dbms/tests/integration/test_external_dictionaries/test.py
浏览文件 @
5a73a986
...
...
@@ -151,6 +151,7 @@ def test_simple_dictionaries(started_cluster):
dct
.
load_data
(
data
)
node
.
query
(
"system reload dictionaries"
)
queries_with_answers
=
[]
for
dct
in
simple_dicts
:
for
row
in
data
:
...
...
@@ -183,6 +184,7 @@ def test_complex_dictionaries(started_cluster):
dct
.
load_data
(
data
)
node
.
query
(
"system reload dictionaries"
)
queries_with_answers
=
[]
for
dct
in
complex_dicts
:
for
row
in
data
:
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录