def test_query_with_date_interval(self):
task = ImportLastCountryOfUserToHiveTask(**self._get_kwargs())
query = task.query()
expected_query = textwrap.dedent("""
USE default;
DROP TABLE IF EXISTS last_country_of_user;
CREATE EXTERNAL TABLE last_country_of_user (
country_name STRING,country_code STRING,username STRING
)
PARTITIONED BY (dt STRING)
ROW FORMAT DELIMITED FIELDS TERMINATED BY '\t'
LOCATION 's3://output/path';
ALTER TABLE last_country_of_user ADD PARTITION (dt = '2014-01-01');
""")
self.assertEquals(query, expected_query)
def test_query_with_date_interval(self):
task = ImportLastCountryOfUserToHiveTask(**self._get_kwargs())
query = task.query()
expected_query = textwrap.dedent(
"""
USE default;
DROP TABLE IF EXISTS last_country_of_user;
CREATE EXTERNAL TABLE last_country_of_user (
country_name STRING,country_code STRING,username STRING
)
PARTITIONED BY (dt STRING)
ROW FORMAT DELIMITED FIELDS TERMINATED BY '\t'
LOCATION 's3://output/path';
ALTER TABLE last_country_of_user ADD PARTITION (dt = '2014-01-01');
"""
)
self.assertEquals(query, expected_query)
