def test_stateloader_should_load_existing_state(tmpdir):
    state.dump({'pids': [1,2]})

    s = state.load()
    
    assert isinstance(s, State)
    assert [1,2] == s.pids
def test_stateloader_should_load_existing_state(tmpdir):
    state.dump({'pids': [1, 2]})

    s = state.load()

    assert isinstance(s, State)
    assert [1, 2] == s.pids
示例#3
0
    def handle(self):
        nodes = self.load_state()

        if 'all' in self.args.nodes:
            for node, info in nodes.iteritems():
                print 'Killing node {0} with pid {1}'.format(node, info['pid'])
                os.kill(info['pid'], 15)
            state.clear()
        else:
            for node in self.args.nodes:
                info = nodes.pop(node)
                print 'Killing node {0} with pid {1}'.format(node, info['pid'])
                os.kill(info['pid'], 15)
            state.dump(nodes)
def test_dump_should_persist_state_in_file():
    s = State()
    s.database = 'mongodb'
    s.pids = [123, 312]
    
    state.dump(s)

    try:
        with open(state.filename) as fileobj:
            data = json.load(fileobj)
    except TypeError:
        pytest.fail("%r should not be empty" % state.filename)
    except ValueError:
        pytest.fail("%r should be JSON" % state.filename)
    
    assert {'database': 'mongodb', 'pids': [123, 312]} == data
def test_dump_should_persist_state_in_file():
    s = State()
    s.database = 'mongodb'
    s.pids = [123, 312]

    state.dump(s)

    try:
        with open(state.filename) as fileobj:
            data = json.load(fileobj)
    except TypeError:
        pytest.fail("%r should not be empty" % state.filename)
    except ValueError:
        pytest.fail("%r should be JSON" % state.filename)

    assert {'database': 'mongodb', 'pids': [123, 312]} == data
示例#6
0
 def save_state(self, nodes):
     state.dump(nodes)
def test_clear_should_remove_state_file():
    state.dump({'foo': 'bar'})

    assert os.path.isfile(state.filename)
    state.clear()
    assert not os.path.isfile(state.filename)
def test_clear_should_remove_state_file():
    state.dump({'foo': 'bar'})

    assert os.path.isfile(state.filename)
    state.clear()
    assert not os.path.isfile(state.filename)