ewma improvements

rootfs/etc/nginx/lua/test/util_test.lua

@@ -12,24 +12,87 @@ end
 describe("lua_ngx_var", function()
   local util = require("util")
 
-  before_each(function()
-    mock_ngx({ var = { remote_addr = "192.168.1.1", [1] = "nginx/regexp/1/group/capturing" } })
-  end)
-
   after_each(function()
     reset_ngx()
-    package.loaded["monitor"] = nil
   end)
 
-  it("returns value of nginx var by key", function()
-    assert.equal("192.168.1.1", util.lua_ngx_var("$remote_addr"))
+  describe("lua_ngx_var", function()
+    before_each(function()
+      mock_ngx({ var = { remote_addr = "192.168.1.1", [1] = "nginx/regexp/1/group/capturing" } })
+    end)
+
+    it("returns value of nginx var by key", function()
+      assert.equal("192.168.1.1", util.lua_ngx_var("$remote_addr"))
+    end)
+
+    it("returns value of nginx var when key is number", function()
+      assert.equal("nginx/regexp/1/group/capturing", util.lua_ngx_var("$1"))
+    end)
+
+    it("returns nil when variable is not defined", function()
+      assert.equal(nil, util.lua_ngx_var("$foo_bar"))
+    end)
   end)
 
-  it("returns value of nginx var when key is number", function()
-    assert.equal("nginx/regexp/1/group/capturing", util.lua_ngx_var("$1"))
-  end)
+  describe("diff_endpoints", function()
+    it("returns removed and added endpoints", function()
+      local old = {
+        { address = "10.10.10.1", port = "8080" },
+        { address = "10.10.10.2", port = "8080" },
+        { address = "10.10.10.3", port = "8080" },
+      }
+      local new = {
+        { address = "10.10.10.1", port = "8080" },
+        { address = "10.10.10.2", port = "8081" },
+        { address = "11.10.10.2", port = "8080" },
+        { address = "11.10.10.3", port = "8080" },
+      }
+      local expected_added = { "10.10.10.2:8081", "11.10.10.2:8080", "11.10.10.3:8080" }
+      table.sort(expected_added)
+      local expected_removed = { "10.10.10.2:8080", "10.10.10.3:8080" }
+      table.sort(expected_removed)
 
-  it("returns nil when variable is not defined", function()
-    assert.equal(nil, util.lua_ngx_var("$foo_bar"))
+      local added, removed = util.diff_endpoints(old, new)
+      table.sort(added)
+      table.sort(removed)
+
+      assert.are.same(expected_added, added)
+      assert.are.same(expected_removed, removed)
+    end)
+
+    it("returns empty results for empty inputs", function()
+      local added, removed = util.diff_endpoints({}, {})
+
+      assert.are.same({}, added)
+      assert.are.same({}, removed)
+    end)
+
+    it("returns empty results for same inputs", function()
+      local old = {
+        { address = "10.10.10.1", port = "8080" },
+        { address = "10.10.10.2", port = "8080" },
+        { address = "10.10.10.3", port = "8080" },
+      }
+      local new = util.deepcopy(old)
+
+      local added, removed = util.diff_endpoints(old, new)
+
+      assert.are.same({}, added)
+      assert.are.same({}, removed)
+    end)
+
+    it("handles endpoints with nil attribute", function()
+      local old = {
+        { address = nil, port = "8080" },
+        { address = "10.10.10.2", port = "8080" },
+        { address = "10.10.10.3", port = "8080" },
+      }
+      local new = util.deepcopy(old)
+      new[2].port = nil
+
+      local added, removed = util.diff_endpoints(old, new)
+      assert.are.same({ "10.10.10.2:nil" }, added)
+      assert.are.same({ "10.10.10.2:8080" }, removed)
+    end)
   end)
 end)