122 lines
3.3 KiB
Python
122 lines
3.3 KiB
Python
from core.builtin_concepts import ReturnValueConcept, BuiltinConcepts
|
|
from core.sheerka import Sheerka, ExecutionContext
|
|
import core.builtin_helpers
|
|
|
|
|
|
def test_i_can_use_expect_one_when_empty():
|
|
sheerka = get_sheerka()
|
|
|
|
res = core.builtin_helpers.expect_one(get_context(sheerka), [])
|
|
assert not res.status
|
|
assert sheerka.isinstance(res.value, BuiltinConcepts.IS_EMPTY)
|
|
|
|
|
|
def test_i_can_use_expect_one_when_too_many_success():
|
|
sheerka = get_sheerka()
|
|
|
|
items = [
|
|
ReturnValueConcept("who", True, "value1"),
|
|
ReturnValueConcept("who", True, "value2"),
|
|
]
|
|
res = core.builtin_helpers.expect_one(get_context(sheerka), items)
|
|
assert not res.status
|
|
assert sheerka.isinstance(res.value, BuiltinConcepts.TOO_MANY_SUCCESS)
|
|
assert res.value.obj == items
|
|
assert res.parents == items
|
|
|
|
|
|
def test_i_can_use_expect_one_when_same_success():
|
|
sheerka = get_sheerka()
|
|
|
|
items = [
|
|
ReturnValueConcept("who", True, "value"),
|
|
ReturnValueConcept("who", True, "value"),
|
|
]
|
|
res = core.builtin_helpers.expect_one(get_context(sheerka), items)
|
|
assert res.status
|
|
assert res.value == items[0].value
|
|
assert res.parents == items
|
|
|
|
|
|
def test_i_can_use_expect_when_only_errors_1():
|
|
sheerka = get_sheerka()
|
|
|
|
items = [
|
|
ReturnValueConcept("who", False, None),
|
|
]
|
|
res = core.builtin_helpers.expect_one(get_context(sheerka), items)
|
|
assert not res.status
|
|
assert sheerka.isinstance(res.value, BuiltinConcepts.TOO_MANY_ERRORS)
|
|
assert res.value.obj == items
|
|
assert res.parents == items
|
|
|
|
|
|
def test_i_can_use_expect_when_only_errors_2():
|
|
sheerka = get_sheerka()
|
|
|
|
items = [
|
|
ReturnValueConcept("who", False, None),
|
|
ReturnValueConcept("who", False, None),
|
|
]
|
|
res = core.builtin_helpers.expect_one(get_context(sheerka), items)
|
|
assert not res.status
|
|
assert sheerka.isinstance(res.value, BuiltinConcepts.TOO_MANY_ERRORS)
|
|
assert res.value.obj == items
|
|
assert res.parents == items
|
|
|
|
|
|
def test_i_can_use_expect_one_when_one_success_1():
|
|
sheerka = get_sheerka()
|
|
|
|
items = [
|
|
ReturnValueConcept("who", True, None),
|
|
]
|
|
res = core.builtin_helpers.expect_one(get_context(sheerka), items)
|
|
assert res.status
|
|
assert res.body == items[0].body
|
|
assert res.parents == items
|
|
|
|
|
|
def test_i_can_use_expect_one_when_one_success_2():
|
|
sheerka = get_sheerka()
|
|
|
|
items = [
|
|
ReturnValueConcept("who", False, None),
|
|
ReturnValueConcept("who", True, None),
|
|
ReturnValueConcept("who", False, None),
|
|
]
|
|
res = core.builtin_helpers.expect_one(get_context(sheerka), items)
|
|
assert res.status
|
|
assert res.body == items[1].body
|
|
assert res.parents == items
|
|
|
|
|
|
def test_i_can_use_expect_one_when_not_a_list_true():
|
|
sheerka = get_sheerka()
|
|
|
|
item = ReturnValueConcept("who", True, None)
|
|
res = core.builtin_helpers.expect_one(get_context(sheerka), item)
|
|
assert res.status
|
|
assert res == item
|
|
|
|
|
|
def test_i_can_use_expect_one_when_not_a_list_false():
|
|
sheerka = get_sheerka()
|
|
|
|
item = ReturnValueConcept("who", False, None)
|
|
res = core.builtin_helpers.expect_one(get_context(sheerka), item)
|
|
|
|
assert not res.status
|
|
assert res == item
|
|
|
|
|
|
def get_sheerka():
|
|
sheerka = Sheerka()
|
|
sheerka.initialize("mem://")
|
|
|
|
return sheerka
|
|
|
|
|
|
def get_context(sheerka):
|
|
return ExecutionContext("test", "xxx", sheerka)
|