1#!/usr/bin/env python3
2from sys import exit
3from test.http_test import HTTPTest
4from misc.wget_file import WgetFile
5
6"""
7    This test ensures that Wget's cookie jar support works correctly.
8"""
9############# File Definitions ###############################################
10File1 = """All happy families are alike;
11Each unhappy family is unhappy in its own way"""
12File2 = "Anyone for chocochip cookies?"
13
14File1_rules = {
15    "SendHeader"        : {
16        "Set-Cookie"    : "sess-id=0213; path=/"
17    }
18}
19File2_rules = {
20    "ExpectHeader"      : {
21        "Cookie"        : "sess-id=0213"
22    }
23}
24
25A_File = WgetFile ("File1", File1, rules=File1_rules)
26B_File = WgetFile ("File2", File2, rules=File2_rules)
27
28WGET_OPTIONS = ""
29WGET_URLS = [["File1", "File2"]]
30
31Files = [[A_File, B_File]]
32
33ExpectedReturnCode = 0
34ExpectedDownloadedFiles = [A_File, B_File]
35
36################ Pre and Post Test Hooks #####################################
37pre_test = {
38    "ServerFiles"       : Files
39}
40test_options = {
41    "WgetCommands"      : WGET_OPTIONS,
42    "Urls"              : WGET_URLS
43}
44post_test = {
45    "ExpectedFiles"     : ExpectedDownloadedFiles,
46    "ExpectedRetcode"   : ExpectedReturnCode
47}
48
49err = HTTPTest (
50                pre_hook=pre_test,
51                test_params=test_options,
52                post_hook=post_test
53).begin ()
54
55exit (err)
56