Beispiel #1
0
def test_shared_memory_initialization():
    shared = botogram.shared.SharedMemory()

    def init1(shared):
        if "a" in shared:
            shared["a"] = 1
        else:
            shared["a"] = 0

    def init2(shared):
        shared["b"] = 1

    shared.register_inits_list("comp1", [init1, init2])
    shared.register_inits_list("comp2", [init1])

    memory1 = shared.of("bot1", "comp1")
    memory2 = shared.of("bot1", "comp2")
    memory3 = shared.of("bot2", "comp1")

    assert memory1["a"] == 0
    assert memory1["b"] == 1
    assert memory2["a"] == 0
    assert "b" not in memory2
    # memory3["a"] should be 1 if the initializer was called multiple times
    assert memory3["a"] == 0

    # memory1b["a"] should be 1 if the initializer was called multiple times
    memory1b = shared.of("bot1", "comp1")
    assert memory1b["a"] == 0
Beispiel #2
0
def test_shared_memory_pickleable():
    shared = botogram.shared.SharedMemory()
    shared.of("bot1", "comp1")["test"] = "test"

    pickled = pickle.loads(pickle.dumps(shared))

    original = shared.of("bot1", "comp1")["test"]
    assert original == pickled.of("bot1", "comp1")["test"]
Beispiel #3
0
def test_shared_memory_pickleable():
    shared = botogram.shared.SharedMemory()
    shared.of("bot1", "comp1")["test"] = "test"

    pickled = pickle.loads(pickle.dumps(shared))

    original = shared.of("bot1", "comp1")["test"]
    assert original == pickled.of("bot1", "comp1")["test"]
Beispiel #4
0
def test_shared_memory_creation():
    shared = botogram.shared.SharedMemory()

    comp1 = shared.of("bot1", "comp1")
    comp2 = shared.of("bot1", "comp2")
    comp1b = shared.of("bot1", "comp1")

    comp1["test"] = "test"

    assert "test" in comp1
    assert "test" in comp1b
    assert not comp2  # empty
    assert comp1["test"] == comp1b["test"]
Beispiel #5
0
def test_switch_driver():
    shared = botogram.shared.SharedMemory()

    # Initialize the memory with some dummy data
    shared.of("bot1", "test1")["a"] = "b"
    shared.of("bot1", "test2")["b"] = "c"

    # Create a new driver
    driver = botogram.shared.LocalDriver()
    shared.switch_driver(driver)

    assert shared.driver == driver
    assert shared.of("bot1", "test1")["a"] == "b"
    assert shared.of("bot1", "test2")["b"] == "c"
Beispiel #6
0
def test_switch_driver():
    shared = botogram.shared.SharedMemory()

    # Initialize the memory with some dummy data
    shared.of("bot1", "test1")["a"] = "b"
    shared.of("bot1", "test2")["b"] = "c"

    # Create a new driver
    driver = botogram.shared.LocalDriver()
    shared.switch_driver(driver)

    assert shared.driver == driver
    assert shared.of("bot1", "test1")["a"] == "b"
    assert shared.of("bot1", "test2")["b"] == "c"
Beispiel #7
0
def test_shared_memory_creation():
    shared = botogram.shared.SharedMemory()

    comp1 = shared.of("bot1", "comp1")
    comp2 = shared.of("bot1", "comp2")
    comp1b = shared.of("bot1", "comp1")
    comp1sub = shared.of("bot1", "comp1", "sub")

    comp1["test"] = "test"

    assert "test" in comp1
    assert "test" in comp1b
    assert not comp2  # empty
    assert comp1["test"] == comp1b["test"]
    assert not comp1sub  # empty

    # memory3sub has more than two parts in a name, so special shared memory
    # methods should not have been applied
    assert not hasattr(comp1sub, "lock")
Beispiel #8
0
def test_shared_memory_creation():
    shared = botogram.shared.SharedMemory()

    comp1 = shared.of("bot1", "comp1")
    comp2 = shared.of("bot1", "comp2")
    comp1b = shared.of("bot1", "comp1")
    comp1sub = shared.of("bot1", "comp1", "sub")

    comp1["test"] = "test"

    assert "test" in comp1
    assert "test" in comp1b
    assert not comp2  # empty
    assert comp1["test"] == comp1b["test"]
    assert not comp1sub  # empty

    # memory3sub has more than two parts in a name, so special shared memory
    # methods should not have been applied
    assert not hasattr(comp1sub, "lock")
Beispiel #9
0
def test_shared_memory_preparers():
    shared = botogram.shared.SharedMemory()

    def init1(shared):
        if "a" in shared:
            shared["a"] = 1
        else:
            shared["a"] = 0

    def init2(shared):
        shared["b"] = 1

    comp = botogram.Component()
    init1_hook = botogram.hooks.MemoryPreparerHook(init1, comp)
    init2_hook = botogram.hooks.MemoryPreparerHook(init2, comp)

    shared.register_preparers_list("comp1", [init1_hook, init2_hook])
    shared.register_preparers_list("comp2", [init1_hook])

    memory1 = shared.of("bot1", "comp1")
    memory2 = shared.of("bot1", "comp2")
    memory3 = shared.of("bot2", "comp1")
    memory3sub = shared.of("bot2", "comp1", "sub")

    assert memory1["a"] == 0
    assert memory1["b"] == 1
    assert memory2["a"] == 0
    assert "b" not in memory2
    # memory3["a"] should be 1 if the initializer was called multiple times
    assert memory3["a"] == 0

    # memory3sub has more than two parts in a name, so no preparer should have
    # been applied
    assert "a" not in memory3sub

    # memory1b["a"] should be 1 if the initializer was called multiple times
    memory1b = shared.of("bot1", "comp1")
    assert memory1b["a"] == 0
Beispiel #10
0
def test_shared_memory_preparers():
    shared = botogram.shared.SharedMemory()

    def init1(shared):
        if "a" in shared:
            shared["a"] = 1
        else:
            shared["a"] = 0

    def init2(shared):
        shared["b"] = 1

    comp = botogram.Component()
    init1_hook = botogram.hooks.MemoryPreparerHook(init1, comp)
    init2_hook = botogram.hooks.MemoryPreparerHook(init2, comp)

    shared.register_preparers_list("comp1", [init1_hook, init2_hook])
    shared.register_preparers_list("comp2", [init1_hook])

    memory1 = shared.of("bot1", "comp1")
    memory2 = shared.of("bot1", "comp2")
    memory3 = shared.of("bot2", "comp1")
    memory3sub = shared.of("bot2", "comp1", "sub")

    assert memory1["a"] == 0
    assert memory1["b"] == 1
    assert memory2["a"] == 0
    assert "b" not in memory2
    # memory3["a"] should be 1 if the initializer was called multiple times
    assert memory3["a"] == 0

    # memory3sub has more than two parts in a name, so no preparer should have
    # been applied
    assert "a" not in memory3sub

    # memory1b["a"] should be 1 if the initializer was called multiple times
    memory1b = shared.of("bot1", "comp1")
    assert memory1b["a"] == 0