Python db_connection.env函数代码示例

def test_empty_user():
    with db_connection.env(login_active=False):
        u = User({})

        assert u.email == 'SAM'
        assert u.name == 'SAM'
        assert u.logged_in is True
        assert u.plan_active is True
        assert u.plan == 'auto'
        assert u.subscription == sub_id
        assert u.viewing == sub_id
        assert u.groups.issuperset({'login', 'subscribed'})
        assert 'logout' not in u.groups
        assert 'unsubscribed' not in u.groups
        assert 'debug' not in u.groups

    with db_connection.env(login_active=True):
        u = User({})

        assert u.email is None
        assert u.name is None
        assert u.logged_in is False
        assert u.plan_active is False
        assert u.plan is None
        assert u.subscription is None
        assert u.viewing is None
        assert u.groups.issuperset({'logout', 'unsubscribed'})
        assert 'login' not in u.groups
        assert 'subscribed' not in u.groups
        assert 'debug' not in u.groups

def test_headlesspost():
    decode_calls = []
    perform_calls = []
    encode_calls = []

    class headless_t(sam.pages.base.HeadlessPost):
        def decode_post_request(self, data):
            decode_calls.append(data.copy())
            return {'a': 1, 'b': 2}

        def perform_post_command(self, request):
            perform_calls.append(request.copy())
            return {'z': 9, 'x': 8}

        def encode_post_response(self, response):
            encode_calls.append(response.copy())
            return 'abc', 123

        def require_ownership(self):
            return True

    with db_connection.env(mock_input=True, login_active=False, mock_session=True):
        # web.input must be set before instantiating the class
        web.input = lambda: {'postinput': 'test'}
        web.ctx['headers'] = []
        p = headless_t()
        resp = p.POST()

        assert decode_calls == [{'postinput': 'test'}]
        assert perform_calls == [{'a': 1, 'b': 2}]
        assert encode_calls == [{'x': 8, 'z': 9}]
        assert resp == '["abc", 123]'

def test_post_upload():
    with db_connection.env(mock_input=True, login_active=False, mock_session=True):
        p = sam.pages.settings.Settings()
        data = {'command': 'upload', 'ds': 'ds12', 'format': 'paloalto', 'file': 'b64,YWJjMTIz'}
        request = p.decode_post_request(data)
        expected = {
            'command': 'upload',
            'ds': 12,
            'format': 'paloalto',
            'file': 'b64,YWJjMTIz'
        }
        assert request == expected
        with pytest.raises(errors.MalformedRequest):
            data = {'command': 'upload', 'format': 'paloalto', 'file': 'b64,YWJjMTIz'}
            p.decode_post_request(data)
        with pytest.raises(errors.MalformedRequest):
            data = {'command': 'upload', 'ds': 'ds12', 'file': 'b64,YWJjMTIz'}
            p.decode_post_request(data)
        with pytest.raises(errors.MalformedRequest):
            data = {'command': 'upload', 'ds': 'ds12', 'format': 'paloalto'}
            p.decode_post_request(data)

        old = sam.models.upload.Uploader
        try:
            sam.models.upload.Uploader = db_connection.Mocker
            p.perform_post_command(expected)
            calls = p.uploadModel.calls
            assert p.uploadModel.constructor == ((db, sub_id, 12, 'paloalto'), {})
            assert calls[0] == ('import_log', ('abc123',), {})
        finally:
            sam.models.upload.Uploader = old

def test_headless():
    decode_calls = []
    perform_calls = []
    encode_calls = []

    class headless_t(sam.pages.base.Headless):
        def decode_get_request(self, data):
            decode_calls.append(data.copy())
            return {'a': 1, 'b': 2}

        def perform_get_command(self, request):
            perform_calls.append(request.copy())
            return {'z': 9, 'x': 8}

        def encode_get_response(self, response):
            encode_calls.append(response.copy())
            return 'abc', 123

    with db_connection.env(mock_input=True, login_active=False, mock_session=True):
        # web.input must be set before instantiating the class
        web.input = lambda: {'input': True}
        web.ctx['headers'] = []
        p = headless_t()
        resp = p.GET()

        assert decode_calls == [{'input': True}]
        assert perform_calls == [{'a': 1, 'b': 2}]
        assert encode_calls == [{'x': 8, 'z': 9}]
        assert resp == '["abc", 123]'

        with pytest.raises(web.nomethod):
            p.POST()

