Restore counter tests from main

lib/kredis/default_values.rb

@@ -1,8 +1,12 @@
+# frozen_string_literal: true
+
 module Kredis::DefaultValues
   extend ActiveSupport::Concern
 
   prepended do
-    attr_writer :default, :default_context
+    attr_writer :default
+
+    # proxying :watch, :unwatch, :exists?
 
     def default
       case @default
@@ -20,6 +24,13 @@ def set_default
 
   def initialize(...)
     super
-    set_default if default.present? && !exists?
+
+    if default
+      proxy.watch do
+        set_default unless proxy.exists?
+
+        proxy.unwatch
+      end
+    end
   end
 end

test/types/counter_test.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require "test_helper"
+require "active_support/core_ext/integer"
 
 class CounterTest < ActiveSupport::TestCase
   setup { @counter = Kredis.counter "mycounter" }
@@ -81,10 +82,41 @@ class CounterTest < ActiveSupport::TestCase
   end
 
   test "default value" do
-    @counter = Kredis.counter "mycounter", default: 5
+    @counter = Kredis.counter "mycounter", default: 10
+    assert_equal 10, @counter.value
+  end
+
+  test "expiring counter with default" do
+    @counter = Kredis.counter "mycounter", default: ->() { 10 }, expires_in: 1.second
+
+    @counter.increment
+    assert_equal 11, @counter.value
+
+    sleep 0.5.seconds
 
     @counter.increment
+    assert_equal 12, @counter.value
+
+    sleep 0.5.seconds
+
+    # Defaults are only set on initialization
+    assert_equal 0, @counter.value
+  end
+
+  test "default via proc" do
+    @counter = Kredis.counter "mycounter", default: ->() { 10 }
+    assert_equal 10, @counter.value
+    @counter.decrement
+    assert_equal 9, @counter.value
+  end
+
+  test "concurrent initialization with default" do
+    5.times.map do
+      Thread.new do
+        Kredis.counter("mycounter", default: 5).increment
+      end
+    end.each(&:join)
 
-    assert_equal 6, @counter.value
+    assert_equal 10, Kredis.counter("mycounter").value
   end
 end