コード例 #1
0
ファイル: test_txraft.py プロジェクト: tehasdf/txraft
    def test_remove_incorrect(self):
        clock = Clock()

        leader_store = MockStoreDontUse(
            entries={
                1: Entry(term=1, payload=1),
                2: Entry(term=2, payload=2),
                3: Entry(term=2, payload=3),
            })
        leader_store.setCurrentTerm(2)
        leader_rpc = MockRPC()
        leader = RaftNode(1, leader_store, leader_rpc, clock=clock)

        follower_store = MockStoreDontUse({
            1: Entry(term=1, payload=1),
            2: Entry(term=5, payload=1)
        })
        follower_rpc = MockRPC()
        follower = RaftNode(2, follower_store, follower_rpc, clock=clock)

        leader_rpc.simpleAddNode(follower)
        follower_rpc.simpleAddNode(leader)

        d = leader._callAppendEntries(follower.id, {})
        res = self.successResultOf(d)

        self.assertEqual(leader_store.log, follower_store.log)
コード例 #2
0
ファイル: test_txraft.py プロジェクト: tehasdf/txraft
    def test_respond_requestVote_oldLog(self):
        store = MockStoreDontUse(entries={
            2: Entry(term=2, payload=1),
            3: Entry(term=3, payload=2)
        })
        store.setCurrentTerm(3)
        rpc = MockRPC()
        clock = Clock()
        node = RaftNode(1, store, rpc, clock=clock)

        resp = node.respond_requestVote(
            RequestVotesCommand(term=4,
                                candidateId='id',
                                lastLogIndex=2,
                                lastLogTerm=2))
        term, result = self.successResultOf(resp)
        self.assertFalse(result)

        resp = node.respond_requestVote(
            RequestVotesCommand(term=4,
                                candidateId='id',
                                lastLogIndex=4,
                                lastLogTerm=2))
        term, result = self.successResultOf(resp)
        self.assertFalse(result)

        resp = node.respond_requestVote(
            RequestVotesCommand(term=4,
                                candidateId='id',
                                lastLogIndex=2,
                                lastLogTerm=3))
        term, result = self.successResultOf(resp)
        self.assertFalse(result)
コード例 #3
0
ファイル: test_txraft.py プロジェクト: tehasdf/txraft
    def test_cluster(self):
        nodes = []
        for num in range(5):
            clock = Clock()
            rpc = MockRPC()
            store = MockStoreDontUse()
            node = RaftNode(num, store, rpc, clock=clock, electionTimeout=1)
            nodes.append((node, rpc, store, clock))

        for node1, rpc, _, _ in nodes:
            for node2, _, _, _ in nodes:
                if node1 is node2:
                    continue
                rpc.simpleAddNode(node2)

        for node, rpc, store, clock in nodes:
            clock.advance(1.0)
コード例 #4
0
ファイル: test_txraft.py プロジェクト: tehasdf/txraft
    def test_three_up(self):
        store1 = MockStoreDontUse()
        store2 = MockStoreDontUse()
        store3 = MockStoreDontUse()
        rpc1 = MockRPC()
        rpc2 = MockRPC()
        rpc3 = MockRPC()

        clock1 = Clock()
        clock2 = Clock()
        clock3 = Clock()

        node1 = RaftNode(1, store1, rpc1, clock=clock1)
        node2 = RaftNode(2, store2, rpc2, clock=clock2)
        node3 = RaftNode(3, store3, rpc3, clock=clock3)

        for rpc in [rpc1, rpc2, rpc3]:
            for node in [node1, node2, node3]:
                rpc.simpleAddNode(node)

        clock1.advance(0.4)
        self.assertIs(node1._state, STATE.LEADER)
コード例 #5
0
ファイル: test_txraft.py プロジェクト: tehasdf/txraft
    def test_respond_requestVote_lowerTerm(self):
        store = MockStoreDontUse()
        store.setCurrentTerm(3)
        rpc = MockRPC()
        clock = Clock()
        node = RaftNode(1, store, rpc, clock=clock)

        resp = node.respond_requestVote(
            RequestVotesCommand(term=2,
                                candidateId='id',
                                lastLogIndex=4,
                                lastLogTerm=4))
        term, result = self.successResultOf(resp)
        self.assertFalse(result)
コード例 #6
0
ファイル: test_txraft.py プロジェクト: tehasdf/txraft
    def test_respond_requestVote(self):
        store = MockStoreDontUse()
        rpc = MockRPC()
        clock = Clock()
        node = RaftNode(1, store, rpc, clock=clock)

        resp = node.respond_requestVote(
            RequestVotesCommand(term=4,
                                candidateId=2,
                                lastLogIndex=4,
                                lastLogTerm=4))

        term, result = self.successResultOf(resp)
        self.assertTrue(result)
        votedFor = self.successResultOf(store.getVotedFor())
        self.assertEqual(votedFor, 2)
コード例 #7
0
ファイル: test_txraft.py プロジェクト: tehasdf/txraft
    def test_respond_appendEntries_empty(self):
        store = MockStoreDontUse()
        rpc = MockRPC()
        clock = Clock()
        node = RaftNode(1, store, rpc, clock=clock)

        newentry = Entry(term=0, payload=1)

        resp = node.respond_appendEntries(
            AppendEntriesCommand(term=0,
                                 leaderId=2,
                                 prevLogIndex=0,
                                 prevLogTerm=0,
                                 entries={},
                                 leaderCommit=1))

        term, result = self.successResultOf(resp)
        self.assertEqual(term, 0)
        self.assertTrue(result)
コード例 #8
0
ファイル: test_txraft.py プロジェクト: tehasdf/txraft
    def test_respond_requestVote_alreadyVoted(self):
        store = MockStoreDontUse()
        store.setVotedFor(3)
        rpc = MockRPC()
        clock = Clock()
        node = RaftNode(1, store, rpc, clock=clock)

        resp = node.respond_requestVote(
            RequestVotesCommand(term=4,
                                candidateId=2,
                                lastLogIndex=4,
                                lastLogTerm=4))
        term, result = self.successResultOf(resp)
        self.assertFalse(result)

        resp = node.respond_requestVote(
            RequestVotesCommand(term=4,
                                candidateId=3,
                                lastLogIndex=4,
                                lastLogTerm=4))
        term, result = self.successResultOf(resp)
        self.assertTrue(result)