def setUpClass(cls):
index_catalog(
'test_catalog',
os.path.join(samples_dir, 'distribution_daily_periodicity.json'),
cls.index)
enqueue_write_csv_task()
super(DtaGeneratorTests, cls).setUpClass()
def setUpClass(cls):
super(CSVTest, cls).setUpClass()
cls.catalog_id = 'csv_dump_test_catalog'
path = os.path.join(samples_dir, 'distribution_daily_periodicity.json')
index_catalog(cls.catalog_id, path, cls.index)
cls.task = GenerateDumpTask()
cls.task.save()
gen = DumpGenerator(cls.task)
gen.generate()
def setUp(self):
GenerateDumpTask.objects.all().delete()
DumpFile.objects.all().delete()
path = os.path.join(samples_dir, 'distribution_daily_periodicity.json')
index_catalog('catalog_one', path, index=self.index)
path = os.path.join(samples_dir, 'leading_nulls_distribution.json')
index_catalog('catalog_two', path, index=self.index)
def setUpClass(cls):
super(XLSXGeneratorTests, cls).setUpClass()
GenerateDumpTask.objects.all().delete()
DumpFile.objects.all().delete()
path = os.path.join(samples_dir, 'distribution_daily_periodicity.json')
index_catalog('catalog_one', path, index=cls.index)
path = os.path.join(samples_dir, 'leading_nulls_distribution.json')
index_catalog('catalog_two', path, index=cls.index)
call_command('generate_dump')
def setUpClass(cls):
super(SQLTests, cls).setUpClass()
DumpFile.objects.all().delete()
Node.objects.all().delete()
Catalog.objects.all().delete()
GenerateDumpTask.objects.all().delete()
path = os.path.join(samples_dir, 'distribution_daily_periodicity.json')
index_catalog('catalog_one', path, index=cls.index)
call_command('generate_dump')
task = GenerateDumpTask.objects.create()
SQLGenerator(task.id).generate()
def test_leading_nulls_distribution(self):
path = os.path.join(samples_dir, 'leading_nulls_distribution.json')
index_catalog('leading_null', path, self.index)
self.task = GenerateDumpTask()
self.task.save()
gen = DumpGenerator(self.task, 'leading_null')
gen.generate()
file = self.task.dumpfile_set.get(file_name=DumpFile.FILENAME_FULL,
file_type=DumpFile.TYPE_CSV,
node__catalog_id='leading_null').file
reader = read_file_as_csv(file)
next(reader) # Header!!!!
self.assertEqual(len(list(reader)), 1) # Un único row, para un único valor del CSV
def setUpClass(cls):
super(ViewTests, cls).setUpClass()
es_client = ElasticInstance.get()
if es_client.indices.exists(cls.index):
es_client.indices.delete(cls.index)
es_client.indices.create(cls.index, body=INDEX_CREATION_BODY)
cls.catalog_id = 'csv_dump_test_catalog'
path = os.path.join(samples_dir, 'distribution_daily_periodicity.json')
index_catalog(cls.catalog_id, path, cls.index)
cls.task = GenerateDumpTask()
cls.task.save()
gen = DumpGenerator(cls.task)
gen.generate()
DumpGenerator(cls.task, cls.catalog_id).generate()
def _index_catalog(self, catalog_path, node=None):
path = os.path.join(SAMPLES_DIR, catalog_path)
utils.index_catalog(CATALOG_ID, path, self.test_index, node=node)