~tribaal/txaws/xss-hardening

« back to all changes in this revision

Viewing changes to txaws/tests/test_credentials.py

  • Committer: Tristan Seligmann
  • Date: 2009-04-27 23:53:32 UTC
  • mfrom: (3.1.5 client)
  • Revision ID: mithrandi@mithrandi.net-20090427235332-l8azkl1kvn5pjtay
mergeĀ lp:~lifeless/txaws/client

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2009 Robert Collins <robertc@robertcollins.net>
 
2
# Licenced under the txaws licence available at /LICENSE in the txaws source.
 
3
 
 
4
import os
 
5
 
 
6
from twisted.trial.unittest import TestCase
 
7
from txaws.tests import TXAWSTestCase
 
8
 
 
9
from txaws.credentials import AWSCredentials
 
10
 
 
11
 
 
12
class TestCredentials(TXAWSTestCase):
 
13
 
 
14
    def test_no_access_errors(self):
 
15
        # Without anything in os.environ, AWSCredentials() blows up
 
16
        os.environ['AWS_SECRET_ACCESS_KEY'] = 'foo'
 
17
        self.assertRaises(Exception, AWSCredentials)
 
18
 
 
19
    def test_no_secret_errors(self):
 
20
        # Without anything in os.environ, AWSCredentials() blows up
 
21
        os.environ['AWS_ACCESS_KEY_ID'] = 'bar'
 
22
        self.assertRaises(Exception, AWSCredentials)
 
23
 
 
24
    def test_found_values_used(self):
 
25
        os.environ['AWS_SECRET_ACCESS_KEY'] = 'foo'
 
26
        os.environ['AWS_ACCESS_KEY_ID'] = 'bar'
 
27
        creds = AWSCredentials()
 
28
        self.assertEqual('foo', creds.secret_key)
 
29
        self.assertEqual('bar', creds.access_key)
 
30
 
 
31
    def test_explicit_access_key(self):
 
32
        os.environ['AWS_SECRET_ACCESS_KEY'] = 'foo'
 
33
        creds = AWSCredentials(access_key='bar')
 
34
        self.assertEqual('foo', creds.secret_key)
 
35
        self.assertEqual('bar', creds.access_key)
 
36
 
 
37
    def test_explicit_secret_key(self):
 
38
        os.environ['AWS_ACCESS_KEY_ID'] = 'bar'
 
39
        creds = AWSCredentials(secret_key='foo')
 
40
        self.assertEqual('foo', creds.secret_key)
 
41
        self.assertEqual('bar', creds.access_key)