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)