Skip to content
This repository was archived by the owner on Apr 8, 2025. It is now read-only.

Commit 9bab6f0

Browse files
committed
Add type provider default for timestamp with time zone
1 parent 8b43bbe commit 9bab6f0

File tree

4 files changed

+14
-7
lines changed

4 files changed

+14
-7
lines changed

README.rst

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -117,21 +117,21 @@ For example given the table as follows::
117117

118118
create table x.demo (
119119
id int,
120-
name string,
121-
country string
120+
name text,
121+
country text
122122
);
123123

124124
The following command can be used to insert 1000 records::
125125

126126
>>> cr8 insert-fake-data --hosts localhost:4200 --table x.demo --num-records 200
127127
Found schema:
128128
{
129-
"country": "string",
129+
"country": "text",
130130
"id": "integer",
131-
"name": "string"
131+
"name": "text"
132132
}
133133
Using insert statement:
134-
insert into "x"."demo" ("country", "id", "name") values (?, ?, ?)
134+
insert into "x"."demo" ("id", "name", "country") values (?, ?, ?)
135135
Will make 1 requests with a bulk size of 200
136136
Generating fake data and executing inserts
137137
<BLANKLINE>

cr8/insert_fake_data.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -98,6 +98,8 @@ class DataFaker:
9898
'ip': operator.attrgetter('ipv4'),
9999
'timestamp': lambda f: partial(
100100
f.date_time_between, start_date='-2y', end_date='now'),
101+
'timestamp with time zone': lambda f: partial(
102+
f.date_time_between, start_date='-2y', end_date='now'),
101103
'string': operator.attrgetter('word'),
102104
'text': operator.attrgetter('word'),
103105
'boolean': operator.attrgetter('boolean'),

tests/test_insert_fake_data.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,12 @@ def test_timestamp_type_default(self):
5454
diff = datetime.datetime(2017, 11, 18, 19, 0, 0) - dt
5555
self.assertLessEqual(diff, datetime.timedelta(seconds=1))
5656

57+
def test_timestamp_with_time_zone_type_default(self):
58+
provider = self.f.provider_for_column('some_ts_column', 'timestamp with time zone')
59+
dt = provider()
60+
diff = datetime.datetime(2017, 11, 18, 19, 0, 0) - dt
61+
self.assertLessEqual(diff, datetime.timedelta(seconds=1))
62+
5763
def test_provider_from_mapping(self):
5864
mapping = {'x': ['random_int', [10, 20]]}
5965
provider = self.f.provider_from_mapping('x', mapping)

tests/test_integration.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,7 @@
1313
crate_dir=crate_dir,
1414
settings={
1515
'cluster.name': 'cr8-tests',
16-
'http.port': '44200-44250',
17-
'bootstrap.system_call_filter': False
16+
'http.port': '44200-44250'
1817
})
1918

2019

0 commit comments

Comments
 (0)