def test_alerts_get_decode():
    with db_connection.env(mock_input=True, login_active=False, mock_session=True):
        a_page = alerts.Alerts()

        data = {}
        request = a_page.decode_get_request(data)
        expected = {
            'subnet': None,
            'severity': 1,
            'time': 604800,
            'sort': 'id',
            'sort_dir': 'DESC',
            'page_size': 50,
            'page_num': 1,
        }
        assert request == expected

        data = {'subnet': '127.0.0.0/24', 'severity': '6', 'time': '1 day', 'sort': 'severity', 'sort_dir': 'aSC',
                'page_size': '10', 'page_num': '2'}
        request = a_page.decode_get_request(data)
        expected = {
            'subnet': '127.0.0.0/24',
            'severity': 6,
            'time': 86400,
            'sort': 'severity',
            'sort_dir': 'ASC',
            'page_size': 10,
            'page_num': 2,
        }
        assert request == expected

def test_download():
    port_f = sam.models.filters.filterTypes.index(sam.models.filters.PortFilter)
    with db_connection.env(mock_input=True, mock_session=True, login_active=False, mock_render=True):
        web.input = lambda: {'download': '1', 'filters': 'ds{ds}|{port};1;2;80'.format(ds=ds_full, port=port_f)}
        web.ctx['headers'] = []
        p = sam.pages.table.Table()
        csv_data = p.GET()
        lines = csv_data.splitlines()
        headers = lines[0]
        lines = lines[1:]
        addresses = [line.partition(',')[0] for line in lines]
        expected = [u'10.0.0.0/8',
                    u'10.20.0.0/16',
                    u'10.20.32.0/24',
                    u'10.20.32.43/32',
                    u'10.24.0.0/16',
                    u'10.24.34.0/24',
                    u'10.24.34.44/32',
                    u'10.24.34.45/32',
                    u'50.0.0.0/8',
                    u'50.64.0.0/16',
                    u'50.64.76.0/24',
                    u'50.64.76.86/32',
                    u'101.0.0.0/8',
                    u'101.99.0.0/16',
                    u'101.99.86.0/24',
                    u'101.99.86.58/32']
        assert addresses == expected
        assert headers == 'Address,Hostname,"Role (0 = client, 1 = server)",' \
                          'Environment,Tags,Bytes handled,Packets handled'

def test_decode_post():
    with db_connection.env(mock_input=True, login_active=False, mock_session=True):
        p = sam.pages.nodes.Nodes()
        data = {'node': '10.20.30.40', 'alias': 'test_alias', 'tags': 'tag1,tag2,,', 'env': 'dev_2', 'extra': 'peanuts'}
        request = p.decode_post_request(data)
        expected = {'node': '10.20.30.40', 'alias': 'test_alias', 'tags': 'tag1,tag2,,', 'env': 'dev_2'}
        assert request == expected

        data = {'node': '10.20', 'alias': 'test_alias'}
        request = p.decode_post_request(data)
        expected = {'node': '10.20', 'alias': 'test_alias'}
        assert request == expected

        data = {'node': '10.20', 'tags': 'tag1,tag2,,'}
        request = p.decode_post_request(data)
        expected = {'node': '10.20', 'tags': 'tag1,tag2,,'}
        assert request == expected

        data = {'node': '10.20', 'env': 'smiles'}
        request = p.decode_post_request(data)
        expected = {'node': '10.20', 'env': 'smiles'}
        assert request == expected

        data = {'node': '10.20', 'env': None, 'tags': None, 'alias': None, 'extra': 'peanuts'}
        request = p.decode_post_request(data)
        expected = {'node': '10.20'}
        assert request == expected

