def testTransfer(self):
        metrics = makeExampleMetrics()

        dataUnits = frozenset(("visit", "filter"))
        dataId = {"visit": 2048, "filter": "Uprime"}

        sc = self.storageClassFactory.getStorageClass("StructuredData")
        ref = self.makeDatasetRef("metric", dataUnits, sc, dataId)

        inputConfig = DatastoreConfig(self.configFile)
        inputConfig['datastore.root'] = os.path.join(self.testDir,
                                                     "./test_input_datastore")
        inputPosixDatastore = PosixDatastore(config=inputConfig,
                                             registry=self.registry)
        outputConfig = inputConfig.copy()
        outputConfig['datastore.root'] = os.path.join(
            self.testDir, "./test_output_datastore")
        outputPosixDatastore = PosixDatastore(config=outputConfig,
                                              registry=DummyRegistry())

        inputPosixDatastore.put(metrics, ref)
        outputPosixDatastore.transfer(inputPosixDatastore, ref)

        metricsOut = outputPosixDatastore.get(ref)
        self.assertEqual(metrics, metricsOut)
Example #2
0
    def testTransfer(self):
        catalog = self.makeExampleCatalog()
        dataUnits = frozenset(("visit", "filter"))
        dataId = {"visit": 12345, "filter": "red"}

        storageClass = self.storageClassFactory.getStorageClass(
            "SourceCatalog")
        ref = self.makeDatasetRef("calexp", dataUnits, storageClass, dataId)

        inputConfig = DatastoreConfig(self.configFile)
        inputConfig['datastore.root'] = os.path.join(self.testDir,
                                                     "./test_input_datastore")
        inputPosixDatastore = PosixDatastore(config=inputConfig,
                                             registry=self.registry)
        outputConfig = inputConfig.copy()
        outputConfig['datastore.root'] = os.path.join(
            self.testDir, "./test_output_datastore")
        outputPosixDatastore = PosixDatastore(config=outputConfig,
                                              registry=DummyRegistry())

        inputPosixDatastore.put(catalog, ref)
        outputPosixDatastore.transfer(inputPosixDatastore, ref)

        catalogOut = outputPosixDatastore.get(ref)
        self.assertCatalogEqual(catalog, catalogOut)