improve test coverage

googleapis · Jan 9, 2020 · bae8997 · bae8997
1 parent f84ea6f
commit bae8997
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/api_core/tests/unit/test_iam.py b/api_core/tests/unit/test_iam.py
@@ -74,10 +74,13 @@ def test___setitem__(self):
         USER = "user:phred@example.com"
         PRINCIPALS = set([USER])
         policy = self._make_one()
-        policy["rolename"] = [USER]
-        assert policy["rolename"] == PRINCIPALS
-        assert len(policy) == 1
-        assert dict(policy) == {"rolename": PRINCIPALS}
+        policy.bindings = [
+            {"role": "role/reader", "members": {USER}},
+            {"role": "role/writer", "members": {USER}},
+        ]
+        assert policy["role/reader"] == PRINCIPALS
+        assert len(policy) == 2
+        assert dict(policy) == {"role/reader": PRINCIPALS, "role/writer": PRINCIPALS}
 
     def test__set_item__overwrite(self):
         USER = "user:phred@example.com"