def saveCredentials(self, creds: Creds):
if not creds:
if os.path.exists(self.config.get(Setting.CREDENTIALS_FILE_PATH)):
os.remove(self.config.get(Setting.CREDENTIALS_FILE_PATH))
self.creds = None
return
with open(self.config.get(Setting.CREDENTIALS_FILE_PATH), "w") as f:
json.dump(creds.serialize(), f)
self.tryLoadCredentials()
async def injector(cleandir, ports, generate_config):
drive_creds = Creds(FakeTime(), "test_client_id", None, "test_access_token", "test_refresh_token")
with open(os.path.join(cleandir, "secrets.yaml"), "w") as f:
f.write("for_unit_tests: \"password value\"\n")
with open(os.path.join(cleandir, "credentials.dat"), "w") as f:
f.write(json.dumps(drive_creds.serialize()))
return Injector([BaseModule(), TestModule(generate_config, ports)])
async def test_token_extra_server(reader: ReaderHelper, coord: Coordinator, ha, drive: DriveSource, restarter, time):
update = {
"config": {
"expose_extra_server": True
},
"snapshot_folder": "unused"
}
assert await reader.postjson("saveconfig", json=update) == {'message': 'Settings saved', "reload_page": False}
await restarter.waitForRestart()
creds = Creds(time, "id", time.now(), "token", "refresh")
serialized = str(base64.b64encode(json.dumps(creds.serialize()).encode("utf-8")), "utf-8")
await reader.get("token?creds={0}&host={1}".format(quote(serialized), quote(reader.getUrl(False))), ingress=False)
assert drive.drivebackend.creds.access_token == 'token'
async def test_token_extra_server(reader: ReaderHelper, coord: Coordinator, ha,
drive: DriveSource, restarter, time):
update = {
"config": {
"expose_extra_server": True
},
"snapshot_folder": "unused"
}
assert await reader.postjson("saveconfig", json=update) == {
'message': 'Settings saved'
}
await restarter.waitForRestart()
creds = Creds(time, "id", time.now(), "token", "refresh")
resp = await reader.get("token?creds=" +
quote(json.dumps(creds.serialize())),
ingress=False)
assert "window.location.assign(\"/\")" in resp
async def injector(cleandir, server_url, ui_port, ingress_port):
drive_creds = Creds(FakeTime(), "test_client_id", None,
"test_access_token", "test_refresh_token")
with open(os.path.join(cleandir, "secrets.yaml"), "w") as f:
f.write("for_unit_tests: \"password value\"\n")
with open(os.path.join(cleandir, "credentials.dat"), "w") as f:
f.write(json.dumps(drive_creds.serialize()))
config = Config.withOverrides({
Setting.DRIVE_URL: server_url,
Setting.HASSIO_URL: server_url + "/",
Setting.HOME_ASSISTANT_URL: server_url + "/homeassistant/api/",
Setting.AUTHENTICATE_URL: server_url + "/drive/authorize",
Setting.DRIVE_REFRESH_URL: server_url + "/oauth2/v4/token",
Setting.DRIVE_AUTHORIZE_URL: server_url + "/o/oauth2/v2/auth",
Setting.DRIVE_TOKEN_URL: server_url + "/token",
Setting.REFRESH_URL: server_url + "/drive/refresh",
Setting.ERROR_REPORT_URL: server_url + "/logerror",
Setting.HASSIO_TOKEN: "test_header",
Setting.SECRETS_FILE_PATH: "secrets.yaml",
Setting.CREDENTIALS_FILE_PATH: "credentials.dat",
Setting.FOLDER_FILE_PATH: "folder.dat",
Setting.RETAINED_FILE_PATH: "retained.json",
Setting.ID_FILE_PATH: "id.json",
Setting.INGRESS_TOKEN_FILE_PATH: "ingress.dat",
Setting.DEFAULT_DRIVE_CLIENT_ID: "test_client_id",
Setting.DEFAULT_DRIVE_CLIENT_SECRET: "test_client_secret",
Setting.BACKUP_DIRECTORY_PATH: cleandir,
Setting.PORT: ui_port,
Setting.INGRESS_PORT: ingress_port
})
# PROBLEM: Something in uploading snapshot chunks hangs between the client and server, so his keeps tests from
# taking waaaaaaaay too long. Remove this line and the @pytest.mark.flaky annotations once the problem is identified.
config.override(Setting.GOOGLE_DRIVE_TIMEOUT_SECONDS, 5)
# logging.getLogger('injector').setLevel(logging.DEBUG)
return Injector([
BaseModule(config),
TestModule(cleandir, server_url, ui_port, ingress_port)
])