tng 0.3.3 → 0.3.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c3b187572f19a15aeb81dfd5be26c940e0e6d2c5a67e9a03a9df7234f11f7850
-  data.tar.gz: aac3a342c8560bead38235275e0c76fec3b53ff2fb1bcec4d29116404031e4b0
+  metadata.gz: ca324d4cd07a9428327d1d75b4d75bfa1aaaf79595b9b308f49b230c7741f50c
+  data.tar.gz: f2e6aab56415c983e235a0f4fe6c9ca20f8bdea6d1441941896bf21fbb1941a1
 SHA512:
-  metadata.gz: 1029c4728e0de909aa0598b5e5279289508c6a64429aac80004bacce00afadfd7a06d954f5790d0e8fba5a5849966a5e279d0cc72bb0bbc2850aae3b9e345c97
-  data.tar.gz: 03dfc617af94c583099e815a834fc90e9126b29343bf8a06c806f6afd2aa4955b36ec1c3a9bee434b2bc9ffd170cfbffd810c0f35b3b6ede171a7b4b0e3d3f78
+  metadata.gz: 05d121c10f312e0fb33d781e312b7f5385001bb6886ee023df6c0e8ff22037501f0e06850e95fe9ca6c83bf6579ea2ff166ab324ac84313aa8c39e6b952609a1
+  data.tar.gz: bf91663c5390fa85787eeaf5b9e290ce4b32c53cd8a29e02c7d82e0d0bc765c5155a8ced68e4d50e5b35eb273896ba3e504a52fd07b87e0f1bb4cb3aea0e76a4

data/Rakefile

@@ -114,6 +114,26 @@ task :build_both do
   system("./build_both.sh") || (puts "❌ Build failed"; exit 1)
 end
 
+task :verify_binaries do
+  puts "🔍 Verifying binary artifacts..."
+  required_go = %w[go-ui-darwin-amd64 go-ui-darwin-arm64 go-ui-linux-amd64 go-ui-linux-arm64]
+  missing = required_go.reject { |f| File.exist?("binaries/#{f}") }
+
+  if missing.any?
+    abort "❌ Cannot release: Missing binaries in binaries/: #{missing.join(', ')}\nRun 'rake build_both' or './build_both.sh' first!"
+  end
+  puts "✅ All required binaries present."
+end
+
+# Add this task to suppress the native artifact warning
+task release: :build_both do
+  # Temporarily rename the binary to avoid the warning during gem build
+  puts "📦 Building gem with pre-compiled binaries..." if File.exist?("binaries/tng.bundle")
+end
+
+# Ensure verify_binaries runs before release
+task release: :verify_binaries
+
 desc "Development build: compile for current OS and build gem"
 task :dev do
   puts "🧹 Cleaning Rust binaries (preserving go-ui)..."

data/bin/tng

@@ -340,7 +340,8 @@ class CLI
       progress.update("Extracting method context...", 60)
       progress.update("Generating method test code...", 80)
 
-      result = Tng::Services::TestGenerator.new(@http_client).run_for_controller_method(controller, method_info, progress: progress)
+      result = Tng::Services::TestGenerator.new(@http_client).run_for_controller_method(controller, method_info,
+                                                                                        progress: progress)
 
       progress.update("Writing test file...", 100)
 
@@ -396,7 +397,8 @@ class CLI
       progress.update("Analyzing method dependencies...", 50)
       progress.update("Generating method test code...", 75)
 
-      result = Tng::Services::TestGenerator.new(@http_client).run_for_model_method(model, method_info, progress: progress)
+      result = Tng::Services::TestGenerator.new(@http_client).run_for_model_method(model, method_info,
+                                                                                   progress: progress)
 
       progress.update("Writing test file...", 100)
 
@@ -416,7 +418,8 @@ class CLI
       progress.update("Analyzing method dependencies...", 50)
       progress.update("Generating method test code...", 75)
 
-      result = Tng::Services::TestGenerator.new(@http_client).run_for_service_method(service, method_info, progress: progress)
+      result = Tng::Services::TestGenerator.new(@http_client).run_for_service_method(service, method_info,
+                                                                                     progress: progress)
 
       progress.update("Writing test file...", 100)
 
@@ -478,7 +481,8 @@ class CLI
       progress.update("Analyzing method dependencies...", 50)
       progress.update("Generating method test code...", 75)
 
-      result = Tng::Services::TestGenerator.new(@http_client).run_for_other_method(other_file, method_info, progress: progress)
+      result = Tng::Services::TestGenerator.new(@http_client).run_for_other_method(other_file, method_info,
+                                                                                   progress: progress)
 
       progress.update("Writing test file...", 100)
 
@@ -653,8 +657,6 @@ class CLI
     end
   end
 
-
-
   def copy_to_clipboard(command)
     if RUBY_PLATFORM.include?("darwin")
       IO.popen("pbcopy", "w") { |io| io.print command }
@@ -671,8 +673,6 @@ class CLI
     end
   end
 
-
-
   def initialize_config_and_clients
     @config_initialized = true
 

data/lib/generators/tng/install_generator.rb

@@ -1,399 +1,70 @@
 # frozen_string_literal: true
 
 require "rails/generators"
-require "yaml"
-require "tng"
-require "pathname"
 
 module Tng
   module Generators
     class InstallGenerator < Rails::Generators::Base
-      desc "Creates a tng.sh Ruby configuration file with comprehensive authentication and authorization support"
-
-      def self.source_root
-        @source_root ||= File.expand_path("templates", __dir__)
-      end
+      desc "Creates a TNG configuration file"
 
       def create_tng_configuration
