def TestWrite(self): self.store.Set('pig', 'grunt') self.store.Write() assert (self.store.Get('pig') == 'grunt') store2 = KV.KeyValueStore(self.temp_file, redhawk.GetVersion()) assert (store2.Get('pig') == 'grunt') store2.Close()
def __init__(self, database_file, parser, store_new=True): self.database_file = database_file self.parser = parser self.store_new = store_new # logging.info("Reading..\n") if self.database_file is None: self.database = None return if not os.path.exists(database_file): logging.warning("Database %s does not exist." % (self.database_file)) self.database = None return if not KVStore.IsValidStore(database_file): print(database_file) logging.error("Not a valid store. Recreating..\n") KVStore.RemoveExistingStore(database_file) KVStore.CreateNewStore(database_file, redhawk.GetVersion()) self.database = KVStore.KeyValueStore(database_file, redhawk.GetVersion()) # logging.info("Done.") return
def TestBadVersionInitKeyValueStore(): temp_file = GetTempFile() KV.CreateNewStore(temp_file, redhawk.GetVersion()) store = shelve.open(temp_file, 'c') store[KV.VERSION_KEY] = '0.0.0' store = KV.KeyValueStore(temp_file, redhawk.GetVersion()) return
def TestCreateAndRemoveStores(): temp_file = GetTempFile() KV.CreateNewStore(temp_file, redhawk.GetVersion()) store = KV.KeyValueStore(temp_file, redhawk.GetVersion()) assert (store.GetVersion() == redhawk.GetVersion()) store.Close() KV.RemoveExistingStore(temp_file) assert (not os.path.exists(temp_file))
def setUp(self): setUp() self.temp_file = GetTempFile() KV.CreateNewStore(self.temp_file, redhawk.GetVersion()) store = shelve.open(self.temp_file, 'c') store['dog'] = 'woof' store['cat'] = 'meow' store.close() self.store = KV.KeyValueStore(self.temp_file, redhawk.GetVersion()) return
def TestChangeVersion(self): self.store.Close() store = shelve.open(self.temp_file, 'c') store[KV.VERSION_KEY] = '0.0.0' store.close() temp_file = GetTempFile() fp = sys.stderr sys.stderr = sys.stdout print("Expected Error :") self.store = KV.KeyValueStore(self.temp_file, redhawk.GetVersion()) sys.stderr = fp assert (self.store.GetVersion() == redhawk.GetVersion())
def Main(args): parser = optparse.OptionParser(usage, description=description) options, args = parser.parse_args(args) if len(args): parser.error("Extra options given. This command takes no options!") database_file = S.GetDatabase() if database_file is None: S.ExitWithError( S.MakeStringFromTemplate( "No database found. Maybe $prog init first?")) store = KVStore.KeyValueStore(database_file, redhawk.GetVersion()) for i in store.GetKeys(): print(U.GetDBPathRelativeToCurrentDirectory(i)) store.Close() return
def Main(args): parser = optparse.OptionParser(usage, description=description) options, args = parser.parse_args(args) if not len(args): parser.error("No files given.") database_file = S.GetDatabase() if database_file is None: S.ExitWithError( S.MakeStringFromTemplate( "No database found. Maybe $prog init first?")) store = KVStore.KeyValueStore(database_file, redhawk.GetVersion()) for f in S.GetSupportedFiles(args): key = S.GetKey(f, database_file) if store.HasKey(key): print("Removing: %s" % (U.GetDBPathRelativeToCurrentDirectory(f))) store.RemoveKey(key) else: print("Not found: %s" % (U.GetDBPathRelativeToCurrentDirectory(f))) store.Close() return
def TestNiceInitKeyValueStore(): temp_file = GetTempFile() KV.CreateNewStore(temp_file, redhawk.GetVersion()) store = KV.KeyValueStore(temp_file, redhawk.GetVersion()) return
def TestRetainOnClose(self): self.store.Set('pig', 'grunt') self.store.Close() self.store = KV.KeyValueStore(self.temp_file, redhawk.GetVersion()) assert (self.store.Get('pig') == 'grunt')