Skip to content

Commit

Permalink
update tests
Browse files Browse the repository at this point in the history
  • Loading branch information
icebob committed May 10, 2018
1 parent 080437d commit 5b50622
Show file tree
Hide file tree
Showing 3 changed files with 13 additions and 3 deletions.
2 changes: 1 addition & 1 deletion src/registry/node-catalog.js
Original file line number Diff line number Diff line change
Expand Up @@ -183,7 +183,7 @@ class NodeCatalog {
const needRegister = node.update(payload, isReconnected);

// Refresh services if 'seq' is greater or it is a reconnected node
if ((needRegister) && node.services) {
if (needRegister && node.services) {
this.registry.registerServices(node, node.services);
}

Expand Down
5 changes: 3 additions & 2 deletions test/unit/registry/node-catalog.spec.js
Original file line number Diff line number Diff line change
Expand Up @@ -181,6 +181,9 @@ describe("Test NodeCatalog.processNodeInfo", () => {
broker.registry.registerServices.mockClear();
broker.broadcastLocal.mockClear();

let node = catalog.get("node-12");
node.update = jest.fn(() => true);

let payload = {
sender: "node-12",
services: [{}, {}, {}]
Expand All @@ -189,8 +192,6 @@ describe("Test NodeCatalog.processNodeInfo", () => {
catalog.processNodeInfo(payload);
expect(catalog.nodes.size).toBe(2);

let node = catalog.get("node-12");

expect(broker.registry.registerServices).toHaveBeenCalledTimes(1);
expect(broker.registry.registerServices).toHaveBeenCalledWith(node, node.services);

Expand Down
9 changes: 9 additions & 0 deletions test/unit/registry/node.spec.js
Original file line number Diff line number Diff line change
Expand Up @@ -71,6 +71,15 @@ describe("Test Node", () => {

expect(node.seq).toBe(8);
});

it("should update 'seq' if reconnected", () => {
let payload = {
seq: 2
};
node.update(payload, true);

expect(node.seq).toBe(2);
});
});

it("should update local info", () => {
Expand Down

0 comments on commit 5b50622

Please sign in to comment.