Codebase list python-faraday / 62d1b14 tests / test_model_events.py
62d1b14

Tree @62d1b14 (Download .tar.gz)

test_model_events.py @62d1b14 — raw · history · blame 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
'''
Faraday Penetration Test IDE
Copyright (C) 2013  Infobyte LLC (http://www.infobytesec.com/)
See the file 'doc/LICENSE' for the license information

'''

import pytest
from tests.factories import HostFactory, ServiceFactory
from faraday.server.models import Host, Workspace


def test_child_parent_verification_event_fails(session, workspace,
                                               second_workspace):
    host = HostFactory.build(workspace=workspace)
    ServiceFactory.build(host=host, workspace=second_workspace)
    with pytest.raises(AssertionError):
        session.commit()

    session.rollback()

    assert session.query(Host).filter(
            Workspace.id == workspace.id
        ).first() is None


def test_child_parent_verification_event_succeeds(session, workspace):
    """
        Asserts that no exception will be raised when workspace are the same.
    """
    host = HostFactory.build(workspace=workspace)
    ServiceFactory.build(host=host, workspace=workspace)
    session.commit()


def test_child_parent_verification_event_fails_update(session, workspace,
                                                      second_workspace):
    host = HostFactory.build(workspace=workspace)
    service = ServiceFactory.build(host=host, workspace=workspace)
    session.commit()
    service.workspace = second_workspace
    session.add(service)
    with pytest.raises(AssertionError):
        session.commit()


def test_child_parent_verification_event_succeds_update(session, workspace):
    host = HostFactory.build(workspace=workspace)
    service = ServiceFactory.build(host=host, workspace=workspace)
    session.commit()
    service.workspace = workspace
    session.add(service)
    session.commit()


def test_child_parent_verification_event_changing_id_fails(session, workspace,
                                                           second_workspace):

    session.add(workspace)
    session.add(second_workspace)
    session.commit()
    host = HostFactory.build(workspace=workspace)
    session.add(host)
    session.commit()
    service = ServiceFactory.build(host=host, workspace_id=second_workspace.id)

    session.add(service)

    with pytest.raises(AssertionError):
        session.commit()