def test_set_target_can_complete_with_http(self):
ctrl = self.make_controller(TargetSetController, conf=self.fixture_for('test.conf'), target='my-target.wight.com')
ctrl.default()
ud = UserData.load()
expect(ud).not_to_be_null()
expect(ud.target).to_equal("http://my-target.wight.com")
def test_can_set_team_and_project(self):
result = self.execute("default-set", team=self.team, project=self.project)
expected = "Default team set to '%s'. Default project set to '%s'." % (self.team, self.project)
expect(result).to_be_like(expected)
ud = UserData.load()
expect(ud.team).to_be_like(self.team)
expect(ud.project).to_be_like(self.project)
def setUp(self):
self.clear_user_data()
self.target = "http://localhost:2368"
self.execute("target-set", self.target)
user_data = UserData.load()
user_data.token = "token-value"
user_data.save()
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_set_target_can_complete_with_http(self):
ctrl = self.make_controller(TargetSetController,
conf=self.fixture_for('test.conf'),
target='my-target.wight.com')
ctrl.default()
ud = UserData.load()
expect(ud).not_to_be_null()
expect(ud.target).to_equal("http://my-target.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_can_set_team_and_project(self):
result = self.execute("default-set",
team=self.team,
project=self.project)
expected = "Default team set to '%s'. Default project set to '%s'." % (
self.team, self.project)
expect(result).to_be_like(expected)
ud = UserData.load()
expect(ud.team).to_be_like(self.team)
expect(ud.project).to_be_like(self.project)
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 default(self):
self.load_conf()
user_data = UserData.load()
if user_data is None:
self.write("No target set.")
return
self.write("Current target set to '%s'." % user_data.target)
def test_can_set_target(self):
target = "http://my-test-target:2324"
# Set target to acc target
result = self.execute("target-set", target)
expected = "Wight target set to '%s'. In order to login with wight, use 'wight login <email>'."
expect(result).to_be_like(expected % target)
ud = UserData.load()
expect(ud.target).to_be_like(target)
def test_can_set_target(self):
target = "http://my-test-target:2324"
# Set target to acc target
result = self.execute("target-set", target)
expected = "Wight target set to '%s'. In order to login with wight, use 'wight login <email>'."
expect(result).to_be_like(expected % target)
ud = UserData.load()
expect(ud.target).to_be_like(target)
def test_set_project(self, mock_stdout):
ctrl = self.make_controller(
SetDefaultController, conf=self.fixture_for("test.conf"), team=None, project=self.project
)
self.authenticate(ctrl)
ctrl.default()
ud = UserData.load()
expect(hasattr(ud, "project")).to_be_true()
expect(ud.project).to_equal(self.project)
expect(hasattr(ud, "team")).to_be_false()
expect(mock_stdout.getvalue()).to_be_like("Default team not set. Default project set to '%s'." % self.project)
def test_set_team_and_project(self, mock_stdout):
ctrl = self.make_controller(
SetDefaultController, conf=self.fixture_for("test.conf"), team=self.team, project=self.project
)
self.authenticate(ctrl)
ctrl.default()
ud = UserData.load()
expect(ud.team).to_equal(self.team)
expect(ud.project).to_equal(self.project)
expect(mock_stdout.getvalue()).to_be_like(
"Default team set to '%s'. Default project set to '%s'." % (self.team, self.project)
)
def test_set_team_and_project(self, mock_stdout):
ctrl = self.make_controller(SetDefaultController,
conf=self.fixture_for('test.conf'),
team=self.team,
project=self.project)
self.authenticate(ctrl)
ctrl.default()
ud = UserData.load()
expect(ud.team).to_equal(self.team)
expect(ud.project).to_equal(self.project)
expect(mock_stdout.getvalue()).to_be_like(
"Default team set to '%s'. Default project set to '%s'." %
(self.team, self.project))
def test_set_team(self, mock_stdout):
ctrl = self.make_controller(SetDefaultController,
conf=self.fixture_for('test.conf'),
team=self.team,
project=None)
self.authenticate(ctrl)
ctrl.default()
ud = UserData.load()
expect(hasattr(ud, "team")).to_be_true()
expect(ud.team).to_equal(self.team)
expect(hasattr(ud, "project")).to_be_false()
expect(mock_stdout.getvalue()).to_be_like(
"Default team set to '%s'. Default project not set." % self.team)
def default(self):
self.load_conf()
user_data = UserData.load()
if user_data is None:
self.abort("No target set.")
return
self.line_break()
self.putsuccess("Current Wight target is '%s%s%s'. In order to login with wight, use '%swight login <email>%s'." % (
self.keyword_color, user_data.target, self.reset_success,
self.commands_color, self.reset_success
))
self.line_break()
def default(self):
self.load_conf()
user_data = UserData.load()
if user_data is None:
self.abort("No target set.")
return
self.line_break()
self.putsuccess(
"Current Wight target is '%s%s%s'. In order to login with wight, use '%swight login <email>%s'."
% (self.keyword_color, user_data.target, self.reset_success,
self.commands_color, self.reset_success))
self.line_break()
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()
def assert_token_is(token):
data = UserData.load()
expect(data.token).to_equal(token)
def __init__(self, label=None, **kw):
super(WightApp, self).__init__(**kw)
self.user_data = UserData.load()
init(autoreset=True)
def test_can_set_team(self):
result = self.execute("default-set", team=self.team)
expected = "Default team set to '%s'. Default project not set." % self.team
expect(result).to_be_like(expected)
ud = UserData.load()
expect(ud.team).to_be_like(self.team)
def test_user_data_returns_none_from_invalid_path(self):
loaded = UserData.load("/some/invalid/path")
expect(loaded).to_be_null()
def test_can_set_project(self):
result = self.execute("default-set", project=self.project)
expected = "Default team not set. Default project set to '%s'." % self.project
expect(result).to_be_like(expected)
ud = UserData.load()
expect(ud.project).to_be_like(self.project)
def test_can_set_team(self):
result = self.execute("default-set", team=self.team)
expected = "Default team set to '%s'. Default project not set." % self.team
expect(result).to_be_like(expected)
ud = UserData.load()
expect(ud.team).to_be_like(self.team)
def test_can_set_project(self):
result = self.execute("default-set", project=self.project)
expected = "Default team not set. Default project set to '%s'." % self.project
expect(result).to_be_like(expected)
ud = UserData.load()
expect(ud.project).to_be_like(self.project)
def test_user_data_returns_none_from_invalid_path(self):
loaded = UserData.load("/some/invalid/path")
expect(loaded).to_be_null()
def assert_token_is(token):
data = UserData.load()
expect(data.token).to_equal(token)
