def test_can_add_manual_proxy_to_desired_capabilities(): proxy = Proxy() proxy.http_proxy = MANUAL_PROXY['httpProxy'] proxy.ftp_proxy = MANUAL_PROXY['ftpProxy'] proxy.no_proxy = MANUAL_PROXY['noProxy'] proxy.sslProxy = MANUAL_PROXY['sslProxy'] proxy.socksProxy = MANUAL_PROXY['socksProxy'] proxy.socksUsername = MANUAL_PROXY['socksUsername'] proxy.socksPassword = MANUAL_PROXY['socksPassword'] proxy.socksVersion = MANUAL_PROXY['socksVersion'] desired_capabilities = {} proxy.add_to_capabilities(desired_capabilities) proxy_capabilities = MANUAL_PROXY.copy() proxy_capabilities['proxyType'] = 'MANUAL' expected_capabilities = {'proxy': proxy_capabilities} assert expected_capabilities == desired_capabilities