-        @test_framework = detect_test_framework
-        @authentication_library = detect_authentication_library
-        @authz_library = detect_authorization_library
-        @factory_library = detect_factory_library
-        @test_examples = detect_test_examples
-
         initializer_path = "config/initializers/tng.rb"
 
         if File.exist?(initializer_path)
           say "Configuration file already exists at #{initializer_path}", :yellow
           if yes?("Do you want to overwrite it? (y/n)")
             remove_file initializer_path
-            create_file initializer_path, ruby_configuration_template
+            create_file initializer_path, configuration_template
             say "TNG configuration file updated!", :green
           else
             say "Skipping configuration file creation.", :blue
-            nil
           end
         else
-          create_file initializer_path, ruby_configuration_template
-          say "tng.sh configuration file created at #{initializer_path}", :green
+          create_file initializer_path, configuration_template
+          say "TNG configuration file created at #{initializer_path}", :green
         end
       end
 
-      def ruby_configuration_template
-        framework_config = generate_framework_config(@test_framework)
-
-        if @authentication_library && @authentication_library != "none"
-          auth_enabled = "config.authentication_enabled = true"
-          auth_comment = " (detected: #{@authentication_library})"
-          auth_lib = "config.authentication_library = \"#{@authentication_library}\""
-        else
-          auth_enabled = "config.authentication_enabled = true"
-          auth_comment = ""
-          auth_lib = "config.authentication_library = nil"
-        end
-
-        if @authz_library && @authz_library != "none"
-          authz_lib = "config.authorization_library = \"#{@authz_library}\""
-          authz_comment = " (detected: #{@authz_library})"
-        else
-          authz_lib = "config.authorization_library = nil"
-          authz_comment = ""
-        end
-
-        factory_lib = "config.factory_library = \"#{@factory_library}\""
-
-        framework_specific = generate_framework_specific_config(@test_framework, framework_config)
-
-        test_examples_config = @test_examples.any? ? @test_examples.inspect : "[]"
-        [
-          "# frozen_string_literal: true",
-          "",
-          "return unless Rails.env.development?",
-          "",
-          "Tng.configure do |config|",
-          "  config.api_key = nil",
-          "  # You dont need to change this url, unless you will instructed by the CLI.",
-          "  config.base_url = \"https://app.tng.sh/\"",
-          "",
-          "  # Testing Framework",
-          "  config.testing_framework = \"#{@test_framework}\"      # Options: minitest, rspec",
-          framework_specific,
-          "  config.mock_library = \"#{framework_config["mock_library"]}\"      # Options: mocha, minitest/mock, rspec-mocks, nil",
-          "  config.http_mock_library = \"#{framework_config["http_mock_library"]}\"       # Options: webmock, vcr, httparty, nil",
-          "  #{factory_lib}        # Options: factory_bot, factory_girl, fabrication, fabricator, fixtures, active_record",
-          "",
-          "  # Test Examples",
-          "  # Example test files for LLM to learn patterns and reduce hallucinations",
-          "  # Format: [{{\"name\" => \"test_name\", \"path\" => \"spec/models/user_spec.rb\"}}]",
-          "  # Leave empty [] to auto-detect from project",
-          "  config.test_examples = #{test_examples_config}",
-          "",
-          "  # Source Code Reading Configuration",
-          "  # When enabled (true), TNG will only read the file where the method is located",
-          "  # and will not analyze other files in the project. This may increase the accuracy of the tests,",
-          "  # but it may also increase the time it takes to generate the tests.",
-          "  config.read_file_source_code = false      # Options: true, false",
-          "",
-          "  # Authentication#{auth_comment}",
-          "  #{auth_enabled}      # Options: true, false",
-          "  #{auth_lib}       # Options: devise, clearance, sorcery, nil",
-          "",
-          "",
-          "  # ⚠️  IMPORTANT: AUTHENTICATION CONFIGURATION REQUIRED ⚠️",
-          "  # You MUST configure your authentication methods below for TNG to work properly.",
-          "  # Uncomment and modify the authentication_methods configuration:",
-          "",
-          "  # Authentication Methods (multiple methods supported)",
-          "  # Supported authentication types: session, devise, jwt, token_auth, basic_auth, oauth, headers, custom, nil",
-          "  # EXAMPLE: Uncomment and modify these examples to match your app's authentication:",
-          "",
-          "  # config.authentication_methods = [",
-          "    # {",
-          "    #   method: \"authenticate_user_via_session!\",",
-          "    #   file_location: \"app/controllers/application_controller.rb\",",
-          "    #   auth_type: \"session\"",
-          "    # },",
-          "    # {",
-          "    #   method: \"authenticate_user_via_api_key!\",",
-          "    #   file_location: \"app/controllers/application_controller.rb\",",
-          "    #   auth_type: \"headers\"",
-          "    # }",
-          "  # ]",
-          "  # ⚠️  Remember to configure your authentication methods above! ⚠️",
-          "",
-          "  # Authorization#{authz_comment}",
-          "  #{authz_lib}      # Options: cancancan, pundit, rolify, nil",
-          "end"
-        ].join("\n") + "\n"
-      end
-
       private
 
