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
