Toggle navigation
Toggle navigation
This project
Loading...
Sign in
서승완
/
kappa
Go to a project
Toggle navigation
Toggle navigation pinning
Projects
Groups
Snippets
Help
Project
Activity
Repository
Graphs
Network
Create a new issue
Commits
Issue Boards
Authored by
James Cooper
2016-06-12 16:11:06 -0700
Browse Files
Options
Browse Files
Download
Email Patches
Plain Diff
Commit
31476b9859c08b046d27c68429ad54d65d15e809
31476b98
1 parent
e207d6e1
add unit tests for Role.delete
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
79 additions
and
0 deletions
tests/unit/test_role.py
tests/unit/test_role.py
0 → 100644
View file @
31476b9
# -*- coding: utf-8 -*-
# Copyright (c) 2015 Mitch Garnaat http://garnaat.org/
#
# Licensed under the Apache License, Version 2.0 (the "License"). You
# may not use this file except in compliance with the License. A copy of
# the License is located at
#
# http://aws.amazon.com/apache2.0/
#
# or in the "license" file accompanying this file. This file 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.
import
unittest
import
random
import
string
from
mock
import
Mock
,
call
from
kappa.role
import
Role
def
randomword
(
length
):
return
''
.
join
(
random
.
choice
(
string
.
lowercase
)
for
i
in
range
(
length
))
class
TestRole
(
unittest
.
TestCase
):
def
setUp
(
self
):
self
.
iam_client
=
Mock
()
policy
=
type
(
'Policy'
,
(
object
,),
{
'arn'
:
None
})
self
.
context
=
type
(
'Context'
,
(
object
,),
{
'name'
:
randomword
(
10
),
'environment'
:
randomword
(
10
),
'policy'
:
policy
})
self
.
role_record
=
{
'RoleName'
:
'
%
s_
%
s'
%
(
self
.
context
.
name
,
self
.
context
.
environment
),
'Arn'
:
randomword
(
10
)
}
self
.
role
=
Role
(
self
.
context
,
None
,
iam_client
=
self
.
iam_client
)
def
_expect_get_role
(
self
):
get_role_resp
=
{
'Role'
:
self
.
role_record
}
self
.
iam_client
.
configure_mock
(
**
{
'call.return_value'
:
get_role_resp
})
return
get_role_resp
def
test_delete_no_ops_if_role_not_found
(
self
):
self
.
iam_client
.
configure_mock
(
**
{
'call.return_value'
:
None
})
self
.
assertEquals
(
None
,
self
.
role
.
delete
())
self
.
assertEquals
(
1
,
self
.
iam_client
.
call
.
call_count
)
def
test_delete_when_role_exists
(
self
):
get_role_resp
=
self
.
_expect_get_role
()
self
.
assertEquals
(
get_role_resp
,
self
.
role
.
delete
())
self
.
iam_client
.
call
.
assert_has_calls
([
call
(
'get_role'
,
RoleName
=
self
.
role
.
name
),
call
(
'delete_role'
,
RoleName
=
self
.
role
.
name
)])
def
test_delete_policy_when_context_arn_exists
(
self
):
self
.
context
.
policy
.
arn
=
randomword
(
10
)
get_role_resp
=
self
.
_expect_get_role
()
self
.
assertEquals
(
get_role_resp
,
self
.
role
.
delete
())
calls
=
[
call
(
'get_role'
,
RoleName
=
self
.
role
.
name
),
call
(
'detach_role_policy'
,
RoleName
=
self
.
role
.
name
,
PolicyArn
=
self
.
context
.
policy
.
arn
),
call
(
'delete_role'
,
RoleName
=
self
.
role
.
name
)]
self
.
iam_client
.
call
.
assert_has_calls
(
calls
)
Please
register
or
login
to post a comment