def test_rulesedit_get_decode():
    reset_dummy_rules()
    r_model = rules.Rules(db, sub_id)
    all_rules = r_model.get_all_rules()
    ids = [rule.id for rule in all_rules]
    # valid requests require a rule id ('id') to be specified

    with db_connection.env(mock_input=True, login_active=False, mock_session=True):
        r = RulesEdit()

        # bad data input
        with pytest.raises(errors.RequiredKey):
            data = {}
            r.decode_get_request(data)
        with pytest.raises(errors.MalformedRequest):
            data = {'id': 'not a number'}
            r.decode_get_request(data)

        # good input data

        data = {'id': ids[0]}
        request = r.decode_get_request(data)
        expected = {
            'id': ids[0]
        }
        assert request == expected

def test_alertdetails_post_perform():
    reset_mock_alerts()
    a_model = AlertsModel(db, sub_id)

    with db_connection.env(mock_input=True, login_active=False, mock_session=True):
        a_page = alerts.AlertDetails()

        all_alerts = a_model.get(AlertFilter())
        a_id1 = [a['id'] for a in all_alerts if a['label'] == 'LDAP Access'][0]
        a_id2 = [a['id'] for a in all_alerts if a['label'] != 'LDAP Access'][0]

        request = {
            'method': 'update_label',
            'id': a_id1,
            'label': 'new label'
        }
        a_page.perform_post_command(request)
        request = {
            'method': 'update_label',
            'id': a_id2,
            'label': 'meh label'
        }
        a_page.perform_post_command(request)

        labels = {row['label'] for row in db.select("s{}_Alerts".format(sub_id), what='label')}
        assert labels == {'new label', 'meh label'}

def test_decode_get():
    with db_connection.env(mock_input=True, login_active=False, mock_session=True):
        p = sam.pages.nodes.Nodes()
        data = {'address': '10.20.30,150.20,59', 'ds': 'ds_13'}
        request = p.decode_get_request(data)
        expected = {'addresses': ['10.20.30', '150.20', '59'], 'flat': False, 'ds': 13}
        assert request == expected

        data = {'address': '50.60.70.80', 'ds': 'ds_13'}
        request = p.decode_get_request(data)
        expected = {'addresses': ['50.60.70.80'], 'flat': False, 'ds': 13}
        assert request == expected

        data = {'ds': '53'}
        request = p.decode_get_request(data)
        expected = {'addresses': [], 'flat': False, 'ds': 53}
        assert request == expected

        data = {'horseradish': 'pickles', 'ds': 'ds_13'}
        request = p.decode_get_request(data)
        expected = {'addresses': [], 'flat': False, 'ds': 13}
        assert request == expected

        data = {'horseradish': 'pickles', 'ds': 'ds_13', 'flat': 'true'}
        request = p.decode_get_request(data)
        expected = {'addresses': [], 'flat': True, 'ds': 13}
        assert request == expected

def test_alerts_post_decode():
    with db_connection.env(mock_input=True, login_active=False, mock_session=True):
        a_page = alerts.Alerts()

        data = {}
        with pytest.raises(errors.MalformedRequest):
            a_page.decode_post_request(data)
        data = {'method': 'delete'}
        with pytest.raises(errors.RequiredKey):
            a_page.decode_post_request(data)
        data = {'method': 'delete', 'id': 'NaN'}
        with pytest.raises(errors.RequiredKey):
            a_page.decode_post_request(data)

        data = {'method': 'delete', 'id': 199}
        request = a_page.decode_post_request(data)
        expected = {'method': 'delete', 'id': 199}
        assert request == expected

        data = {'method': 'delete_all'}
        request = a_page.decode_post_request(data)
        expected = {'method': 'delete_all', 'id': None}
        assert request == expected

        data = {'method': 'delete_all', 'id': 201}
        request = a_page.decode_post_request(data)
        expected = {'method': 'delete_all', 'id': None}
        assert request == expected

