Python assert_called_with Examples

Example #1

File: test_platforms.py Project: kyphan/celery

 def test_with_guid(self, initgroups, setuid, setgid, parse_gid, parse_uid):
     parse_uid.return_value = 5001
     parse_gid.return_value = 50001
     maybe_drop_privileges(uid="user", gid="group")
     parse_uid.assert_called_with("user")
     parse_gid.assert_called_with("group")
     setgid.assert_called_with(50001)
     initgroups.assert_called_with(5001, 50001)
     setuid.assert_called_with(5001)

Example #2

 def test_with_guid(self, initgroups, setuid, setgid, parse_gid,
                    parse_uid):
     parse_uid.return_value = 5001
     parse_gid.return_value = 50001
     maybe_drop_privileges(uid="user", gid="group")
     parse_uid.assert_called_with("user")
     parse_gid.assert_called_with("group")
     setgid.assert_called_with(50001)
     initgroups.assert_called_with(5001, 50001)
     setuid.assert_called_with(5001)

Example #3

File: test_platforms.py Project: kyphan/celery

        def test_with_uid(self, initgroups, setuid, setgid, getpwuid, parse_uid):
            class pw_struct(object):
                pw_gid = 50001

            getpwuid.return_value = pw_struct()
            parse_uid.return_value = 5001
            maybe_drop_privileges(uid="user")
            parse_uid.assert_called_with("user")
            getpwuid.assert_called_with(5001)
            setgid.assert_called_with(50001)
            initgroups.assert_called_with(5001, 50001)
            setuid.assert_called_with(5001)

Example #4

        def test_with_uid(self, initgroups, setuid, setgid, getpwuid,
                          parse_uid):
            class pw_struct(object):
                pw_gid = 50001

            getpwuid.return_value = pw_struct()
            parse_uid.return_value = 5001
            maybe_drop_privileges(uid="user")
            parse_uid.assert_called_with("user")
            getpwuid.assert_called_with(5001)
            setgid.assert_called_with(50001)
            initgroups.assert_called_with(5001, 50001)
            setuid.assert_called_with(5001)