ubicloud/spec/serializers/load_balancer_spec.rb

# frozen_string_literal: true

require_relative "../spec_helper"

RSpec.describe Serializers::LoadBalancer do
  describe ".serialize_internal" do
    it "serializes an lb correctly" do
      lb = LoadBalancer.new(name: "test")
      lb.associations[:ports] = [LoadBalancerPort.new(src_port: 1, dst_port: 5)]
      expect(lb).to receive(:display_location).and_return("hetzner")
      expect(lb).to receive(:ubid).and_return("1234")
      expect(lb).to receive(:hostname).and_return("something.com")
      expect(lb).to receive(:algorithm).and_return("roundrobin")
      expect(lb).to receive(:stack).and_return("dual")
      expect(lb).to receive(:health_check_endpoint).and_return("/")
      expect(lb).to receive(:health_check_protocol).and_return("tcp")
      expect(lb).to receive(:cert_enabled).and_return(false)

      expected_result = {
        id: "1234",
        name: "test",
        location: "hetzner",
        hostname: "something.com",
        algorithm: "roundrobin",
        stack: "dual",
        health_check_endpoint: "/",
        health_check_protocol: "tcp",
        src_port: 1,
        dst_port: 5,
        cert_enabled: false
      }

      expect(described_class.serialize_internal(lb)).to eq(expected_result)
    end

    it "serializes an lb correctly2" do
      lb = LoadBalancer.new(name: "test")
      lb.associations[:ports] = []
      expect(lb).to receive(:display_location).and_return("hetzner")
      expect(lb).to receive(:ubid).and_return("1234")
      expect(lb).to receive(:hostname).and_return("something.com")
      expect(lb).to receive(:algorithm).and_return("roundrobin")
      expect(lb).to receive(:stack).and_return("dual")
      expect(lb).to receive(:health_check_endpoint).and_return("/")
      expect(lb).to receive(:health_check_protocol).and_return("tcp")

      expected_result = {
        id: "1234",
        name: "test",
        location: "hetzner",
        hostname: "something.com",
        algorithm: "roundrobin",
        stack: "dual",
        health_check_endpoint: "/",
        health_check_protocol: "tcp",
        src_port: nil,
        dst_port: nil,
        cert_enabled: nil
      }

      expect(described_class.serialize_internal(lb)).to eq(expected_result)
    end
  end
end