# Path: lamvery/clients/base.py
# class BaseClient:
# 
#     __metaclass__ = ABCMeta
# 
#     def __init__(self, region=None, profile=None, dry_run=False):
#         self._session = boto3.session.Session(
#             profile_name=profile, region_name=region)
#         self._dry_run = dry_run
#         self._sts = self._session.client('sts')
# 
#     def get_account_id(self):
#         return self._sts.get_caller_identity().get('Account')


from unittest import TestCase
from nose.tools import eq_
from mock import Mock
from lamvery.clients.base import BaseClient
# -*- coding: utf-8 -*-



class TestClient(BaseClient):
    pass


class BaseClientTestCase(TestCase):

    def setUp(self):
        self.client = TestClient(region='us-east-1')
        self.client._sts = Mock()

    def get_account_id(self):
        self.client._sts.get_caller_identity = Mock(return_value={'Account': 'foo'})

# Path: lamvery/clients/base.py
# class BaseClient:
# 
#     __metaclass__ = ABCMeta
# 
#     def __init__(self, region=None, profile=None, dry_run=False):
#         self._session = boto3.session.Session(
#             profile_name=profile, region_name=region)
#         self._dry_run = dry_run
#         self._sts = self._session.client('sts')
# 
#     def get_account_id(self):
#         return self._sts.get_caller_identity().get('Account')


# Path: tests/lamvery/clients/base_test.py
from unittest import TestCase
from nose.tools import eq_
from mock import Mock
from lamvery.clients.base import BaseClient

# -*- coding: utf-8 -*-



class TestClient(BaseClient):
    pass


class BaseClientTestCase(TestCase):

    def setUp(self):
        self.client = TestClient(region='us-east-1')
        self.client._sts = Mock()

    def get_account_id(self):
        self.client._sts.get_caller_identity = Mock(return_value={'Account': 'foo'})

        eq_(self.client.get_account_id(), 'foo')