def test_LoginSucessWithLoginAndYesConfirm(post, getpass, raw_input, capsys):
class Response:
"""It is the post()'s return value"""
def __init__(self):
self.ok = True
def json(self):
return {"token": "token"}
loggedin()
data = {"password": TestPassword}
am = AuthManager("target", dbn)
post.return_value = Response()
am.login(TestUser, TestEmail2)
out, err = capsys.readouterr()
raw_input.assert_called_with(QHaveLoggedIn)
post.assert_called_with("target/users/{0}/tokens".format(TestEmail2), data=json.dumps(data))
assert (out.strip(), ILogin)
# confirm the cfgdb's content
with offtheshelf.openDB(dbn) as db:
users = db.get_collection("users")
x = users.find({"name": TestUser, "email": TestEmail2, "default": True})
assert x != None
assert 1 == len(x)
def test_RemoveUserWithLoginAndNoConfirm(delete, raw_input, capsys):
loggedin()
am = AuthManager("target", dbn)
am.removeUser()
out, err = capsys.readouterr()
raw_input.assert_called_with(QRemoveUser)
delete.not_called()
assert out.strip() == "Abort."
def test_GetCurrentUserWithLogin(capsys):
loggedin()
am = AuthManager("target", dbn)
x = am.getCurrentUser()
out, err = capsys.readouterr()
assert out.strip() == "Current user is: %s, email: %s" % (TestUser, TestEmail)
assert type(x) == type({})
assert x["name"] == TestUser
def test_RemoveUserWithLoginAndYesConfirm(delete, raw_input, capsys):
loggedin()
am = AuthManager("target", dbn)
# capture = py.io.StdCaptureFD(in_=False)
am.removeUser()
out, err = capsys.readouterr()
raw_input.assert_called_with(QRemoveUser)
delete.assert_called_with("target/users", headers=TestAuthHeader)
assert out.strip() == IRemoveUser
def test_LoginStatus(capsys):
am = AuthManager("target", dbn)
am.status()
out, err = capsys.readouterr()
assert out.strip() == "You're not logged in!"
loggedin()
am.status()
out, err = capsys.readouterr()
assert out.strip() == "You: %s have logged into %s !" % (TestUser, "target")
def test_CreateUser(post, getpass, capsys):
am = AuthManager("target", dbn)
# capture = py.io.StdCaptureFD(in_=False)
data = {"email": TestEmail, "password": TestPassword}
am.createUser(TestUser, TestEmail)
out, err = capsys.readouterr()
# getpass.assert_called_with("Please input your password: "******"Confirm: ")
post.assert_called_with("target/users", data=json.dumps(data))
assert out.strip() == ICreateUser % TestEmail
def test_LoginFailed(post, getpass, raw_input, capsys):
class Response:
"""It is the post()'s return value"""
def __init__(self):
self.ok = False
self.content = "Error!!"
def json(self):
return {"token": "token"}
data = {"password": TestPassword}
am = AuthManager("target", dbn)
post.return_value = Response()
am.login(TestUser, TestEmail)
out, err = capsys.readouterr()
post.assert_called_with("target/users/{0}/tokens".format(TestEmail), data=json.dumps(data))
assert out.strip() == ELogin % post.return_value.content
def test_LoginSucessWithLoginAndNoConfirm(post, getpass, raw_input, capsys):
class Response:
"""It is the post()'s return value"""
def __init__(self):
self.ok = True
def json(self):
return {"token": "token"}
loggedin()
data = {"password": TestPassword}
am = AuthManager("target", dbn)
post.return_value = Response()
am.login(TestUser, TestEmail2)
out, err = capsys.readouterr()
# out, err = capsys.reset()
raw_input.assert_called_with(QHaveLoggedIn)
post.not_called()
assert out.strip() == "Abort."
def test_RemoveUserWithoutLogin(delete, capsys):
am = AuthManager("target", dbn)
am.removeUser()
out, err = capsys.readouterr()
delete.not_called()
assert out.strip() == ELoginFirst
def test_GetCurrentUserWhthoutLogin(capsys):
am = AuthManager("target", dbn)
x = am.getCurrentUser()
out, err = capsys.readouterr()
assert out.strip() == "There is no default user."
def test_LogoutSucess(capsys):
loggedin()
am = AuthManager("target", dbn)
am.logout()
out, err = capsys.readouterr()
assert out.strip() == ILogout
