def test_validate_password_returns_false(self):
""" Evaluate validate_password() returns false
:return: false
:rtype: boolean
"""
p = password_client.generate_password()
res = password_client.validate_password("wrongpassword", p["hash"])
self.assertTrue(res is False)
def test_validate_password_returns_true(self):
""" Evaluate validate_password() returns true
:return: true
:rtype: boolean
"""
p = password_client.generate_password()
res = password_client.validate_password(p["password"], p["hash"])
self.assertTrue(res is True)
def test_generate_password_returns_password_and_hash_as_strings(self):
""" Evaluate generate_password() returns password and hash as string object
:return: password and hash object as dict
:rsample: {"password": "******", "hash": "string"}
"""
res = password_client.generate_password()
self.assertTrue(type(res["password"]) is str)
self.assertTrue(type(res["hash"]) is str)
def test_generate_password_returns_dict_object(self):
""" Evaluate generate_password() returns dict object
:return: password and hash object as dict
:rsample: {"password": "******", "hash": "string"}
:rtype: dict
"""
res = password_client.generate_password()
self.assertTrue(type(res) is dict)
def test_generate_password_returns_password_and_hash(self):
""" Evaluate generate_password() returns password and hash
:return: password and hash object as dict
:rsample: {"password": "******", "hash": "string"}
"""
res = password_client.generate_password()
self.assertTrue("password" in res)
self.assertTrue("password" in res)
self.assertTrue("hash" in res)
def main(req: func.HttpRequest,
outdoc: func.Out[func.Document]) -> func.HttpResponse:
logging.info('ProtectMasterBot caught GitHub Apps callback.')
# Get and set params from the request
try:
installation_id = req.params.get('installation_id')
except (KeyError, ValueError):
logging.error('Params were not valid')
return func.HttpResponse("Sorry, something went wrong!",
status_code=500)
# Get and set params from environment variables
try:
app_id = os.environ["gh_app_id"]
app_pem = os.environ["gh_app_pem"]
except (KeyError, ValueError):
logging.error('Could not read environment variables')
return func.HttpResponse("Sorry, something went wrong!",
status_code=500)
gh_token_client = github_token_client.GitHubTokenClient(
app_id, app_pem, installation_id)
gh_org = gh_token_client.get_installation()
# Generate password
credentials = password_client.generate_password()
password = credentials["password"]
hash = credentials["hash"]
try:
# Data to store in CosmosDB
outdata = {
"id": gh_org,
"installation_id": installation_id,
"protection_json": "",
"mention": "",
"password_hash": hash
}
outdoc.set(func.Document.from_json(json.dumps(outdata)))
except:
logging.error('Could not store data in CosmosDB')
return func.HttpResponse("Sorry, something went wrong!",
status_code=500)
try:
text = f"""
<html><head>
<link rel="stylesheet" href="https://stackpath.bootstrapcdn.com/bootstrap/4.3.1/css/bootstrap.min.css" integrity="sha384-ggOyR0iXCbMQv3Xipma34MD+dH/1fQ784/j6cY/iJTQUOhcWr7x9JvoRxT2MZw1T" crossorigin="anonymous">
<script src="https://code.jquery.com/jquery-3.3.1.slim.min.js" integrity="sha384-q8i/X+965DzO0rT7abK41JStQIAqVgRVzpbzo5smXKp4YfRvH+8abtTE1Pi6jizo" crossorigin="anonymous"></script>
<scrsrc="https://cdnjs.cloudflare.com/ajax/libs/popper.js/1.14.7/umd/popper.min.js" integrity="sha384-UO2eT0CpHqdSJQ6hJty5KVphtPhzWj9WO1clHTMGa3JDZwrnQq4sF86dIHNDz0W1" crossorigin="anonymous"></scrsrc=>
<scrsrc="https://stackpath.bootstrapcdn.com/bootstrap/4.3.1/js/bootstrap.min.js" integrity="sha384-JjSmVgyd0p3pXB1rRibZUAYoIIy6OrQ6VrjIEaFf/nJGzIxFDsf4x0xIM+B07jRM" crossorigin="anonymous"></scrsrc=>
</head>
<body>
<center>
<div style="font-size:17px;max-width:800px;margin:40px auto;">
<h1>ProtectMasterBot</h1>
<h4>The branch protection is applied to your organization <strong>"{ gh_org }"</strong>.</h4>
<p>Please note the below password to configure setting. </p>
<p>The password is only shown here <strong>ONCE</strong>.</p>
<p>DO NOT share this password with others.</p>
<div style="background-color:#efefef;border-radius:5px;padding:10px;width:600px;">
<h4>Password: { password }</h4>
</div>
<hr>
<a href="/api/EditRule?org={gh_org}&password={password}"><div class="btn btn-primary"> Go To Setting </div></a>
</div>
</center>
<body></html>
"""
return func.HttpResponse(text,
status_code=200,
headers={"Content-Type": "text/html"})
except:
return func.HttpResponse("Sorry, something went wrong!",
status_code=500)