def test_alertdetails_post_decode():
    with db_connection.env(mock_input=True, login_active=False, mock_session=True):
        a_page = alerts.AlertDetails()

        data = {}
        with pytest.raises(errors.RequiredKey):
            a_page.decode_post_request(data)
        data = {'method': 'bananas'}
        with pytest.raises(errors.RequiredKey):
            a_page.decode_post_request(data)
        data = {'method': 'update_label'}
        with pytest.raises(errors.RequiredKey):
            a_page.decode_post_request(data)
        data = {'method': 'update_label', 'id': 'non-number', 'label': 'new label'}
        with pytest.raises(errors.RequiredKey):
            a_page.decode_post_request(data)
        data = {'method': 'update_label', 'id': '55'}
        with pytest.raises(errors.RequiredKey):
            a_page.decode_post_request(data)
        data = {'method': 'update_label', 'id': '55', 'label': 'new label'}
        request = a_page.decode_post_request(data)
        expected = {
            'method': 'update_label',
            'id': 55,
            'label': 'new label'
        }
        assert request == expected

def test_decode_connection_string():
    with db_connection.env(mock_input=True, mock_session=True):
        p = sam.pages.login.Login_LDAP()
        cs = 'ldaps://ipa.demo1.freeipa.org/CN=users,CN=accounts,DC=demo1,DC=freeipa,DC=org'
        address, ns = p.decode_connection_string(cs)
        assert address == 'ldaps://ipa.demo1.freeipa.org'
        assert ns == 'CN=users,CN=accounts,DC=demo1,DC=freeipa,DC=org'

def test_connect():
    # TODO: I don't control this test server. This could break.
    with db_connection.env(mock_input=True, mock_session=True):
        p = sam.pages.login.Login_LDAP()
        p.server_address = 'ldaps://ipa.demo1.freeipa.org'
        p.namespace = 'CN=users,CN=accounts,DC=demo1,DC=freeipa,DC=org'

        # wrong password
        request = {
            'user': 'bob',
            'password': 'bobpass'
        }
        with pytest.raises(errors.AuthenticationError):
            p.perform_post_command(request)

        # correct details
        #request = {
        #    'user': 'admin',
        #    'password': 'Secret123',
        #}
        #assert p.perform_post_command(request)

        # wrong url
        p.server_address = 'ldaps://ipa.demo1.garbage.org'
        with pytest.raises(errors.AuthenticationError):
            p.perform_post_command(request)

def test_perform_post_command():
    # request can be one of:
    #   {'method': 'enable' | 'disable' | 'reset_all'}
    #   {'method': accept|reject|ignore, 'warning_id': ###}
    #   {'method': reset, 'host': 'a.b.c.d'}
    test_warnings.populate_db_warnings()

    with db_connection.env(mock_input=True, login_active=False, mock_session=True):
        ad = ADPlugin()
        ap = model_ap.ADPlugin(db, sub_id)
        ws = ap.get_warnings(show_all=True)
        i = ws[7]['id']

        assert ad.perform_post_command({'method': 'disable'}) == 'success'
        assert ad.perform_post_command({'method': 'enable'}) == 'success'
        assert ad.perform_post_command({'method': 'accept', 'warning_id': i}) == 'success'
        assert ad.perform_post_command({'method': 'reject', 'warning_id': i}) == 'success'
        assert ad.perform_post_command({'method': 'ignore', 'warning_id': i}) == 'success'

        with pytest.raises(errors.MalformedRequest):
            ad.perform_post_command({'method': 'accept', 'warning_id': -200})
        with pytest.raises(errors.MalformedRequest):
            ad.perform_post_command({'method': 'reject', 'warning_id': -200})
        with pytest.raises(errors.MalformedRequest):
            ad.perform_post_command({'method': 'ignore', 'warning_id': -200})
        with pytest.raises(errors.MalformedRequest):
            ad.perform_post_command({'method': 'reset_all'})
        with pytest.raises(errors.MalformedRequest):
            ad.perform_post_command({'method': 'reset', 'host': '10.20.30.40'})
        with pytest.raises(errors.MalformedRequest):
            ad.perform_post_command({'method': 'bogus'})