예제 #1
0
파일: tests.py 프로젝트: TJKenney/muckrock
    def test_agency_get_proxy_info(self):
        """Test an agencies get_proxy_info method"""
        agency_ = AgencyFactory()
        proxy_info = agency_.get_proxy_info()
        eq_(proxy_info['proxy'], False)
        eq_(proxy_info['missing_proxy'], False)
        assert_not_in('from_user', proxy_info)
        assert_not_in('warning', proxy_info)

        proxy_placeholder = UserFactory(username='******')
        agency_ = AgencyFactory(requires_proxy=True)
        proxy_info = agency_.get_proxy_info()
        eq_(proxy_info['proxy'], True)
        eq_(proxy_info['missing_proxy'], True)
        eq_(proxy_info['from_user'], proxy_placeholder)
        assert_in('warning', proxy_info)

        proxy = UserFactory(
            profile__acct_type='proxy',
            profile__state=agency_.jurisdiction.legal.abbrev,
        )
        proxy_info = agency_.get_proxy_info()
        eq_(proxy_info['proxy'], True)
        eq_(proxy_info['missing_proxy'], False)
        eq_(proxy_info['from_user'], proxy)
        assert_in('warning', proxy_info)
예제 #2
0
파일: tests.py 프로젝트: WPMedia/muckrock
    def test_agency_get_proxy_info(self):
        """Test an agencies get_proxy_info method"""
        agency_ = AgencyFactory()
        proxy_info = agency_.get_proxy_info()
        eq_(proxy_info["proxy"], False)
        eq_(proxy_info["missing_proxy"], False)
        assert_not_in("from_user", proxy_info)
        assert_not_in("warning", proxy_info)

        agency_ = AgencyFactory(requires_proxy=True)
        proxy_info = agency_.get_proxy_info()
        eq_(proxy_info["proxy"], True)
        eq_(proxy_info["missing_proxy"], True)
        assert_not_in("from_user", proxy_info)
        assert_in("warning", proxy_info)

        proxy = UserFactory(
            profile__proxy=True, profile__state=agency_.jurisdiction.legal.abbrev
        )
        proxy_info = agency_.get_proxy_info()
        eq_(proxy_info["proxy"], True)
        eq_(proxy_info["missing_proxy"], False)
        eq_(proxy_info["from_user"], proxy)
        assert_in("warning", proxy_info)