tng 0.2.6 → 0.2.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 32a8b662c4dfe266510a9191d5cb6fbb2b0d52372db4e020884967efeb761c52
-  data.tar.gz: f91b5768b12f8715fc400cef1da192d1a5b9e2ca7c941095f108b438e1674499
+  metadata.gz: 39e06ff3d2ea9fdf295f87ac41e01e3fef4365899a128d7e013d913c568253f4
+  data.tar.gz: dd42840fd11e8c165b9777e540ae85c5eaa7047fd3b3923299cab3392431d0ee
 SHA512:
-  metadata.gz: 670688569bc80522d0a55b1c1625428443a015d1a1432dd8ce1851e6d7ca30dbfa528cee00d961a6b67ffeb9493678c34678766541de16596571e07448a52e1a
-  data.tar.gz: 9bce7c524936fcf7d8c8a06f7c55c546c7ea4e1757034e7812c3eb721a363bade6086b3e16ac97db64ffce1399d5c1df4c9702c90bfd656369a4456895621535
+  metadata.gz: 188224ee7db41585b1f4bf6737a549ba6a2b7830b07b8ba31959a98b930901f07979a1e321d00fc129dd82321c8c35686571462c152534c1f7537f5c2f4f563c
+  data.tar.gz: 0c5f179a3a833042e48cf06df02bf878e0ab7478975bbe328920831994d1097cd4fbf093bb3e8f6be236a7e172027d704f5746fc2e8272179e475b0132622621

data/README.md

@@ -124,9 +124,6 @@ Tng.configure do |config|
   config.api_key = ENV["TNG_API_KEY"]
   config.base_url = "https://app.tng.sh/"
   
-  config.read_source_code = true
-  config.read_test_code = true
-  
   config.testing_framework = "rspec"
   config.assertion_style = "expect"
   config.describe_style = true
@@ -189,10 +186,6 @@ end
 Tng.configure do |config|
   config.api_key = ENV["TNG_API_KEY"]
   config.base_url = "https://app.tng.sh/"
-  
-  config.read_source_code = true
-  config.read_test_code = true
-  
   config.testing_framework = "rspec"
   config.assertion_style = "expect"
   config.let_style = true

data/bin/tng

@@ -130,15 +130,15 @@ class CLI
   def preprocess_arguments(argv)
     normalized = []
     positional_args = []
-    
+
     argv.each_with_index do |arg, index|
       case arg
       when /^(?:--)?(file|f)=(.+)$/
-        normalized << "--file=#{$2}"
+        normalized << "--file=#{::Regexp.last_match(2)}"
       when /^(?:--)?(method|m)=(.+)$/
-        normalized << "--method=#{$2}"
+        normalized << "--method=#{::Regexp.last_match(2)}"
       when /^(help|h)=(.+)$/
-        normalized << "--help=#{$2}"
+        normalized << "--help=#{::Regexp.last_match(2)}"
       when /^--file$/, /^-f$/
         normalized << arg
       when /^--method$/, /^-m$/
@@ -156,29 +156,25 @@ class CLI
         end
       end
     end
-    
+
     if positional_args.length >= 2
       has_file = normalized.any? { |a| a.match?(/^--file/) }
       has_method = normalized.any? { |a| a.match?(/^--method/) }
-      
-      unless has_file
-        normalized << "--file=#{positional_args[0]}"
-      end
-      unless has_method
-        normalized << "--method=#{positional_args[1]}"
-      end
+
+      normalized << "--file=#{positional_args[0]}" unless has_file
+      normalized << "--method=#{positional_args[1]}" unless has_method
     elsif positional_args.length == 1
       arg = positional_args[0]
       has_file = normalized.any? { |a| a.match?(/^--file/) }
       has_method = normalized.any? { |a| a.match?(/^--method/) }
-      
-      if !has_file && (arg.end_with?('.rb') || arg.include?('/'))
+
+      if !has_file && (arg.end_with?(".rb") || arg.include?("/"))
         normalized << "--file=#{arg}"
-      elsif !has_method && !arg.include?('/')
+      elsif !has_method && !arg.include?("/")
         normalized << "--method=#{arg}"
       end
     end
-    
+
     normalized
   end
 

data/lib/generators/tng/install_generator.rb

@@ -3,6 +3,7 @@
 require "rails/generators"
 require "yaml"
 require "tng"
