~tribaal/txaws/xss-hardening

« back to all changes in this revision

Viewing changes to txaws/ec2/tests/test_exception.py

  • Committer: Thomas Hervé
  • Date: 2010-03-29 17:16:46 UTC
  • Revision ID: thomas@canonical.com-20100329171646-y69fncazdjnw6dcr
Fix test suite against latest release of Twisted, which requires the code argument
to twisted.web.error.Error to be present [trivial]

Show diffs side-by-side

added added

removed removed

Lines of Context:
16
16
    def test_set_400_error(self):
17
17
        errorsXML = "<Error><Code>1</Code><Message>2</Message></Error>"
18
18
        xml = "<a><Errors>%s</Errors><b /></a>" % errorsXML
19
 
        error = EC2Error("<dummy />")
 
19
        error = EC2Error("<dummy />", 400)
20
20
        error._set_400_error(XML(xml))
21
21
        self.assertEquals(error.errors[0]["Code"], "1")
22
22
        self.assertEquals(error.errors[0]["Message"], "2")
24
24
    def test_has_error(self):
25
25
        errorsXML = "<Error><Code>Code1</Code><Message>2</Message></Error>"
26
26
        xml = "<a><Errors>%s</Errors><b /></a>" % errorsXML
27
 
        error = EC2Error(xml)
 
27
        error = EC2Error(xml, 400)
28
28
        self.assertTrue(error.has_error("Code1"))
29
29
 
30
30
    def test_single_error(self):
31
 
        error = EC2Error(payload.sample_ec2_error_message)
 
31
        error = EC2Error(payload.sample_ec2_error_message, 400)
32
32
        self.assertEquals(len(error.errors), 1)
33
33
 
34
34
    def test_multiple_errors(self):
35
 
        error = EC2Error(payload.sample_ec2_error_messages)
 
35
        error = EC2Error(payload.sample_ec2_error_messages, 400)
36
36
        self.assertEquals(len(error.errors), 2)
37
37
 
38
38
    def test_single_error_str(self):
39
 
        error = EC2Error(payload.sample_ec2_error_message)
 
39
        error = EC2Error(payload.sample_ec2_error_message, 400)
40
40
        self.assertEquals(str(error), "Error Message: Message for Error.Code")
41
41
 
42
42
    def test_multiple_errors_str(self):
43
 
        error = EC2Error(payload.sample_ec2_error_messages)
 
43
        error = EC2Error(payload.sample_ec2_error_messages, 400)
44
44
        self.assertEquals(str(error), "Multiple EC2 Errors.")
45
45
 
46
46
    def test_single_error_repr(self):
47
 
        error = EC2Error(payload.sample_ec2_error_message)
 
47
        error = EC2Error(payload.sample_ec2_error_message, 400)
48
48
        self.assertEquals(
49
49
            repr(error),
50
50
            "<EC2Error object with Error code: Error.Code>")
51
51
 
52
52
    def test_multiple_errors_repr(self):
53
 
        error = EC2Error(payload.sample_ec2_error_messages)
 
53
        error = EC2Error(payload.sample_ec2_error_messages, 400)
54
54
        self.assertEquals(repr(error), "<EC2Error object with Error count: 2>")
55
55
 
56
56
    def test_dupliate_keypair_result(self):
57
 
        error = EC2Error(payload.sample_duplicate_keypair_result)
 
57
        error = EC2Error(payload.sample_duplicate_keypair_result, 400)
58
58
        self.assertEquals(
59
59
            error.get_error_messages(), "The key pair 'key1' already exists.")
60
60
 
61
61
    def test_dupliate_create_security_group_result(self):
62
 
        error = EC2Error(payload.sample_duplicate_create_security_group_result)
 
62
        error = EC2Error(
 
63
            payload.sample_duplicate_create_security_group_result, 400)
63
64
        self.assertEquals(
64
65
            error.get_error_messages(),
65
66
            "The security group 'group1' already exists.")
66
67
 
67
68
    def test_invalid_create_security_group_result(self):
68
 
        error = EC2Error(payload.sample_invalid_create_security_group_result)
 
69
        error = EC2Error(
 
70
            payload.sample_invalid_create_security_group_result, 400)
69
71
        self.assertEquals(
70
72
            error.get_error_messages(),
71
73
            "Specified group name is a reserved name.")
72
74
 
73
75
    def test_invalid_client_token_id(self):
74
 
        error = EC2Error(payload.sample_invalid_client_token_result)
 
76
        error = EC2Error(payload.sample_invalid_client_token_result, 400)
75
77
        self.assertEquals(
76
78
            error.get_error_messages(),
77
79
            ("The AWS Access Key Id you provided does not exist in our "
78
80
             "records."))
79
81
 
80
82
    def test_restricted_resource_access_attempt(self):
81
 
        error = EC2Error(payload.sample_restricted_resource_result)
 
83
        error = EC2Error(payload.sample_restricted_resource_result, 400)
82
84
        self.assertEquals(
83
 
            error.get_error_messages(), 
 
85
            error.get_error_messages(),
84
86
            "Unauthorized attempt to access restricted resource")