mokkku 0.0.8 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a918d20e114108edde5cd017fbbb18329cac43272aba65110042e3a27fb7f77c
-  data.tar.gz: 2c0114ee979c6a69640f3e128d27cc56f1d2fd879fc260afbed13c35a0e476c0
+  metadata.gz: 57725aee6c5242ed58219debf928fcb0ad98c39b64f2505d5a2f31f20c5007bc
+  data.tar.gz: 7f1b8dbc79ab2df8ed26ee364deb7478a80a7f427fb7ce1cea764b0c337443df
 SHA512:
-  metadata.gz: '0029fbefa5b22bbd3001b701a051a80194389c6a585a44cb4f585542d378f505dab66ca09e40ea04e4c77b9d16c6471acbe9284537e6b9f2ce31bcb1b57fc618'
-  data.tar.gz: 78697aed768e3c0476be71e2cf69dbdc09e3afb1b8cc30f7cc676e689245d4b0e529e1e604b63dca182fc802a0f597d84115b2054890da5e2c7e44f7b78002c7
+  metadata.gz: 45428b6c93726c284b0ad40373e79ef8d2b265c33f604f6cab3f9211080713b70f8346f29816451faaee8801383e991892b405c423a03e3393a8f03ed9810568
+  data.tar.gz: 51005dd2a8e3a3f34c3d8ae4b9a66366d729f2c8b3cc057244596684555b5b5358f4fd3a3546b48dd45ca61ff60907d1cdf8d87b5c2c9fb057fae5bee9f7172a

data/lib/mokkku/mocked_object.rb

@@ -1,6 +1,10 @@
 module Mokkku
   class MockedObject
-    def initialize(mocks)
+    attr_reader :mocked_class
+    attr_accessor :selected_object
+
+    def initialize(mocked_class, mocks)
+      @mocked_class = mocked_class
       @mocks = mocks
       @selected_object = nil
     end
@@ -18,12 +22,7 @@ module Mokkku
     end
 
     def method_missing(method_name, *args, &block)
-      if @selected_object.nil?
-        @selected_object = mocked_objects.sample(random: Mokkku::Random)
-        @selected_object.public_send(method_name)
-      else
-        @selected_object.public_send(method_name)
-      end
+      @selected_object.public_send(method_name)
     end
   end
 end

data/lib/mokkku/uniq_utils.rb

@@ -0,0 +1,62 @@
+require 'set'
+
+module Mokkku
+  class UniqueUtils
+    class << self
+      def add_instance(generator, max_retries)
+        instances[generator.mocked_class] ||= Mokkku::UniqueUtils.new(generator, max_retries)
+      end
+
+      def instances
+        Thread.current[:mokkku_unique_utils] ||= {}
+      end
+
+      def clear
+        instances.each_value(&:clear)
+        instances.clear
+      end
+    end
+
+    def initialize(generator, max_retries)
+      @generator = generator
+      @max_retries = max_retries
+    end
+
+    def clear
+      previous_results.clear
+    end
+
+    private
+
+    def method_missing(name, *args, **kwargs)
+      if @generator.selected_object.nil? && @generator.send(:mocked_objects).first.to_h.keys.include?(name)
+        @max_retries.times do
+          next_object = @generator.send(:mocked_objects).sample(random: Mokkku::Random)
+        
+          next if previous_results.include?(next_object)
+
+          previous_results << next_object
+          @generator.instance_variable_set(:@selected_object, next_object)
+          break
+        end
+
+        if @generator.selected_object.nil?
+          previous_results.clear
+          next_object = @generator.send(:mocked_objects).sample(random: Mokkku::Random)
+          previous_results << next_object
+          @generator.instance_variable_set(:@selected_object, next_object)
+        end
+      end
+
+      @generator.public_send(name, *args, **kwargs)
+    end
+
+    def respond_to_missing?(name, *args)
+      @generator.respond_to?(name, *args) || super
+    end
+
+    def previous_results
+      @previous_results ||= Set.new
+    end
+  end
+end

data/lib/mokkku/utils.rb

@@ -1,5 +1,6 @@
 require 'yaml'
 require_relative 'mocked_object'
+require_relative 'uniq_utils'
 
 module Mokkku
   module Utils
@@ -11,11 +12,12 @@ module Mokkku
         mock_path = File.join(Mokkku.configuration.mocks_path, "#{mock_file_name}.yml")
         data = File.read(mock_path)
         parsed_data = YAML.safe_load(data, symbolize_names: true)
-        mocked_object = Mokkku::MockedObject.new(parsed_data)
+        mocked_object = Mokkku::MockedObject.new(const_name.to_s, parsed_data)
+        unique_mocked_object = Mokkku::UniqueUtils.add_instance(mocked_object, 100)
 
-        const_set const_name, mocked_object
+        const_set const_name, unique_mocked_object
 
-        mocked_object
+        unique_mocked_object
       end
     end
   end

data/lib/mokkku/version.rb

@@ -3,7 +3,7 @@ module Mokkku
     module_function
 
     def to_s
-      '0.0.8'
+      '0.1.1'
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mokkku
 version: !ruby/object:Gem::Version
-  version: 0.0.8
+  version: 0.1.1
 platform: ruby
 authors:
 - Paweł Dąbrowski
@@ -103,6 +103,7 @@ files:
 - lib/mokkku/generators/gemini.rb
 - lib/mokkku/generators/gpt.rb
 - lib/mokkku/mocked_object.rb
+- lib/mokkku/uniq_utils.rb
 - lib/mokkku/utils.rb
 - lib/mokkku/version.rb
 - lib/mokkku/yaml_sanitizer.rb