+require "pathname"
 
 module Tng
   module Generators
@@ -18,6 +19,7 @@ module Tng
         @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"
 
@@ -61,6 +63,8 @@ module Tng
         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",
           "",
@@ -70,8 +74,6 @@ module Tng
           "  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/\"",
-          "  config.read_source_code = true      # Options: true, false",
-          "  config.read_test_code = true        # Options: true, false",
           "",
           "  # Testing Framework",
           "  config.testing_framework = \"#{@test_framework}\"      # Options: minitest, rspec",
@@ -80,6 +82,12 @@ module Tng
           "  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}",
+          "",
           "  # Authentication#{auth_comment}",
           "  #{auth_enabled}      # Options: true, false",
           "  #{auth_lib}       # Options: devise, clearance, sorcery, nil",
@@ -190,6 +198,156 @@ module Tng
         "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))
+          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"
           {

data/lib/tng/api/http_client.rb

@@ -51,6 +51,19 @@ module Tng
       response
     end
 
+    def patch(path, payload: {}, headers: {})
+      merged_headers = json_default_headers.merge(headers)
+
+      response = HTTPX.with(timeout: @timeout).patch(
+        "#{@api_endpoint}/#{path}",
+        json: payload,
+        headers: merged_headers
+      )
+
+      debug_response("PATCH #{path}", response) if debug_enabled?
+      response
+    end
+
     def ping
       response = HTTPX.with(timeout: @timeout).get("#{@api_endpoint}/ping")
       debug_response("GET /ping", response) if debug_enabled?

data/lib/tng/services/test_generator.rb

@@ -76,6 +76,8 @@ module Tng
         file_result = Tng::Utils.save_test_file(result.to_json)
         return unless file_result
 
+        trigger_cleanup(job_id)
+
         file_result.merge(generation_time: generation_time)
       end
 
@@ -204,6 +206,12 @@ module Tng
         end
       end
 
+      def trigger_cleanup(job_id)
+        @http_client.patch("#{CONTENT_RESPONSES_PATH}/#{job_id}/cleanup")
+      rescue StandardError => e
+        debug_log("Cleanup request failed: #{e.message}") if debug_enabled?
+      end
+
       def debug_enabled?
         ENV["DEBUG"] == "1"
       end

data/lib/tng/version.rb

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

data/lib/tng.rb

@@ -53,8 +53,6 @@ module Tng
   @config = {
     api_key: nil,
     base_url: "https://app.tng.sh/",
-    read_source_code: true,
-    read_test_code: true,
     testing_framework: "minitest",
     authentication_enabled: false,
     authorization_library: nil,
@@ -62,7 +60,8 @@ module Tng
     authentication_methods: [],
     mock_library: "minitest/mock",
     http_mock_library: "webmock",
-    factory_library: "active_record"
+    factory_library: "active_record",
+    test_examples: []
   }
 
   def self.configure
@@ -89,22 +88,6 @@ module Tng
     @config[:base_url]
   end
 
-  def self.read_source_code=(value)
-    @config[:read_source_code] = value
-  end
-
-  def self.read_source_code
-    @config[:read_source_code]
-  end
-
-  def self.read_test_code=(value)
-    @config[:read_test_code] = value
-  end
-
-  def self.read_test_code
-    @config[:read_test_code]
-  end
-
   def self.testing_framework=(value)
     @config[:testing_framework] = value
     initialize_framework_defaults(value)
@@ -258,6 +241,14 @@ module Tng
     @config[:subject_style]
   end
 
+  def self.test_examples=(value)
+    @config[:test_examples] = value
+  end
+
+  def self.test_examples
+    @config[:test_examples]
+  end
+
   def self.authentication_configured?
     return false unless authentication_enabled
     return false if authentication_methods.nil? || authentication_methods.empty?

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tng
 version: !ruby/object:Gem::Version
-  version: 0.2.6
+  version: 0.2.8
 platform: ruby
 authors:
 - ralucab
@@ -287,7 +287,7 @@ post_install_message: "┌ TNG ────────────────
   \                                                            │\n│  • bundle exec
   tng --help  - Show help information                 │\n│                                                                    │\n│
   \ \U0001F4A1 Generate tests for individual methods with precision            │\n└────────────────────────────────────────────────────────────
-  v0.2.6 ┘\n"
+  v0.2.8 ┘\n"
 rdoc_options: []
 require_paths:
 - lib