tng 0.2.7 → 0.2.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c30c794fbce7688603e0535f1673b9c755627d090846074a55f49614eb61eb59
-  data.tar.gz: 487772b58677a33f9f54719bcac234a17f7be7905f0ef00001d1fae259e06704
+  metadata.gz: 5f6e0805bc5c9a7dbb6d83b8298fcbc099a14c70e118c02743545a29dfbc8182
+  data.tar.gz: 070b0bbc5c54a91b01a2e38113588e7838bb0ff1d8217e0f666695d953a7ea6a
 SHA512:
-  metadata.gz: 720a2884c77eb6aa515a9fdcc765554725c08a45afbf7cf278ed03025067d13200550acf86d50edc34334d503fd76d61ef6114d26b5b4d0e6884166f1e74c302
-  data.tar.gz: b607b077576e8cddc418e58ad2647ea8a1e5e2079ce9ad34420cde5a8b41c6d66d8a1f69f80d105365acb1818be161352b051a1d01e91b2e494b73dd2e5bafd4
+  metadata.gz: f46c75ee3f7fa1cc0e22bdb46687c600184e9f2435f8321be71bc1623c33c7a1494326d0cfcd2030017bbe94244f39869870a16f245d9abc3e2ee8b307dedf98
+  data.tar.gz: 1c971f70e79e8cf9d9bd99ce779e3d244bee955908aec4ab679fd7a72c491e68a624b2fa56ee8195ecef2e42ba2b3df2828bb7dcc8278caa5374692c68522807

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
 
@@ -831,7 +827,7 @@ class CLI
 
       info_msg = [
         @pastel.decorate("#{Tng::UI::Theme.icon(:config)} File: #{result[:file_path]}", Tng::UI::Theme.color(:info)),
-        @pastel.decorate("#{Tng::UI::Theme.icon(:marker)} Run: #{result[:run_command]}", Tng::UI::Theme.color(:warning))
+        @pastel.decorate("#{Tng::UI::Theme.icon(:marker)} Run: #{result[:run_command]}", Tng::UI::Theme.color(:secondary))
       ].join("\n")
       puts center_text(info_msg)
       puts
@@ -849,7 +845,7 @@ class CLI
 
       info_msg = [
         @pastel.decorate("#{Tng::UI::Theme.icon(:config)} File: #{result[:file_path]}", Tng::UI::Theme.color(:info)),
-        @pastel.decorate("#{Tng::UI::Theme.icon(:marker)} Run: #{result[:run_command]}", Tng::UI::Theme.color(:warning))
+        @pastel.decorate("#{Tng::UI::Theme.icon(:marker)} Run: #{result[:run_command]}", Tng::UI::Theme.color(:secondary))
       ].join("\n")
       puts center_text(info_msg)
       puts

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/analyzers/model.rb

@@ -33,32 +33,50 @@ module Tng
           instance_methods = model_class.public_instance_methods(false)
           class_methods = model_class.public_methods(false) - Class.public_methods
 
-          model_file = model_class.const_source_location(model_class.name.split("::").last)&.first
+          model_file = Object.const_source_location(model_class.name)&.first
 
-          model_methods = if model_file && File.exist?(model_file)
-                            source_code = File.read(model_file)
-                            result = Prism.parse(source_code)
+          if model_file && File.exist?(model_file)
+            source_code = File.read(model_file)
+            result = Prism.parse(source_code)
 
-                            defined_methods = []
-                            extract_method_names(result.value, defined_methods)
+            defined_methods = []
+            synthetic_methods = []
+            scopes = []
+            validations = []
+            extract_method_names(result.value, defined_methods, synthetic_methods, scopes, validations)
 
-                            filtered_instance_methods = instance_methods.select do |method_name|
-                              method = model_class.instance_method(method_name)
-                              next false unless method.owner == model_class
+            filtered_instance_methods = instance_methods.select do |method_name|
+              method = model_class.instance_method(method_name)
+              next false unless method.owner == model_class
 
-                              defined_methods.include?(method_name.to_s)
-                            end
+              defined_methods.include?(method_name.to_s)
+            end
 
-                            filtered_class_methods = class_methods.select do |method_name|
-                              defined_methods.include?(method_name.to_s)
-                            end
+            filtered_class_methods = class_methods.select do |method_name|
+              defined_methods.include?(method_name.to_s)
+            end
 
