def test_run(self): zookeeper = flexmock() ds_factory = flexmock(appscale_datastore_batch.DatastoreFactory) ds_factory.should_receive("getDatastore").and_return(FakeDatastore()) flexmock(datastore_distributed).should_receive( 'DatastoreDistributed').and_return() fake_restore = flexmock( DatastoreRestore('app_id', 'backup/dir', zookeeper, "cassandra")) # Test with failure to get the restore lock. fake_restore.should_receive('get_restore_lock').and_return(False) flexmock(time).should_receive('sleep').and_return() # Test with successfully obtaining the restore lock. fake_restore.should_receive('get_restore_lock').and_return('some/path') fake_restore.should_receive('run_restore').and_return() # ... and successfully releasing the lock. zookeeper.should_receive('release_lock_with_path').and_return(True) self.assertEquals(None, fake_restore.run()) # ... and failure to release the lock. zookeeper.should_receive('release_lock_with_path').\ and_raise(ZKTransactionException) self.assertEquals(None, fake_restore.run())
def test_get_restore_lock(self): zookeeper = flexmock() zookeeper.should_receive("get_lock_with_path").and_return(True) fake_restore = flexmock( DatastoreRestore('app_id', 'backup/dir', zookeeper, "cassandra")) # Test with successfully obtaining the restore lock. self.assertEquals(True, fake_restore.get_restore_lock())
def test_run_restore(self): zookeeper = flexmock() fake_restore = flexmock( DatastoreRestore('app_id', 'backup/dir', zookeeper, "cassandra")) flexmock(glob).should_receive('glob').and_return(['some/file.backup']) fake_restore.should_receive('read_from_file_and_restore').and_return() fake_restore.run_restore()
def test_init(self): zookeeper = flexmock() fake_restore = flexmock( DatastoreRestore('app_id', 'backup/dir', zookeeper, "cassandra"))