def test_saving_with_project(self):
ud = UserData(target="target")
ud.set_default(project="project-blah")
ud.save()
loaded = UserData.load()
expect(hasattr(loaded, "project")).to_be_true()
expect(loaded.project).to_equal("project-blah")
expect(hasattr(loaded, "team")).to_be_false()
def test_saving_with_project(self):
ud = UserData(target="target")
ud.set_default(project="project-blah")
ud.save()
loaded = UserData.load()
expect(hasattr(loaded, "project")).to_be_true()
expect(loaded.project).to_equal("project-blah")
expect(hasattr(loaded, "team")).to_be_false()
def test_user_data_returns_from_default_path(self):
if exists(UserData.DEFAULT_PATH):
os.remove(UserData.DEFAULT_PATH)
ud = UserData(target="http://target3.wight.com")
ud.save()
loaded = UserData.load()
expect(loaded).not_to_be_null()
expect(loaded.target).to_equal("http://target3.wight.com")
def test_user_data_deserializes_properly(self):
ud = UserData(target="http://target2.wight.com")
directory = tempfile.mkdtemp()
path = join(directory, '.wight-user-data')
ud.save(path)
loaded = UserData.load(path)
expect(loaded.target).to_equal("http://target2.wight.com")
def test_user_data_returns_from_default_path(self):
if exists(UserData.DEFAULT_PATH):
os.remove(UserData.DEFAULT_PATH)
ud = UserData(target="http://target3.wight.com")
ud.save()
loaded = UserData.load()
expect(loaded).not_to_be_null()
expect(loaded.target).to_equal("http://target3.wight.com")
def test_user_data_deserializes_properly(self):
ud = UserData(target="http://target2.wight.com")
directory = tempfile.mkdtemp()
path = join(directory, '.wight-user-data')
ud.save(path)
loaded = UserData.load(path)
expect(loaded.target).to_equal("http://target2.wight.com")
def test_user_data_serializes_properly(self):
ud = UserData(target="http://target2.wight.com")
directory = tempfile.mkdtemp()
path = join(directory, '.wight-user-data')
ud.save(path)
with open(path, 'r') as text:
obj = loads(text.read())
expect(obj).to_include("target")
expect(obj['target']).to_equal("http://target2.wight.com")
def test_user_data_serializes_properly(self):
ud = UserData(target="http://target2.wight.com")
directory = tempfile.mkdtemp()
path = join(directory, '.wight-user-data')
ud.save(path)
with open(path, 'r') as text:
obj = loads(text.read())
expect(obj).to_include("target")
expect(obj['target']).to_equal("http://target2.wight.com")
def default(self):
self.load_conf()
target = self.arguments.target
self.line_break()
self.putsuccess("Wight target set to '%s%s%s'. In order to login with wight, use '%swight login%s'." % (
self.keyword_color, target, self.reset_success,
self.commands_color, self.reset_success
))
self.line_break()
ud = UserData.load()
if ud is None:
ud = UserData(target=target)
else:
ud.target = target
ud.save()
def default(self):
self.load_conf()
target = self.arguments.target
target_parsed = urlparse(target)
if not target_parsed.scheme:
target = "http://%s" % target_parsed.geturl()
self.line_break()
self.putsuccess("Wight target set to '%s%s%s'. In order to login with wight, use '%swight login <email>%s'." % (
self.keyword_color, target, self.reset_success,
self.commands_color, self.reset_success
))
self.line_break()
ud = UserData.load()
if ud is None:
ud = UserData(target=target)
else:
ud.target = target
ud.save()
def default(self):
self.load_conf()
target = self.arguments.target
target_parsed = urlparse(target)
if not target_parsed.scheme:
target = "http://%s" % target_parsed.geturl()
self.line_break()
self.putsuccess(
"Wight target set to '%s%s%s'. In order to login with wight, use '%swight login <email>%s'."
% (self.keyword_color, target, self.reset_success,
self.commands_color, self.reset_success))
self.line_break()
ud = UserData.load()
if ud is None:
ud = UserData(target=target)
else:
ud.target = target
ud.save()