def test_cli_op_line_numbers(self):
inventory = make_inventory()
state = State(inventory, Config())
connect_all(state)
state.current_deploy_filename = __file__
pyinfra.is_cli = True
pseudo_state.set(state)
# Add op to both hosts
for name in ('anotherhost', 'somehost'):
pseudo_host.set(inventory.get_host(name))
server.shell('echo hi') # note this is called twice but on *the same line*
# Add op to just the second host - using the pseudo modules such that
# it replicates a deploy file.
pseudo_host.set(inventory.get_host('anotherhost'))
first_pseudo_hash = server.user('anotherhost_user').hash
first_pseudo_call_line = getframeinfo(currentframe()).lineno - 1
# Add op to just the first host - using the pseudo modules such that
# it replicates a deploy file.
pseudo_host.set(inventory.get_host('somehost'))
second_pseudo_hash = server.user('somehost_user').hash
second_pseudo_call_line = getframeinfo(currentframe()).lineno - 1
pseudo_state.reset()
pseudo_host.reset()
pyinfra.is_cli = False
# Ensure there are two ops
op_order = state.get_op_order()
assert len(op_order) == 3
# And that the two ops above were called in the expected order
assert op_order[1] == first_pseudo_hash
assert op_order[2] == second_pseudo_hash
# And that they have the expected line numbers
assert state.op_line_numbers_to_hash.get((first_pseudo_call_line,)) == first_pseudo_hash
assert state.op_line_numbers_to_hash.get((second_pseudo_call_line,)) == second_pseudo_hash
# Ensure somehost has two ops and anotherhost only has the one
assert len(state.ops[inventory.get_host('somehost')]) == 2
assert len(state.ops[inventory.get_host('anotherhost')]) == 2
def test_cli_op_line_numbers(self):
inventory = make_inventory()
state = State(inventory, Config())
connect_all(state)
state.current_deploy_filename = __file__
pyinfra.is_cli = True
ctx_state.set(state)
# Add op to both hosts
for name in ("anotherhost", "somehost"):
ctx_host.set(inventory.get_host(name))
server.shell(
"echo hi") # note this is called twice but on *the same line*
# Add op to just the second host - using the context modules such that
# it replicates a deploy file.
ctx_host.set(inventory.get_host("anotherhost"))
first_context_hash = server.user("anotherhost_user").hash
# Add op to just the first host - using the context modules such that
# it replicates a deploy file.
ctx_host.set(inventory.get_host("somehost"))
second_context_hash = server.user("somehost_user").hash
ctx_state.reset()
ctx_host.reset()
pyinfra.is_cli = False
# Ensure there are two ops
op_order = state.get_op_order()
assert len(op_order) == 3
# And that the two ops above were called in the expected order
assert op_order[1] == first_context_hash
assert op_order[2] == second_context_hash
# Ensure somehost has two ops and anotherhost only has the one
assert len(state.ops[inventory.get_host("somehost")]) == 2
assert len(state.ops[inventory.get_host("anotherhost")]) == 2