-                            filtered_instance_methods + filtered_class_methods
-                          else
-                            []
-                          end
+            # Return array of hashes with type information
+            instance_method_hashes = filtered_instance_methods.map do |name|
+              { name: name.to_s, type: "instance_method" }
+            end
+            class_method_hashes = filtered_class_methods.map do |name|
+              { name: name.to_s, type: "class_method" }
+            end
+            synthetic_method_hashes = synthetic_methods.map do |name|
+              { name: name.to_s, type: "synthetic" }
+            end
+            scope_hashes = scopes.map do |name|
+              { name: name.to_s, type: "scope" }
+            end
+            validation_hashes = validations.map do |name|
+              { name: name.to_s, type: "validation" }
+            end
 
-          model_methods.map { |method_name| { name: method_name.to_s } }
+            instance_method_hashes + class_method_hashes + synthetic_method_hashes + scope_hashes + validation_hashes
+          else
+            []
+          end
         rescue NameError => e
           puts "❌ Could not load model class #{model_name}: #{e.message}"
           []
@@ -68,7 +86,7 @@ module Tng
         end
       end
 
-      def self.extract_method_names(node, methods)
+      def self.extract_method_names(node, methods, synthetic_methods = [], scopes = [], validations = [])
         return unless node.is_a?(Prism::Node)
 
         case node
@@ -81,18 +99,37 @@ module Tng
             first_arg = node.arguments.arguments.first
             if first_arg.is_a?(Prism::SymbolNode)
               scope_name = first_arg.value
-              methods << scope_name if scope_name
+              scopes << scope_name if scope_name
+            end
+          # Handle validation macros
+          elsif node.name.to_s.start_with?("validate") && node.arguments&.arguments&.any?
+            if node.name == :validate
+              # Custom validation: validate :method_name -> extract actual method
+              node.arguments.arguments.each do |arg|
+                if arg.is_a?(Prism::SymbolNode)
+                  validation_method = arg.value
+                  validations << validation_method if validation_method
+                end
+              end
+            else
+              # Built-in validations: validates :attr, :attr2, ... -> add to validations
+              node.arguments.arguments.each do |arg|
+                if arg.is_a?(Prism::SymbolNode)
+                  attr_name = arg.value
+                  validations << attr_name if attr_name
+                end
+              end
             end
           end
         when Prism::SingletonClassNode
           # Methods inside class << self blocks
           node.body&.child_nodes&.each do |child|
-            extract_method_names(child, methods)
+            extract_method_names(child, methods, synthetic_methods, scopes, validations)
           end
         end
 
         node.child_nodes.each do |child|
-          extract_method_names(child, methods)
+          extract_method_names(child, methods, synthetic_methods, scopes, validations)
         end
       end
     end

data/lib/tng/analyzers/service.rb

@@ -47,7 +47,7 @@ module Tng
           end
 
           # Fallback to const_source_location if no method source found
-          service_file ||= service_class.const_source_location(service_class.name.split("::").last)&.first
+          service_file ||= Object.const_source_location(service_class.name)&.first
 
           service_methods = if service_file && File.exist?(service_file)
                               source_code = File.read(service_file)

data/lib/tng/services/test_generator.rb

@@ -87,13 +87,13 @@ module Tng
 
         case type
         when :controller
-          Tng.send_request_for_controller(name, file_object[:path], method_info[:name], *config)
+          Tng.send_request_for_controller(name, file_object[:path], method_info, *config)
         when :model
-          Tng.send_request_for_model(name, file_object[:path], method_info[:name], *config)
+          Tng.send_request_for_model(name, file_object[:path], method_info, *config)
         when :service
-          Tng.send_request_for_service(name, file_object[:path], method_info[:name], *config)
+          Tng.send_request_for_service(name, file_object[:path], method_info, *config)
         when :other
-          Tng.send_request_for_other(name, file_object[:path], method_info[:name], *config)
+          Tng.send_request_for_other(name, file_object[:path], method_info, *config)
         end
       end
 

data/lib/tng/services/user_app_config.rb

@@ -34,6 +34,18 @@ 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
 
@@ -71,6 +83,19 @@ module Tng
       rescue StandardError
         nil
       end
+
+      def self.read_test_file_content(relative_path)
+        return nil unless relative_path
+
+        rails_root = defined?(Rails) && Rails.root ? Rails.root.to_s : Dir.pwd
+        full_path = File.join(rails_root, relative_path)
+
+        return nil unless File.exist?(full_path) && File.readable?(full_path)
+
+        File.read(full_path)
+      rescue StandardError
+        nil
+      end
     end
   end
 end

data/lib/tng/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Tng
-  VERSION = "0.2.7"
+  VERSION = "0.2.9"
 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.7
+  version: 0.2.9
 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.7 ┘\n"
+  v0.2.9 ┘\n"
 rdoc_options: []
 require_paths:
 - lib