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_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."