mokkku 0.0.8 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a918d20e114108edde5cd017fbbb18329cac43272aba65110042e3a27fb7f77c
-  data.tar.gz: 2c0114ee979c6a69640f3e128d27cc56f1d2fd879fc260afbed13c35a0e476c0
+  metadata.gz: 4cd54dda1bd58c2eb99881690a02c99b7962c018bdf07132122d5240d6885c2f
+  data.tar.gz: 5c61996856695c6f53b8015eaa9c7923bf48a067969e8a9f4337db82cfc58b54
 SHA512:
-  metadata.gz: '0029fbefa5b22bbd3001b701a051a80194389c6a585a44cb4f585542d378f505dab66ca09e40ea04e4c77b9d16c6471acbe9284537e6b9f2ce31bcb1b57fc618'
-  data.tar.gz: 78697aed768e3c0476be71e2cf69dbdc09e3afb1b8cc30f7cc676e689245d4b0e529e1e604b63dca182fc802a0f597d84115b2054890da5e2c7e44f7b78002c7
+  metadata.gz: 15a72e5e0e60cb7a2ef7b9febcc3dc17dfa01d3393358ee4a147654e2e915eafee3c7d55a9fb174b83433d2e724117fa79c4f33ca23d89d7be4b731b2ed2f0dc
+  data.tar.gz: 76acb0e6fba9015ee604d9f19eaf9834ef66c8d1a752cade074753f4d7d69f19f2f783e1d7846a2c81beb4c9cab4f93a9aa7d44d6b8017440af5141c33775d6d

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,61 @@
+require 'set'
+
+module Mokkku
+  class UniqueUtils
+    RetryLimitExceeded = Class.new(StandardError)
+
+    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?
+          raise RetryLimitExceeded, "Retry limit exceeded for #{name}"
+        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.0'
     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.0
 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
@@ -124,7 +125,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.10
+rubygems_version: 3.2.32
 signing_key:
 specification_version: 4
 summary: Generate mock data that is unique for your application