# frozen_string_literal: true

module Lint
  module HyperLogLog
    def test_pfadd
      target_version "2.8.9" do
        assert_equal true, r.pfadd("foo", "s1")
        assert_equal true, r.pfadd("foo", "s2")
        assert_equal false, r.pfadd("foo", "s1")

        assert_equal 2, r.pfcount("foo")
      end
    end

    def test_variadic_pfadd
      target_version "2.8.9" do
        assert_equal true, r.pfadd("foo", ["s1", "s2"])
        assert_equal true, r.pfadd("foo", ["s1", "s2", "s3"])

        assert_equal 3, r.pfcount("foo")
      end
    end

    def test_pfcount
      target_version "2.8.9" do
        assert_equal 0, r.pfcount("foo")

        assert_equal true, r.pfadd("foo", "s1")

        assert_equal 1, r.pfcount("foo")
      end
    end

    def test_variadic_pfcount
      target_version "2.8.9" do
        assert_equal 0, r.pfcount(["{1}foo", "{1}bar"])

        assert_equal true, r.pfadd("{1}foo", "s1")
        assert_equal true, r.pfadd("{1}bar", "s1")
        assert_equal true, r.pfadd("{1}bar", "s2")

        assert_equal 2, r.pfcount("{1}foo", "{1}bar")
      end
    end

    def test_variadic_pfcount_expanded
      target_version "2.8.9" do
        assert_equal 0, r.pfcount("{1}foo", "{1}bar")

        assert_equal true, r.pfadd("{1}foo", "s1")
        assert_equal true, r.pfadd("{1}bar", "s1")
        assert_equal true, r.pfadd("{1}bar", "s2")

        assert_equal 2, r.pfcount("{1}foo", "{1}bar")
      end
    end

    def test_pfmerge
      target_version '2.8.9' do
        r.pfadd 'foo', 's1'
        r.pfadd 'bar', 's2'

        assert_equal true, r.pfmerge('res', 'foo', 'bar')
        assert_equal 2, r.pfcount('res')
      end
    end

    def test_variadic_pfmerge_expanded
      redis.pfadd('{1}foo', %w[foo bar zap a])
      redis.pfadd('{1}bar', %w[a b c foo])
      assert_equal true, redis.pfmerge('{1}baz', '{1}foo', '{1}bar')
    end
  end
end