def new(self, fqdn): """Generates a new token for a fqdn if it does not yet exist and return it. This function will return False if the fqdn is invalid or if the token already exists :param fqdn: Fully-Qualified Domain-Name for the host :type fqdn: str :returns: Token for the new host, or False if an error occurred :rtype: str, bool """ if not checks.valid_fqdn(fqdn): return False if self.get(fqdn): log.warning('Token for {0} already exists'.format(fqdn)) return False token = utils.gentoken() self._store[fqdn] = token self.save() return token
def setUp(self): if os.path.exists(STORE): os.unlink(STORE) self.data = {TEST_HOST: utils.gentoken()} open(STORE, 'w').write(json.dumps(self.data)) self.store = tokens.TokenStore(STORE)
def test_valid_fields(self): data = {TEST_HOST: utils.gentoken()} assert self.store.validate_store(data) is True
def test_fqdn_field_invalid(self): data = {12345: utils.gentoken()} assert self.store.validate_store(data) is False