mirror of https://github.com/moparisthebest/wget
60 lines
1.4 KiB
Python
Executable File
60 lines
1.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
from sys import exit
|
|
from test.http_test import HTTPTest
|
|
from misc.wget_file import WgetFile
|
|
|
|
"""
|
|
This test ensures that Wget stores the cookie even in the event of a
|
|
401 Unauthorized Response
|
|
"""
|
|
TEST_NAME = "Basic Cookie 401 Response"
|
|
############# File Definitions ###############################################
|
|
File1 = """All happy families are alike;
|
|
Each unhappy family is unhappy in its own way"""
|
|
File2 = "Anyone for chocochip cookies?"
|
|
|
|
File1_rules = {
|
|
"SendHeader" : {
|
|
"Set-Cookie" : "sess-id=0213; path=/"
|
|
},
|
|
"Response" : 401
|
|
}
|
|
File2_rules = {
|
|
"ExpectHeader" : {
|
|
"Cookie" : "sess-id=0213"
|
|
},
|
|
}
|
|
|
|
A_File = WgetFile ("File1", File1, rules=File1_rules)
|
|
B_File = WgetFile ("File2", File2, rules=File2_rules)
|
|
|
|
WGET_OPTIONS = ""
|
|
WGET_URLS = [["File1", "File2"]]
|
|
|
|
Files = [[A_File, B_File]]
|
|
|
|
ExpectedReturnCode = 6
|
|
ExpectedDownloadedFiles = [B_File]
|
|
|
|
################ Pre and Post Test Hooks #####################################
|
|
pre_test = {
|
|
"ServerFiles" : Files
|
|
}
|
|
test_options = {
|
|
"WgetCommands" : WGET_OPTIONS,
|
|
"Urls" : WGET_URLS
|
|
}
|
|
post_test = {
|
|
"ExpectedFiles" : ExpectedDownloadedFiles,
|
|
"ExpectedRetcode" : ExpectedReturnCode
|
|
}
|
|
|
|
err = HTTPTest (
|
|
name=TEST_NAME,
|
|
pre_hook=pre_test,
|
|
test_params=test_options,
|
|
post_hook=post_test
|
|
).begin ()
|
|
|
|
exit (err)
|