-      def detect_test_framework
-        rails_root = defined?(Rails) && Rails.root ? Rails.root.to_s : Dir.pwd
-
-        return "rspec" if File.exist?(File.join(rails_root, ".rspec")) ||
-                          File.exist?(File.join(rails_root, "spec", "spec_helper.rb")) ||
-                          Dir.exist?(File.join(rails_root, "spec"))
-
-        return "minitest" if Dir.exist?(File.join(rails_root, "test")) ||
-                             File.exist?(File.join(rails_root, "test", "test_helper.rb"))
-
-        begin
-          if defined?(Bundler)
-            gemfile_specs = Bundler.load.specs
-            return "rspec" if gemfile_specs.any? { |spec| spec.name == "rspec-rails" }
-            return "minitest" if gemfile_specs.any? { |spec| spec.name == "minitest" }
-          end
-        rescue StandardError
-          # Fallback if Bundler isn't available
-        end
-
-        "minitest" # Default fallback
-      end
-
-      def detect_authentication_library
-        return unless defined?(Bundler)
-
-        begin
-          gemfile_specs = Bundler.load.specs
-          return "devise" if gemfile_specs.any? { |spec| spec.name == "devise" }
-          return "clearance" if gemfile_specs.any? { |spec| spec.name == "clearance" }
-          return "sorcery" if gemfile_specs.any? { |spec| spec.name == "sorcery" }
-        rescue StandardError
-          # Fallback if Bundler isn't available
-          nil
-        end
-
-        nil
-      end
-
-      def detect_authorization_library
-        return unless defined?(Bundler)
-
-        gemfile_specs = Bundler.load.specs
-        return "cancancan" if gemfile_specs.any? { |spec| spec.name == "cancancan" }
-        return "pundit" if gemfile_specs.any? { |spec| spec.name == "pundit" }
-        return "rolify" if gemfile_specs.any? { |spec| spec.name == "rolify" }
-
-        nil
-      end
-
-      def detect_factory_library
-        return "active_record" unless defined?(Bundler)
-
-        rails_root = defined?(Rails) && Rails.root ? Rails.root.to_s : Dir.pwd
-
-        # Check for fixtures in test/fixtures or spec/fixtures
-        test_fixtures_path = File.join(rails_root, "test", "fixtures")
-        spec_fixtures_path = File.join(rails_root, "spec", "fixtures")
-
-        return "fixtures" if Dir.exist?(test_fixtures_path) && !Dir.empty?(test_fixtures_path)
-
-        return "fixtures" if Dir.exist?(spec_fixtures_path) && !Dir.empty?(spec_fixtures_path)
-
-        gemfile_specs = Bundler.load.specs
-        return "factory_bot" if gemfile_specs.any? { |spec| spec.name.match(/factory_bot/) }
-        return "factory_girl" if gemfile_specs.any? { |spec| spec.name.match(/factory_girl/) }
-        return "fabrication" if gemfile_specs.any? { |spec| spec.name.match(/fabrication/) }
-        return "fabricator" if gemfile_specs.any? { |spec| spec.name.match(/fabricator/) }
-
-        "active_record"
-      rescue StandardError
-        # Fallback if Bundler isn't available
-        "active_record"
-      end
-
-      def detect_test_examples
-        rails_root = defined?(Rails) && Rails.root ? Rails.root.to_s : Dir.pwd
-        examples = []
-
-        # Determine test directory based on framework
-        test_dir = @test_framework == "rspec" ? "spec" : "test"
-
-        # Comprehensive exclusion list for build artifacts, caches, dependencies
-        exclude_dirs = [
-          ".git", "log", "tmp", "vendor", "node_modules",
-          "coverage", "public", "storage", "db", "config",
-          "lib/tasks", "bin", ".bundle", ".vscode", ".idea"
-        ]
-
-        # Also exclude files larger than 500KB (probably not good examples)
-        max_file_size = 500 * 1024 # 500KB
-
-        # Find all test directories and subdirectories
-        test_directories = find_test_directories(rails_root, test_dir, exclude_dirs)
-
-        # Determine collection strategy based on folder structure
-        main_test_dir = File.join(rails_root, test_dir)
-        has_subfolders = test_directories.size > 1
-
-        if has_subfolders
-          # If we have subfolders, take 2 files per directory
-          collected_files = []
-          test_directories.each do |dir_path|
-            dir_files = find_valid_test_files_in_directory(dir_path, max_file_size)
-            # Take up to 2 files from this directory
-            collected_files.concat(dir_files.first(2))
+      def configuration_template
+        <<~RUBY
+          # frozen_string_literal: true
+
+          return unless Rails.env.development?
+
+          Tng.configure do |config|
+            config.api_key = ENV["TNG_API_KEY"]
+            config.base_url = ENV["API_BASE_URI"] || "https://app.tng.sh/"
+
+            # Test Helper File (optional - auto-detected if not set)
+            # Uncomment to override auto-detection with a custom path
+            # Default auto-detection checks: spec/rails_helper.rb, spec/spec_helper.rb, test/test_helper.rb
+            # config.test_helper_path = "spec/rails_helper.rb"
+
+            # ⚠️  IMPORTANT: AUTHENTICATION CONFIGURATION REQUIRED ⚠️
+            # You MUST configure your authentication methods below for TNG to work properly.
+            # Uncomment and modify the authentication_methods configuration:
+
+            # Authentication Methods (multiple methods supported)
+            # Supported authentication types: session, devise, jwt, token_auth, basic_auth, oauth, headers, custom, nil
+            # EXAMPLE: Uncomment and modify these examples to match your app's authentication:
+
+            # config.authentication_methods = [
+            #   {
+            #     method: "authenticate_user_via_session!",
+            #     file_location: "app/controllers/application_controller.rb",
+            #     auth_type: "session"
+            #   },
+            #   {
+            #     method: "authenticate_user_via_api_key!",
+            #     file_location: "app/controllers/application_controller.rb",
+            #     auth_type: "headers"
+            #   }
+            # ]
+            # ⚠️  Remember to configure your authentication methods above! ⚠️
           end
