def testCreateCubeSet(self): cubeSetName = 'test-' + str(uuid.uuid4()) csvFileName = cubeSetName + '.csv' shutil.copyfile('testdata.csv', cubeSetName + '.csv') with open('test_binnings.json') as binnings_file: binnings = json.load(binnings_file) with open('test_agg.json') as agg_file: aggs = json.load(agg_file) cs = CubeSetService('testdb') cs.createCubeSet("testOwner", cubeSetName, cubeSetName, csvFileName, binnings, aggs) cubeSet = cs.getCubeSet(cubeSetName) self.assertTrue(cubeSet != None) self.assertTrue(cubeSet['name'] == cubeSetName) self.assertTrue(cubeSet['displayName'] == cubeSetName) self.assertTrue(cubeSet['owner'] == 'testOwner') self.assertTrue(cubeSet['csvFileName'] == csvFileName) self.assertTrue(cubeSet['sourceCube'] == cubeSetName + '_source') self.assertTrue(cubeSet['binnedCube'] == cubeSetName + '_binned') binnedCubeName = cubeSet['binnedCube'] self.assertTrue(len(cubeSet['aggCubes']) == 3) self.assertTrue(binnedCubeName + '_' + aggs[0]['name'] in cubeSet['aggCubes']) self.assertTrue(binnedCubeName + '_' + aggs[1]['name'] in cubeSet['aggCubes']) self.assertTrue(binnedCubeName + '_' + aggs[2]['name'] in cubeSet['aggCubes']) os.remove(csvFileName)
def testDeleteCubeSet(self): cubeSetName = 'test-' + str(uuid.uuid4()) csvFileName = cubeSetName + '.csv' shutil.copyfile('testdata.csv', cubeSetName + '.csv') cs = CubeSetService('testdb') cs.createCubeSet("testOwner", cubeSetName, cubeSetName, csvFileName, None, None) cs.deleteCubeSet(cubeSetName) cubeSet = cs.getCubeSet(cubeSetName) self.assertTrue(cubeSet == None) os.remove(csvFileName)
def testDeleteCubeSet2(self): cubeSetName = 'test-' + str(uuid.uuid4()) csvFileName = cubeSetName + '.csv' shutil.copyfile('testdata.csv', cubeSetName + '.csv') with open('test_binnings.json') as binnings_file: binnings = json.load(binnings_file) with open('test_agg.json') as agg_file: aggs = json.load(agg_file) cs = CubeSetService('testdb') cs.createCubeSet("testOwner", cubeSetName, cubeSetName, csvFileName, binnings, aggs) cs.deleteCubeSet(cubeSetName) cubeSet = cs.getCubeSet(cubeSetName) self.assertTrue(cubeSet == None) os.remove(csvFileName)
def testUpdateCubeSetDisplayName(self): cubeSetName = 'test-' + str(uuid.uuid4()) csvFileName = cubeSetName + '.csv' shutil.copyfile('testdata.csv', cubeSetName + '.csv') cs = CubeSetService('testdb') cs.createCubeSet("testOwner", cubeSetName, cubeSetName, csvFileName, None, None) cs.updateCubeSetDisplayName(cubeSetName, 'new name') cubeSet = cs.getCubeSet(cubeSetName) self.assertTrue(cubeSet != None) self.assertTrue(cubeSet['name'] == cubeSetName) self.assertTrue(cubeSet['displayName'] == 'new name') self.assertTrue(cubeSet['owner'] == 'testOwner') self.assertTrue(cubeSet['csvFileName'] == csvFileName) self.assertTrue(cubeSet['sourceCube'] == cubeSetName + '_source') os.remove(csvFileName)