← Back to branch summary

~andrewjbeach/juju-ci-tools/isolate-autoload

« back to all changes in this revision

Viewing changes to tests/test_clean_resources.py

  • Committer: Aaron Bentley
  • Date: 2016-11-07 14:56:42 UTC
  • mfrom: (1699.2.11 indirect-config-7)
  • Revision ID: aaron.bentley@canonical.com-20161107145642-r31e3j50k4riberv
Accept client.env in make_substrate_manager and it callees.

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/test_clean_resources.py
1
1
from argparse import Namespace
2
2
from mock import (
3
 
    MagicMock,
 
3
    call,
4
4
    patch,
5
5
)
6
6
 
56
56
 
57
57
    def asses_clean(self, all_region, call_count):
58
58
        args = Namespace(env='foo', verbose=0, all_regions=all_region)
59
 
        clean_resources.AWSAccount = MagicMock(spec=['manager_from_config'])
60
 
        with patch('clean_resources.SimpleEnvironment.from_config',
61
 
                   return_value=get_aws_env()) as cr_mock:
62
 
            clean(args)
 
59
        env = get_aws_env()
 
60
        with patch.object(clean_resources.AWSAccount,
 
61
                          'from_boot_config') as fbc_mock:
 
62
            with patch('clean_resources.SimpleEnvironment.from_config',
 
63
                       return_value=env) as cr_mock:
 
64
                clean(args)
63
65
        self.assertEqual(
64
 
            clean_resources.AWSAccount.manager_from_config.call_count,
 
66
            fbc_mock.call_count,
65
67
            call_count)
66
 
        mfc_mock = clean_resources.AWSAccount.manager_from_config.return_value
67
 
        ctx_mock = mfc_mock.__enter__.return_value
 
68
        regions = get_regions(args, env)
 
69
        calls = [call(cr_mock.return_value, region=r) for r in regions]
 
70
        self.assertEqual(fbc_mock.call_args_list, calls)
 
71
        ctx_mock = fbc_mock.return_value.__enter__.return_value
68
72
        self.assertEqual(ctx_mock.iter_security_groups.call_count, call_count)
69
73
        self.assertEqual(
70
74
            ctx_mock.iter_instance_security_groups.call_count, call_count)