23 lines
676 B
Python
23 lines
676 B
Python
|
import pytest
|
||
|
|
|
||
|
|
from data.model.organization import get_organization, get_organizations
|
||
|
|
from data.model.user import mark_namespace_for_deletion
|
||
|
|
from data.queue import WorkQueue
|
||
|
|
from test.fixtures import *
|
||
|
|
|
||
|
|
@pytest.mark.parametrize('deleted', [
|
||
|
|
(True),
|
||
|
|
(False),
|
||
|
|
])
|
||
|
|
def test_get_organizations(deleted, initialized_db):
|
||
|
|
# Delete an org.
|
||
|
|
deleted_org = get_organization('sellnsmall')
|
||
|
|
queue = WorkQueue('testgcnamespace', lambda db: db.transaction())
|
||
|
|
mark_namespace_for_deletion(deleted_org, [], queue)
|
||
|
|
|
||
|
|
orgs = get_organizations(deleted=deleted)
|
||
|
|
assert orgs
|
||
|
|
|
||
|
|
deleted_found = [org for org in orgs if org.id == deleted_org.id]
|
||
|
|
assert bool(deleted_found) == deleted
|