예제 #1
0
class DataServerTestCase(unittest.TestCase):

    def setUp(self):
        config_path = os.path.abspath(os.path.join(
            os.path.dirname(__file__), "config.yaml"))
        if not os.path.isfile(config_path):
            self.raiseConfigException(config_path)
        config = yaml.load(open(config_path, 'r').read())
        if not "aws_access_key_id" in config or "aws_secret_access_key" not in config:
            self.raiseConfigException(config_path)
        self.uuid = hashlib.sha256("%s%s%s" % (
            config["aws_access_key_id"],
            config["aws_secret_access_key"], 
            self.__class__.__name__)).hexdigest()
        self.aws_access_key_id = config["aws_access_key_id"]
        self.aws_secret_access_key = config["aws_secret_access_key"]
        self.aws_s3_storage_bucket = "%s_storage" % self.uuid
        self.aws_sdb_reservation_domain = "%s_reservation" % self.uuid
        self.dataserver = DataServer( 
            aws_access_key_id = self.aws_access_key_id, 
            aws_secret_access_key = self.aws_secret_access_key,
            aws_s3_storage_bucket = self.aws_s3_storage_bucket, 
            aws_sdb_reservation_domain = self.aws_sdb_reservation_domain,
            port = 5001
        )
        return self.dataserver.start()
    
    def tearDown(self):
        deferreds = []  
        deferreds.append(self.dataserver.shutdown())
        d = DeferredList(deferreds)
        d.addCallback(self._tearDownCallback)
        return d
        
    def _tearDownCallback(self, data):
        deferreds = []        
        deferreds.append(self.dataserver.clearStorage())  
        d = DeferredList(deferreds)
        d.addCallback(self._tearDownCallback2)
        return d

    def _tearDownCallback2(self, data):
        s3 = AmazonS3(self.aws_access_key_id, self.aws_secret_access_key)
        deferreds = []        
        deferreds.append(s3.deleteBucket(self.aws_s3_storage_bucket))        
        d = DeferredList(deferreds)
        return d

    def test_01_clearStorage(self):
        d = self.dataserver.clearStorage()
        return d 
예제 #2
0
 def setUp(self):
     config_path = os.path.abspath(os.path.join(
         os.path.dirname(__file__), "config.yaml"))
     if not os.path.isfile(config_path):
         self.raiseConfigException(config_path)
     config = yaml.load(open(config_path, 'r').read())
     if not "aws_access_key_id" in config or "aws_secret_access_key" not in config:
         self.raiseConfigException(config_path)
     self.uuid = hashlib.sha256("%s%s%s" % (
         config["aws_access_key_id"],
         config["aws_secret_access_key"], 
         self.__class__.__name__)).hexdigest()
     self.aws_access_key_id = config["aws_access_key_id"]
     self.aws_secret_access_key = config["aws_secret_access_key"]
     self.aws_s3_storage_bucket = "%s_storage" % self.uuid
     self.aws_sdb_reservation_domain = "%s_reservation" % self.uuid
     self.dataserver = DataServer( 
         aws_access_key_id = self.aws_access_key_id, 
         aws_secret_access_key = self.aws_secret_access_key,
         aws_s3_storage_bucket = self.aws_s3_storage_bucket, 
         aws_sdb_reservation_domain = self.aws_sdb_reservation_domain, 
         port = 5001
     )