-        else
-          # If no subfolders, take up to 5 files from main test directory
-          collected_files = find_valid_test_files_in_directory(main_test_dir, max_file_size).first(5)
-        end
-
-        # Sort all collected files by modification time (recent first), then by size (smaller files first)
-        collected_files.sort_by! { |f| [-File.mtime(f).to_i, File.size(f)] }
-
-        # Convert to examples format
-        collected_files.first(5).each do |test_file|
-          examples << {
-            "name" => File.basename(test_file),
-            "path" => Pathname.new(test_file).relative_path_from(Pathname.new(rails_root)).to_s
-          }
-        rescue StandardError
-          next
-        end
-
-        examples
-      rescue StandardError
-        # Return empty array if anything goes wrong
-        []
-      end
-
-      def find_test_directories(rails_root, test_dir, exclude_dirs)
-        directories = []
-        main_test_path = File.join(rails_root, test_dir)
-
-        return [] unless Dir.exist?(main_test_path)
-
-        # Add main test directory
-        directories << main_test_path
-
-        # Find all subdirectories recursively
-        Dir.glob(File.join(main_test_path, "**/*/")).each do |dir|
-          # Skip excluded directories
-          next if exclude_dirs.any? { |excluded| dir.include?("/#{excluded}/") }
-
-          # Skip if it's not a directory with actual test files
-          next unless test_files?(dir)
-
-          directories << dir
-        end
-
-        directories.uniq
-      end
-
-      def test_files?(dir_path)
-        return false unless Dir.exist?(dir_path)
-
-        pattern = @test_framework == "rspec" ? "*_spec.rb" : "*_test.rb"
-        Dir.glob(File.join(dir_path, pattern)).any?
-      end
-
-      def find_valid_test_files_in_directory(dir_path, max_file_size)
-        return [] unless Dir.exist?(dir_path)
-
-        pattern = @test_framework == "rspec" ? "*_spec.rb" : "*_test.rb"
-        files = []
-
-        Dir.glob(File.join(dir_path, pattern)).each do |test_file|
-          next if File.size(test_file) > max_file_size
-          next unless valid_test_file?(test_file)
-
-          files << test_file
-        rescue StandardError
-          next
-        end
-
-        files
-      end
-
-      def valid_test_file?(file_path)
-        return false unless File.exist?(file_path) && File.readable?(file_path)
-        return false if File.size(file_path) == 0
-
-        begin
-          # Read first 1KB of the file
-          content = File.read(file_path, 1024)
-
-          # Check for Ruby test indicators based on framework
-          test_indicators = if @test_framework == "rspec"
-                              [
-                                "describe ",      # RSpec describe blocks
-                                "context ",       # RSpec context blocks
-                                "it ",            # RSpec examples
-                                "specify ",       # RSpec specify
-                                "expect(",        # RSpec expectations
-                                "before(",        # RSpec hooks
-                                "let(",           # RSpec let
-                                "subject ",       # RSpec subject
-                                'require "spec_helper"', # RSpec spec helper
-                                'require "rails_helper"' # RSpec rails helper
-                              ]
-                            else # minitest
-                              [
-                                "def test_",      # Minitest test methods
-                                "class.*Test",    # Test classes
-                                "assert ",        # Assertions
-                                "refute ",        # Refutations
-                                'require "test_helper"', # Minitest helper
-                                'require "minitest"', # Minitest require
-                                "MiniTest::Test" # Minitest base class
-                              ]
-                            end
-
-          # Must contain at least one test indicator
-          return false unless test_indicators.any? { |indicator| content.include?(indicator) }
-
-          # Basic Ruby syntax check - should not have obvious non-Ruby content
-          return false if content.scan(/#!/).size > 5 || content.include?("<?xml")
-
-          true
-        rescue StandardError
-          false
-        end
-      end
-
-      def generate_framework_config(framework)
-        if framework == "minitest"
-          {
-            "test_style" => "test_block", # Options: spec, unit, test_block
-            "setup_style" => true, # Options: true, false
-            "assertion_style" => "assert/refute", # Options: assert/refute, assert/assert_not, must/wont
-            "teardown_style" => false, # Options: true, false
-            "http_mock_library" => "webmock", # Options: webmock, vcr, httparty, none
-            "mock_library" => "minitest/mock" # Options: mocha, minitest/mock, rspec-mocks, none
-          }
-        else # rspec
-          {
-            "describe_style" => true, # Options: true, false
-            "context_style" => "context", # Options: context, describe
-            "it_style" => "it", # Options: it, specify
-            "before_style" => "before", # Options: before, setup
-            "after_style" => "after", # Options: after, teardown
-            "let_style" => true, # Options: true, false
-            "subject_style" => true, # Options: true, false
-            "mock_library" => "rspec-mocks", # Options: rspec-mocks, mocha, flexmock, none
-            "http_mock_library" => "webmock" # Options: webmock, vcr, httparty, none
-          }
-        end
-      end
-
-      def generate_framework_specific_config(framework, framework_config)
-        if framework == "minitest"
-          "  config.test_style = \"#{framework_config["test_style"]}\"           # Options: spec, unit, test_block\n" +
-            "  config.setup_style = #{framework_config["setup_style"]}                  # Options: true, false\n" +
-            "  config.assertion_style = \"#{framework_config["assertion_style"]}\"   # Options: assert/refute, assert/assert_not, must/wont\n" +
-            "  config.teardown_style = #{framework_config["teardown_style"]}              # Options: true, false"
-        else # rspec
-          "  config.describe_style = #{framework_config["describe_style"]}                  # Options: true, false\n" +
-            "  config.context_style = \"#{framework_config["context_style"]}\"                  # Options: context, describe\n" +
-            "  config.it_style = \"#{framework_config["it_style"]}\"                  # Options: it, specify\n" +
-            "  config.before_style = \"#{framework_config["before_style"]}\"                  # Options: before, setup\n" +
-            "  config.after_style = \"#{framework_config["after_style"]}\"                  # Options: after, teardown\n" +
-            "  config.let_style = #{framework_config["let_style"]}                  # Options: true, false\n" +
-            "  config.subject_style = #{framework_config["subject_style"]}                  # Options: true, false"
-        end
+        RUBY
       end
     end
   end

data/lib/tng/analyzers/model.rb

@@ -30,7 +30,8 @@ module Tng
           # Load the model class
           model_class = model_name.constantize
 
-          instance_methods = model_class.public_instance_methods(false)
+          instance_methods = model_class.public_instance_methods(false) +
+                              model_class.private_instance_methods(false)
           class_methods = model_class.public_methods(false) - Class.public_methods
 
           model_file = Object.const_source_location(model_class.name)&.first

data/lib/tng/analyzers/other.rb

@@ -79,8 +79,10 @@ module Tng
 
         case node
         when Prism::DefNode
-          # Only add methods that are public
-          methods << node.name.to_s if current_visibility == :public
+          # Add methods that are public, OR specifically 'initialize' (which is conventionally private but often needs testing)
+          if current_visibility == :public || node.name.to_s == "initialize"
+            methods << node.name.to_s
+          end
         when Prism::CallNode
           # Handle visibility modifiers (private, protected, public)
           if node.receiver.nil? && node.arguments.nil?

data/lib/tng/analyzers/service.rb

@@ -31,7 +31,8 @@ module Tng
           # Load the service class
           service_class = service_name.constantize
 
-          instance_methods = service_class.public_instance_methods(false)
+          instance_methods = service_class.public_instance_methods(false) +
+                              service_class.private_instance_methods(false)
           class_methods = service_class.public_methods(false) - Class.public_methods
 
           # Try to get source file from any method, fallback to const_source_location

data/lib/tng/services/test_generator.rb

@@ -36,7 +36,6 @@ module Tng
         generate_test_for_type(other_file, method_info, :other, progress: progress)
       end
 
-
       def generate_test_for_type(file_object, method_info, type, progress: nil)
         start_time = Time.now
 
@@ -50,14 +49,11 @@ module Tng
         job_data = JSON.parse(response.body)
 
         # Check for authentication errors
-        if response.status == 401
-          return { error: :auth_failed, message: "Invalid or missing API key" }
-        end
+        return { error: :auth_failed, message: "Invalid or missing API key" } if response.status == 401
 
         # Check for forbidden responses (usage limits or invalid keys)
-        if response.status == 403
-          return { error: :auth_failed, message: "API key expired or usage limit reached" }
-        end
+        return { error: :auth_failed, message: "API key expired or usage limit reached" } if response.status == 403
+
         job_id = job_data["job_id"]
 
         return unless job_id
@@ -114,15 +110,15 @@ module Tng
           # We create 4 generic steps for the agents
           # Note: indices are relative to the current session, so we just append them.
           # But we need their indices to update them later.
-          # Since we can't ask progress for current index easily without hacking, 
+          # Since we can't ask progress for current index easily without hacking,
           # we rely on the fact that we call update 4 times.
-          
+
           # Use a base offset if we could know it, but we can't reliably.
           # Actually, if we use explicit_step, we need absolute indices.
           # Let's assume the previous steps were 0, 1, 2, 3 based on bin/tng.
           # So we start at 4.
-          base_idx = 4 
-          
+          base_idx = 4
+
           progress.update("Context Builder: Pending...", nil, step_increment: true)
           agent_step_indices["context_agent_status"] = base_idx
 
@@ -146,10 +142,10 @@ module Tng
 
           # Calculate pseudo-percentage
           if progress
-             pct = ((attempts.to_f / max_attempts.to_f) * 95.0).to_i
-             # Update global progress bar without adding a new step (explicit_step=4 updates Context Builder line, but we want to just update percentage?)
-             # The progress bar component takes the percent from the *last* update message.
-             # So we can pass specific percent when we update any agent step.
+            pct = ((attempts.to_f / max_attempts.to_f) * 95.0).to_i
+          # Update global progress bar without adding a new step (explicit_step=4 updates Context Builder line, but we want to just update percentage?)
+          # The progress bar component takes the percent from the *last* update message.
+          # So we can pass specific percent when we update any agent step.
           else
             pct = 0
           end
@@ -166,32 +162,32 @@ module Tng
           begin
             status_data = JSON.parse(status_response.body.to_s)
             status = status_data["status"]
-            
+
             # Update UI with granular info
             if progress && status_data["info"].is_a?(Hash)
               info = status_data["info"]
-              
+
               agent_step_indices.each do |key, step_idx|
-                 item_data = info[key]
-                 next unless item_data
-
-                 agent_status = "pending"
-                 values = []
-
-                 if item_data.is_a?(Hash)
-                   agent_status = item_data["status"] || "pending"
-                   values = item_data["values"] || []
-                 else
-                   agent_status = item_data.to_s
-                 end
-
-                 label = case key
-                         when "context_agent_status" then "Context Builder"
-                         when "style_agent_status" then "Style Analyzer"
-                         when "logical_issue_status" then "Logic Analyzer"
-                         when "behavior_expert_status" then "Logic Generator"
-                         else key
-                         end
+                item_data = info[key]
+                next unless item_data
+
+                agent_status = "pending"
+                values = []
+
+                if item_data.is_a?(Hash)
+                  agent_status = item_data["status"] || "pending"
+                  values = item_data["values"] || []
+                else
+                  agent_status = item_data.to_s
+                end
+
+                label = case key
+                        when "context_agent_status" then "Context Builder"
+                        when "style_agent_status" then "Style Analyzer"
+                        when "logical_issue_status" then "Logic Analyzer"
+                        when "behavior_expert_status" then "Logic Generator"
+                        else key
+                        end
 
                 msg = if agent_status == "processing"
                         "#{label}: Processing..."
@@ -213,8 +209,8 @@ module Tng
 
                 # Pass percentage only on the last step (Logic Generator) to keep main bar moving?
                 # Or pass it on all updates.
-                p = (step_idx == agent_step_indices["behavior_expert_status"]) ? pct : nil
-                
+                p = step_idx == agent_step_indices["behavior_expert_status"] ? pct : nil
+
                 progress.update(msg, p, step_increment: false, explicit_step: step_idx)
               end
             end
@@ -222,7 +218,7 @@ module Tng
             case status
             when "completed"
               debug_log("Test generation completed!") if debug_enabled?
-              return status_data["result"]
+              return status_data["result"].merge("analysis" => status_data["analysis"])
             when "failed"
               debug_log("Test generation failed: #{status_data["error"]}") if debug_enabled?
               return { error: :generation_failed, message: status_data["error"] }

data/lib/tng/services/user_app_config.rb

@@ -35,17 +35,6 @@ module Tng
         end
         Tng.config[:authentication_entry_points_with_source] = auth_entry_points_with_source
 
-        # Add source content to test examples for API requests
-        if Tng.config[:test_examples]&.any?
-          Tng.config[:test_examples] = Tng.config[:test_examples].map do |example|
-            next example unless example.is_a?(Hash) && example["path"]
-
-            example.merge("source" => read_test_file_content(example["path"]))
-          rescue StandardError
-            example
-          end
-        end
-
         Tng.config
       end
 

data/lib/tng/ui/go_ui_session.rb

@@ -381,8 +381,10 @@ module Tng
       def update(message, percent = nil, step_increment: true, explicit_step: nil)
         step_idx = if explicit_step
                      explicit_step
+                   elsif step_increment
+                     @step
                    else
-                     step_increment ? @step : (@step > 0 ? @step - 1 : 0)
+                     (@step > 0 ? @step - 1 : 0)
                    end
 
         data = { type: "step", step: step_idx, message: message }
@@ -403,4 +405,3 @@ module Tng
     end
   end
 end
-

data/lib/tng/utils.rb

@@ -80,9 +80,7 @@ module Tng
       puts "📋 Raw API response: #{test_content[0..200]}..." if ENV["DEBUG"]
       parsed_response = JSON.parse(test_content)
 
-      if parsed_response["error"]
-        return { error: :generation_failed, message: parsed_response["error"] }
-      end
+      return { error: :generation_failed, message: parsed_response["error"] } if parsed_response["error"]
       # Validate required fields
       unless parsed_response["file_content"]
         return { error: :invalid_response, message: "API response missing file_content" }
@@ -90,16 +88,16 @@ module Tng
 
       # Handle both possible field names for file path
       file_path = parsed_response["test_file_path"] || parsed_response["file_path"] || parsed_response["file_name"] || parsed_response["file"]
-      unless file_path
-        return { error: :invalid_response, message: "API response missing file path" }
-      end
+      return { error: :invalid_response, message: "API response missing file path" } unless file_path
 
       begin
-        File.write(file_path, parsed_response["file_content"])
+        clean_content = cleanup_file_content(parsed_response["file_content"])
+        File.write(file_path, clean_content)
       rescue Errno::ENOENT
         # Create directory if it doesn't exist
         FileUtils.mkdir_p(File.dirname(file_path))
-        File.write(file_path, parsed_response["file_content"])
+        clean_content = cleanup_file_content(parsed_response["file_content"])
+        File.write(file_path, clean_content)
       end
       absolute_path = File.expand_path(file_path)
 
@@ -123,22 +121,47 @@ module Tng
       }
     end
 
