-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
10 changed files
with
328 additions
and
44 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,5 @@ | ||
| .tox/ | ||
| local/ | ||
|
|
||
| *.egg-info/ | ||
| *.pyc |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,20 @@ | ||
| language: python | ||
| sudo: false | ||
|
|
||
| matrix: | ||
| include: | ||
| - python: '2.7' | ||
| env: TOXENV=py27 | ||
| - python: '3.3' | ||
| env: TOXENV=py33 | ||
| - python: '3.4' | ||
| env: TOXENV=py34 | ||
| - python: '3.5' | ||
| env: TOXENV=py35 | ||
| - python: '3.6' | ||
| env: TOXENV=py36 | ||
|
|
||
| install: | ||
| - pip install tox | ||
| script: | ||
| - tox -e ${TOXENV} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,3 +1,3 @@ | ||
| __all__ = ['main'] | ||
|
|
||
| from r53update import main | ||
| from .r53update import main |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Empty file.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,154 @@ | ||
| #!/bin/env python | ||
| # -*- coding: utf-8 -*- | ||
| # | ||
| # R53Update Dynamic DNS Updater | ||
| # (C)2014 Takuya Sawada All rights reserved. | ||
| # | ||
| # Licensed under the Apache License, Version 2.0 (the "License"); | ||
| # you may not use this file except in compliance with the License. | ||
| # You may obtain a copy of the License at | ||
| # | ||
| # http://www.apache.org/licenses/LICENSE-2.0 | ||
| # | ||
| # Unless required by applicable law or agreed to in writing, software | ||
| # distributed under the License is distributed on an "AS IS" BASIS, | ||
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| # See the License for the specific language governing permissions and | ||
| # limitations under the License. | ||
| # | ||
| from ..r53update import R53UpdateApp | ||
| from botocore import exceptions | ||
|
|
||
| import unittest | ||
| import string | ||
| import random | ||
| import mock | ||
| import os | ||
| import io | ||
|
|
||
| class TestAppContext(unittest.TestCase): | ||
| def setUp(self): | ||
| # keep original function for reference from mock function | ||
| self.__open__ = open | ||
| self.__isfile__ = os.path.isfile | ||
|
|
||
| try: | ||
| import __builtin__ | ||
| self.builtins = '__builtin__' | ||
| except: | ||
| self.builtins = 'builtins' | ||
|
|
||
| # key-pair test data | ||
| self.profile = { | ||
| 'env': (self.randomstr(20), self.randomstr(40)), | ||
| 'default': (self.randomstr(20), self.randomstr(40)), | ||
| 'test': (self.randomstr(20), self.randomstr(40)) | ||
| } | ||
|
|
||
| def randomstr(self, length): | ||
| return ''.join([random.choice(string.digits + string.ascii_uppercase) for i in range(12)]) | ||
|
|
||
| ## | ||
| # boto core checks if config file exists before load it. | ||
| # in order to use mock file instead of actual config file, | ||
| # we must use this mock function which always return True | ||
| # if specified path is the config file. | ||
| # | ||
| # Ref) https://github.com/boto/botocore/blob/develop/botocore/configloader.py | ||
| # | ||
| def mock_isfile(self, path): | ||
| if path == os.path.expanduser('~/.aws/config'): | ||
| return True | ||
| else: | ||
| return self.__isfile__(path) | ||
|
|
||
| def mock_file(self, path, *args, **kwargs): | ||
| if path == os.path.expanduser('~/.aws/config'): | ||
| return io.StringIO( | ||
| u"[default]\n" | ||
| u"aws_access_key_id = %s\n" | ||
| u"aws_secret_access_key = %s\n" | ||
| u"region = ap-northeast-1\n" | ||
| u"output = json\n" | ||
| u"\n" | ||
| u"[profile test]\n" | ||
| u"aws_access_key_id = %s\n" | ||
| u"aws_secret_access_key = %s\n" | ||
| u"region = ap-northeast-2\n" | ||
| u"output = json\n" | ||
|
|
||
| % (self.profile['default'] + self.profile['test']) | ||
| ) | ||
| else: | ||
| return self.__open__(path, *args, **kwargs) | ||
|
|
||
| # check credential with environment variable | ||
| def test_env(self): | ||
| AWS_ACCESS_KEY, AWS_SECRET_KEY = self.profile['env'] | ||
|
|
||
| ENV = { | ||
| 'AWS_ACCESS_KEY_ID': AWS_ACCESS_KEY, | ||
| 'AWS_SECRET_ACCESS_KEY': AWS_SECRET_KEY | ||
| } | ||
|
|
||
| with mock.patch.dict('os.environ', ENV): | ||
| ctx = R53UpdateApp.Context() | ||
|
|
||
| creds = ctx.session.get_credentials() | ||
| self.assertNotEqual(creds, None) | ||
|
|
||
| self.assertEqual(creds.access_key, AWS_ACCESS_KEY) | ||
| self.assertEqual(creds.secret_key, AWS_SECRET_KEY) | ||
|
|
||
| self.assertNotEqual(ctx.getR53Client(), None) | ||
|
|
||
| # check credential with profile [default] (implicit) | ||
| def test_profile_default_implicit(self): | ||
| with mock.patch('os.path.isfile', side_effect=self.mock_isfile): | ||
| with mock.patch('%s.open' % self.builtins, side_effect=self.mock_file): | ||
| AWS_ACCESS_KEY, AWS_SECRET_KEY = self.profile['default'] | ||
| ctx = R53UpdateApp.Context() | ||
|
|
||
| creds = ctx.session.get_credentials() | ||
| self.assertNotEqual(creds, None) | ||
|
|
||
| self.assertEqual(creds.access_key, AWS_ACCESS_KEY) | ||
| self.assertEqual(creds.secret_key, AWS_SECRET_KEY) | ||
|
|
||
| self.assertNotEqual(ctx.getR53Client(), None) | ||
|
|
||
| # check credential with profile [default] (explicit) | ||
| def test_profile_default_explicit(self): | ||
| with mock.patch('os.path.isfile', side_effect=self.mock_isfile): | ||
| with mock.patch('%s.open' % self.builtins, side_effect=self.mock_file): | ||
| AWS_ACCESS_KEY, AWS_SECRET_KEY = self.profile['default'] | ||
| ctx = R53UpdateApp.Context('default') | ||
|
|
||
| creds = ctx.session.get_credentials() | ||
| self.assertEqual(creds.access_key, AWS_ACCESS_KEY) | ||
| self.assertEqual(creds.secret_key, AWS_SECRET_KEY) | ||
|
|
||
| self.assertNotEqual(ctx.getR53Client(), None) | ||
|
|
||
| # check credential with profile [test] | ||
| def test_profile_test(self): | ||
| with mock.patch('os.path.isfile', side_effect=self.mock_isfile): | ||
| with mock.patch('%s.open' % self.builtins, side_effect=self.mock_file): | ||
| AWS_ACCESS_KEY, AWS_SECRET_KEY = self.profile['test'] | ||
| ctx = R53UpdateApp.Context('test') | ||
|
|
||
| creds = ctx.session.get_credentials() | ||
| self.assertEqual(creds.access_key, AWS_ACCESS_KEY) | ||
| self.assertEqual(creds.secret_key, AWS_SECRET_KEY) | ||
|
|
||
| self.assertNotEqual(ctx.getR53Client(), None) | ||
|
|
||
| # check not exist profile cause `ProfileNotFound` exception | ||
| def test_profile_not_found(self): | ||
| with (mock.patch('os.environ.get', return_value=None)): | ||
| with mock.patch('os.path.isfile', side_effect=self.mock_isfile): | ||
| with mock.patch('%s.open' % self.builtins, side_effect=self.mock_file): | ||
| with self.assertRaises(exceptions.ProfileNotFound): | ||
| ctx = R53UpdateApp.Context('notexist') | ||
| ctx.getR53Client() | ||
|
|
Oops, something went wrong.