-    def self.fixture_content
-      factory_library = Tng.factory_library || "active_record"
-
-      case factory_library.downcase
-      when "fixtures"
-        load_all_fixtures_data
-      when "fabrication", "fabricator"
-        load_all_fabricator_data
-      when "factory_girl", "factory_bot"
-        load_all_factory_data
-      when "active_record"
-        load_all_active_record_data
-      else
-        puts "⚠️  Warning: Unknown factory library '#{factory_library}'. Using Active Record object creation as fallback."
-        load_all_fixtures_data
+    def self.cleanup_file_content(content)
+      return content unless content.is_a?(String)
+
+      cleaned = content.dup
+
+      # Strip trailing whitespace first
+      cleaned.rstrip!
+
+      # Remove trailing JSON artifacts that shouldn't be in Ruby code
+      # Common patterns: ends with `end"` or `end"\n}` or just `}`
+      loop do
+        original = cleaned.dup
+
+        # Remove trailing lone } or " that shouldn't be there
+        cleaned.sub!(/\n\s*\}\s*\z/, "\n")
+        cleaned.sub!(/"\s*\z/, "")
+        cleaned.sub!(/\n\s*"\s*\z/, "\n")
+        cleaned.sub!(/end"\s*\z/, "end")
+
+        break if cleaned == original
       end
+
+      # Ensure file ends with single newline
+      cleaned.rstrip!
+      cleaned << "\n" unless cleaned.empty?
+
+      cleaned
+    end
+
+    def self.fixture_content
+      # Auto-detect: try fixtures first, then factory_bot, then fabricator
+      fixtures = load_all_fixtures_data
+      return fixtures if fixtures.any?
+
+      factories = load_all_factory_data
+      return factories if factories.any?
+
+      fabricators = load_all_fabricator_data
+      return fabricators if fabricators.any?
+
+      {}
     end
 
     def self.load_all_fixtures_data

data/lib/tng/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Tng
-  VERSION = "0.3.3"
+  VERSION = "0.3.5"
 end

data/lib/tng.rb

@@ -66,16 +66,8 @@ module Tng
   @config = {
     api_key: nil,
     base_url: "https://app.tng.sh/",
-    testing_framework: "minitest",
-    authentication_enabled: false,
-    authorization_library: nil,
-    authentication_library: nil,
-    authentication_methods: [],
-    mock_library: "minitest/mock",
-    http_mock_library: "webmock",
-    factory_library: "active_record",
-    test_examples: [],
-    read_file_source_code: false
+    test_helper_path: nil,
+    authentication_methods: []
   }
 
   def self.configure
@@ -102,37 +94,12 @@ module Tng
     @config[:base_url]
   end
 
-  def self.testing_framework=(value)
-    @config[:testing_framework] = value
-    initialize_framework_defaults(value)
+  def self.test_helper_path=(value)
+    @config[:test_helper_path] = value
   end
 
-  def self.testing_framework
-    @config[:testing_framework]
-  end
-
-  def self.authentication_enabled=(value)
-    @config[:authentication_enabled] = value
-  end
-
-  def self.authentication_enabled
-    @config[:authentication_enabled]
-  end
-
-  def self.authentication_library=(value)
-    @config[:authentication_library] = value
-  end
-
-  def self.authentication_library
-    @config[:authentication_library]
-  end
-
-  def self.authorization_library=(value)
-    @config[:authorization_library] = value
-  end
-
-  def self.authorization_library
-    @config[:authorization_library]
+  def self.test_helper_path
+    @config[:test_helper_path]
   end
 
   def self.authentication_methods=(value)
@@ -142,178 +109,4 @@ module Tng
   def self.authentication_methods
     @config[:authentication_methods]
   end
-
-  def self.mock_library=(value)
-    @config[:mock_library] = value
-  end
-
-  def self.mock_library
-    @config[:mock_library]
-  end
-
-  def self.http_mock_library=(value)
-    @config[:http_mock_library] = value
-  end
-
-  def self.http_mock_library
-    @config[:http_mock_library]
-  end
-
-  def self.factory_library=(value)
-    @config[:factory_library] = value
-  end
-
-  def self.factory_library
-    @config[:factory_library]
-  end
-
-  def self.test_style=(value)
-    @config[:test_style] = value
-  end
-
-  def self.test_style
-    @config[:test_style]
-  end
-
-  def self.setup_style=(value)
-    @config[:setup_style] = value
-  end
-
-  def self.setup_style
-    @config[:setup_style]
-  end
-
-  def self.assertion_style=(value)
-    @config[:assertion_style] = value
-  end
-
-  def self.assertion_style
-    @config[:assertion_style]
-  end
-
-  def self.teardown_style=(value)
-    @config[:teardown_style] = value
-  end
-
-  def self.teardown_style
-    @config[:teardown_style]
-  end
-
-  def self.describe_style=(value)
-    @config[:describe_style] = value
-  end
-
-  def self.describe_style
-    @config[:describe_style]
-  end
-
-  def self.context_style=(value)
-    @config[:context_style] = value
-  end
-
-  def self.context_style
-    @config[:context_style]
-  end
-
-  def self.it_style=(value)
-    @config[:it_style] = value
-  end
-
-  def self.it_style
-    @config[:it_style]
-  end
-
-  def self.before_style=(value)
-    @config[:before_style] = value
-  end
-
-  def self.before_style
-    @config[:before_style]
-  end
-
-  def self.after_style=(value)
-    @config[:after_style] = value
-  end
-
-  def self.after_style
-    @config[:after_style]
-  end
-
-  def self.let_style=(value)
-    @config[:let_style] = value
-  end
-
-  def self.let_style
-    @config[:let_style]
-  end
-
-  def self.subject_style=(value)
-    @config[:subject_style] = value
-  end
-
-  def self.subject_style
-    @config[:subject_style]
-  end
-
-  def self.test_examples=(value)
-    @config[:test_examples] = value
-  end
-
-  def self.test_examples
-    @config[:test_examples]
-  end
-
-  def self.read_file_source_code=(value)
-    @config[:read_file_source_code] = value
-  end
-
-  def self.read_file_source_code
-    @config[:read_file_source_code]
-  end
-
-  def self.authentication_configured?
-    return false unless authentication_enabled
-    return false if authentication_methods.nil? || authentication_methods.empty?
-
-    authentication_methods.all? do |method|
-      method.is_a?(Hash) &&
-        method.key?(:method) && !method[:method].to_s.strip.empty? &&
-        method.key?(:file_location) && !method[:file_location].to_s.strip.empty? &&
-        method.key?(:auth_type) && !method[:auth_type].to_s.strip.empty?
-    end
-  end
-
-  def self.initialize_framework_defaults(framework)
-    if framework == "minitest"
-      @config.merge!({
-                       test_style: "spec",
-                       setup_style: false,
-                       assertion_style: "assert",
-                       teardown_style: false
-                     })
-      @config.delete(:describe_style)
-      @config.delete(:context_style)
-      @config.delete(:it_style)
-      @config.delete(:before_style)
-      @config.delete(:after_style)
-      @config.delete(:let_style)
-      @config.delete(:subject_style)
-    elsif framework == "rspec"
-      @config.merge!({
-                       describe_style: true,
-                       context_style: "context",
-                       it_style: "it",
-                       before_style: "before",
-                       after_style: "after",
-                       let_style: true,
-                       subject_style: true
-                     })
-      @config.delete(:test_style)
-      @config.delete(:setup_style)
-      @config.delete(:assertion_style)
-      @config.delete(:teardown_style)
-    end
-  end
-
-  initialize_framework_defaults(@config[:testing_framework])
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tng
 version: !ruby/object:Gem::Version
-  version: 0.3.3
+  version: 0.3.5
 platform: ruby
 authors:
 - ralucab
@@ -165,6 +165,7 @@ files:
 - binaries/tng-darwin-arm64.bundle
 - binaries/tng-linux-arm64.so
 - binaries/tng-linux-x86_64.so
+- binaries/tng.bundle
 - lib/generators/tng/install_generator.rb
 - lib/tng.rb
 - lib/tng/analyzers/controller.rb
@@ -194,33 +195,30 @@ metadata:
   source_code_uri: https://github.com/tng-sh/tng-rails-public
   license_uri: https://github.com/tng-sh/tng-rails-public/blob/main/LICENSE.md
 post_install_message: "┌ TNG ───────────────────────────────────────────────────────────────┐\n│
-  \                                                                   │\n│  \e[32;1m✅
-  Tng installed successfully!\e[0m                                     │\n│                                                                    │\n│
-  \ \e[31;1m\U0001F4CB SETUP REQUIRED\e[0m                                                  │\n│
-  \                                                                   │\n│  \e[37m1.
-  Generate configuration:\e[0m                                        │\n│  \e[34m
-  \  rails g tng:install\e[0m                                            │\n│                                                                    │\n│
-  \ \e[37m2. Edit configuration file:\e[0m                                       │\n│
-  \ \e[34m   config/initializers/tng.rb\e[0m                                     │\n│
-  \                                                                   │\n│  \e[37m3.
-  Add your license key:\e[0m                                          │\n│  \e[34m
-  \  config.api_key = 'your-license-key-here'\e[0m                       │\n│                                                                    │\n│
-  \ \e[37m\U0001F4CB Check documentation for the correct authentication setup\e[0m
-  \       │\n│                                                                    │\n│
-  \ \e[31;1m\U0001F680 Usage:\e[0m                                                          │\n│
-  \                                                                   │\n│  \e[37mInteractive
-  mode:\e[0m                                                 │\n│  \e[32m• bundle
-  exec tng\e[0m\e[37m  - Interactive CLI with method selection\e[0m        │\n│                                                                    │\n│
-  \ \e[37mFeatures:\e[0m                                                         │\n│
-  \ \e[32m• Test 20+ file types: Controllers, Models, Services + Jobs, \e[0m     │\n│
-  \ \e[32mHelpers, Lib, Policies, Presenters, Mailers, GraphQL, and more\e[0m    │\n│
-  \ \e[32m• Select specific methods to test\e[0m                                 │\n│
-  \ \e[32m• Search and filter through methods\e[0m                               │\n│
-  \                                                                   │\n│  \e[37mHelp:\e[0m
-  \                                                            │\n│  \e[32m• bundle
-  exec tng --help\e[0m\e[37m  - Show help information\e[0m                 │\n│                                                                    │\n│
-  \ \e[37m\U0001F4A1 Generate tests for individual methods with precision\e[0m            │\n└────────────────────────────────────────────────────────────
-  v0.3.3 ┘\n"
+  \                                                                   │\n│  ✅ Tng
+  installed successfully!                                     │\n│                                                                    │\n│
+  \ \U0001F4CB SETUP REQUIRED                                                  │\n│
+  \                                                                   │\n│  1. Generate
+  configuration:                                        │\n│     rails g tng:install
+  \                                           │\n│                                                                    │\n│
+  \ 2. Edit configuration file:                                       │\n│     config/initializers/tng.rb
+  \                                    │\n│                                                                    │\n│
+  \ 3. Add your license key:                                          │\n│     config.api_key
+  = 'your-license-key-here'                       │\n│                                                                    │\n│
+  \ \U0001F4CB Check documentation for the correct authentication setup        │\n│
+  \                                                                   │\n│  \U0001F680
+  Usage:                                                          │\n│                                                                    │\n│
+  \ Interactive mode:                                                 │\n│  • bundle
+  exec tng  - Interactive CLI with method selection        │\n│                                                                    │\n│
+  \ Features:                                                         │\n│  • Test
+  20+ file types: Controllers, Models, Services + Jobs,      │\n│  Helpers, Lib, Policies,
+  Presenters, Mailers, GraphQL, and more    │\n│  • Select specific methods to test
+  \                                │\n│  • Search and filter through methods                               │\n│
+  \                                                                   │\n│  Help:
+  \                                                            │\n│  • bundle exec
+  tng --help  - Show help information                 │\n│                                                                    │\n│
+  \ \U0001F4A1 Generate tests for individual methods with precision            │\n└────────────────────────────────────────────────────────────
+  v0.3.5 ┘\n"
 rdoc_options: []
 require_paths:
 - lib