spec_guardian 0.1.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 48673c89d4e9875932318e835524bf63b4dc03eb779dc1f6a603b3cca2ac2eef
+  data.tar.gz: 78200eb6a421aec33340e58ae22674068b5ab7d05aaac03f5d2b83f5eb176af1
+SHA512:
+  metadata.gz: 6ab6111033bbfe79842df12c1a526ea93785080e627d2bfda161e295b9e24b7d618899bdbdbfbb273b05f47f4efa975e038affa83e1002d276a442126b660c12
+  data.tar.gz: af439dc047b822ecf3ce303114ed7405923eda8bfa4e6b9ccf8b3b07af5c0cf7a42ae0fd718f84583c8f0efb285221647bb35850a06c0e059598acd46057b26a

data/.gitignore

@@ -0,0 +1,13 @@
+/.bundle/
+/.yardoc
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+/vendor/bundle/
+spec_guardian-*.gem
+
+# rspec failure tracking
+.rspec_status

data/.prettierrc

@@ -0,0 +1,27 @@
+{
+  "tabWidth": 2,
+  "useTabs": false,
+  "semi": true,
+  "singleQuote": true,
+  "jsxSingleQuote": false,
+  "trailingComma": "all",
+  "bracketSpacing": true,
+  "bracketSameLine": false,
+  "arrowParens": "avoid",
+  "printWidth": 120,
+  "rubyPlugins": "plugin/trailing_comma",
+  "overrides": [
+    {
+      "files": ["babel.config.js"],
+      "options": {
+        "trailingComma": "es5"
+      }
+    },
+    {
+      "files": ["*.css", "*.scss"],
+      "options": {
+        "singleQuote": false
+      }
+    }
+  ]
+}

data/.rspec

@@ -0,0 +1,3 @@
+--format documentation
+--color
+--require spec_helper

data/.rubocop.yml

@@ -0,0 +1,21 @@
+# Overwrite or add rules to create your own house style
+#
+# # Use `[a, [b, c]]` not `[ a, [ b, c ] ]`
+# Layout/SpaceInsideArrayLiteralBrackets:
+#   Enabled: false
+Style/StringLiterals:
+  Enabled: true
+  EnforcedStyle: single_quotes
+Style/TrailingCommaInHashLiteral:
+  Enabled: true
+  EnforcedStyle: no_comma
+Style/EmptyMethod:
+  Enabled: true
+  EnforcedStyle: expanded
+Style/Documentation:
+  Enabled: false
+Style/FrozenStringLiteralComment:
+  Enabled: false
+Metrics/MethodLength:
+  Enabled: true
+  Max: 25

data/.ruby-version

@@ -0,0 +1 @@
+3.3.6

data/CHANGELOG.md

@@ -0,0 +1,5 @@
+## [Unreleased]
+
+## [0.1.0] - 2025-02-27
+
+- Initial release

data/Gemfile

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+source 'https://rubygems.org'
+
+ruby '3.3.6'
+
+# Specify your gem's dependencies in spec_guardian.gemspec
+gemspec

data/Gemfile.lock

@@ -0,0 +1,305 @@
+PATH
+  remote: .
+  specs:
+    spec_guardian (0.1.2)
+      anthropic (>= 0.3.2)
+      rails (>= 6.0)
+      rake (>= 12.0)
+      thor (>= 1.2)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    actioncable (8.0.1)
+      actionpack (= 8.0.1)
+      activesupport (= 8.0.1)
+      nio4r (~> 2.0)
+      websocket-driver (>= 0.6.1)
+      zeitwerk (~> 2.6)
+    actionmailbox (8.0.1)
+      actionpack (= 8.0.1)
+      activejob (= 8.0.1)
+      activerecord (= 8.0.1)
+      activestorage (= 8.0.1)
+      activesupport (= 8.0.1)
+      mail (>= 2.8.0)
+    actionmailer (8.0.1)
+      actionpack (= 8.0.1)
+      actionview (= 8.0.1)
+      activejob (= 8.0.1)
+      activesupport (= 8.0.1)
+      mail (>= 2.8.0)
+      rails-dom-testing (~> 2.2)
+    actionpack (8.0.1)
+      actionview (= 8.0.1)
+      activesupport (= 8.0.1)
+      nokogiri (>= 1.8.5)
+      rack (>= 2.2.4)
+      rack-session (>= 1.0.1)
+      rack-test (>= 0.6.3)
+      rails-dom-testing (~> 2.2)
+      rails-html-sanitizer (~> 1.6)
+      useragent (~> 0.16)
+    actiontext (8.0.1)
+      actionpack (= 8.0.1)
+      activerecord (= 8.0.1)
+      activestorage (= 8.0.1)
+      activesupport (= 8.0.1)
+      globalid (>= 0.6.0)
+      nokogiri (>= 1.8.5)
+    actionview (8.0.1)
+      activesupport (= 8.0.1)
+      builder (~> 3.1)
+      erubi (~> 1.11)
+      rails-dom-testing (~> 2.2)
+      rails-html-sanitizer (~> 1.6)
+    activejob (8.0.1)
+      activesupport (= 8.0.1)
+      globalid (>= 0.3.6)
+    activemodel (8.0.1)
+      activesupport (= 8.0.1)
+    activerecord (8.0.1)
+      activemodel (= 8.0.1)
+      activesupport (= 8.0.1)
+      timeout (>= 0.4.0)
+    activestorage (8.0.1)
+      actionpack (= 8.0.1)
+      activejob (= 8.0.1)
+      activerecord (= 8.0.1)
+      activesupport (= 8.0.1)
+      marcel (~> 1.0)
+    activesupport (8.0.1)
+      base64
+      benchmark (>= 0.3)
+      bigdecimal
+      concurrent-ruby (~> 1.0, >= 1.3.1)
+      connection_pool (>= 2.2.5)
+      drb
+      i18n (>= 1.6, < 2)
+      logger (>= 1.4.2)
+      minitest (>= 5.1)
+      securerandom (>= 0.3)
+      tzinfo (~> 2.0, >= 2.0.5)
+      uri (>= 0.13.1)
+    anthropic (0.3.2)
+      event_stream_parser (>= 0.3.0, < 2.0.0)
+      faraday (>= 1)
+      faraday-multipart (>= 1)
+    ast (2.4.2)
+    base64 (0.2.0)
+    benchmark (0.4.0)
+    bigdecimal (3.1.9)
+    builder (3.3.0)
+    coderay (1.1.3)
+    concurrent-ruby (1.3.5)
+    connection_pool (2.5.0)
+    crass (1.0.6)
+    date (3.4.1)
+    diff-lcs (1.6.0)
+    drb (2.2.1)
+    erubi (1.13.1)
+    event_stream_parser (1.0.0)
+    faraday (2.12.2)
+      faraday-net_http (>= 2.0, < 3.5)
+      json
+      logger
+    faraday-multipart (1.1.0)
+      multipart-post (~> 2.0)
+    faraday-net_http (3.4.0)
+      net-http (>= 0.5.0)
+    globalid (1.2.1)
+      activesupport (>= 6.1)
+    i18n (1.14.7)
+      concurrent-ruby (~> 1.0)
+    io-console (0.8.0)
+    irb (1.15.1)
+      pp (>= 0.6.0)
+      rdoc (>= 4.0.0)
+      reline (>= 0.4.2)
+    json (2.10.1)
+    language_server-protocol (3.17.0.4)
+    lint_roller (1.1.0)
+    logger (1.6.6)
+    loofah (2.24.0)
+      crass (~> 1.0.2)
+      nokogiri (>= 1.12.0)
+    mail (2.8.1)
+      mini_mime (>= 0.1.1)
+      net-imap
+      net-pop
+      net-smtp
+    marcel (1.0.4)
+    method_source (1.1.0)
+    mini_mime (1.1.5)
+    minitest (5.25.4)
+    multipart-post (2.4.1)
+    net-http (0.6.0)
+      uri
+    net-imap (0.5.6)
+      date
+      net-protocol
+    net-pop (0.1.2)
+      net-protocol
+    net-protocol (0.2.2)
+      timeout
+    net-smtp (0.5.1)
+      net-protocol
+    nio4r (2.7.4)
+    nokogiri (1.18.3-aarch64-linux-gnu)
+      racc (~> 1.4)
+    nokogiri (1.18.3-aarch64-linux-musl)
+      racc (~> 1.4)
+    nokogiri (1.18.3-arm-linux-gnu)
+      racc (~> 1.4)
+    nokogiri (1.18.3-arm-linux-musl)
+      racc (~> 1.4)
+    nokogiri (1.18.3-arm64-darwin)
+      racc (~> 1.4)
+    nokogiri (1.18.3-x86_64-darwin)
+      racc (~> 1.4)
+    nokogiri (1.18.3-x86_64-linux-gnu)
+      racc (~> 1.4)
+    nokogiri (1.18.3-x86_64-linux-musl)
+      racc (~> 1.4)
+    parallel (1.26.3)
+    parser (3.3.7.1)
+      ast (~> 2.4.1)
+      racc
+    pp (0.6.2)
+      prettyprint
+    prettyprint (0.2.0)
+    pry (0.15.2)
+      coderay (~> 1.1)
+      method_source (~> 1.0)
+    psych (5.2.3)
+      date
+      stringio
+    racc (1.8.1)
+    rack (3.1.10)
+    rack-session (2.1.0)
+      base64 (>= 0.1.0)
+      rack (>= 3.0.0)
+    rack-test (2.2.0)
+      rack (>= 1.3)
+    rackup (2.2.1)
+      rack (>= 3)
+    rails (8.0.1)
+      actioncable (= 8.0.1)
+      actionmailbox (= 8.0.1)
+      actionmailer (= 8.0.1)
+      actionpack (= 8.0.1)
+      actiontext (= 8.0.1)
+      actionview (= 8.0.1)
+      activejob (= 8.0.1)
+      activemodel (= 8.0.1)
+      activerecord (= 8.0.1)
+      activestorage (= 8.0.1)
+      activesupport (= 8.0.1)
+      bundler (>= 1.15.0)
+      railties (= 8.0.1)
+    rails-dom-testing (2.2.0)
+      activesupport (>= 5.0.0)
+      minitest
+      nokogiri (>= 1.6)
+    rails-html-sanitizer (1.6.2)
+      loofah (~> 2.21)
+      nokogiri (>= 1.15.7, != 1.16.7, != 1.16.6, != 1.16.5, != 1.16.4, != 1.16.3, != 1.16.2, != 1.16.1, != 1.16.0.rc1, != 1.16.0)
+    railties (8.0.1)
+      actionpack (= 8.0.1)
+      activesupport (= 8.0.1)
+      irb (~> 1.13)
+      rackup (>= 1.0.0)
+      rake (>= 12.2)
+      thor (~> 1.0, >= 1.2.2)
+      zeitwerk (~> 2.6)
+    rainbow (3.1.1)
+    rake (13.2.1)
+    rdoc (6.12.0)
+      psych (>= 4.0.0)
+    regexp_parser (2.10.0)
+    reline (0.6.0)
+      io-console (~> 0.5)
+    rspec (3.13.0)
+      rspec-core (~> 3.13.0)
+      rspec-expectations (~> 3.13.0)
+      rspec-mocks (~> 3.13.0)
+    rspec-core (3.13.3)
+      rspec-support (~> 3.13.0)
+    rspec-expectations (3.13.3)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.13.0)
+    rspec-mocks (3.13.2)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.13.0)
+    rspec-support (3.13.2)
+    rubocop (1.73.1)
+      json (~> 2.3)
+      language_server-protocol (~> 3.17.0.2)
+      lint_roller (~> 1.1.0)
+      parallel (~> 1.10)
+      parser (>= 3.3.0.2)
+      rainbow (>= 2.2.2, < 4.0)
+      regexp_parser (>= 2.9.3, < 3.0)
+      rubocop-ast (>= 1.38.0, < 2.0)
+      ruby-progressbar (~> 1.7)
+      unicode-display_width (>= 2.4.0, < 4.0)
+    rubocop-ast (1.38.1)
+      parser (>= 3.3.1.0)
+    rubocop-packaging (0.5.2)
+      rubocop (>= 1.33, < 2.0)
+    rubocop-performance (1.24.0)
+      lint_roller (~> 1.1)
+      rubocop (>= 1.72.1, < 2.0)
+      rubocop-ast (>= 1.38.0, < 2.0)
+    rubocop-rspec (3.5.0)
+      lint_roller (~> 1.1)
+      rubocop (~> 1.72, >= 1.72.1)
+    rubocop-shopify (2.15.1)
+      rubocop (~> 1.51)
+    rubocop-thread_safety (0.7.0)
+      lint_roller (~> 1.1)
+      rubocop (~> 1.72, >= 1.72.1)
+    ruby-progressbar (1.13.0)
+    securerandom (0.4.1)
+    stringio (3.1.5)
+    thor (1.3.2)
+    timeout (0.4.3)
+    tzinfo (2.0.6)
+      concurrent-ruby (~> 1.0)
+    unicode-display_width (3.1.4)
+      unicode-emoji (~> 4.0, >= 4.0.4)
+    unicode-emoji (4.0.4)
+    uri (1.0.3)
+    useragent (0.16.11)
+    websocket-driver (0.7.7)
+      base64
+      websocket-extensions (>= 0.1.0)
+    websocket-extensions (0.1.5)
+    zeitwerk (2.7.2)
+
+PLATFORMS
+  aarch64-linux-gnu
+  aarch64-linux-musl
+  arm-linux-gnu
+  arm-linux-musl
+  arm64-darwin
+  x86_64-darwin
+  x86_64-linux-gnu
+  x86_64-linux-musl
+
+DEPENDENCIES
+  pry (~> 0.15.2)
+  rspec (~> 3.0)
+  rubocop (~> 1.73, >= 1.73.1)
+  rubocop-packaging (~> 0.5.2)
+  rubocop-performance (~> 1.24)
+  rubocop-rspec (~> 3.5)
+  rubocop-shopify (~> 2.15, >= 2.15.1)
+  rubocop-thread_safety (~> 0.7.0)
+  spec_guardian!
+
+RUBY VERSION
+   ruby 3.3.6p108
+
+BUNDLED WITH
+   2.6.3

data/LICENSE.md

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Darren Terhune
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

data/README.md

@@ -0,0 +1,84 @@
+# SpecGuardian
+
+A Ruby gem that uses AI (Claude) to automatically generate test files for your Rails application code.
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'spec_guardian'
+```
+
+And then execute:
+
+```ruby
+bundle install
+```
+
+Or install it yourself as:
+
+```ruby
+gem install spec_guardian
+```
+
+## Configuration
+
+Create an initializer file at `config/initializers/spec_guardian.rb` by running the install command to create this file `rails g spec_guardian:install`
+
+```ruby
+SpecGuardian.configure do |config|
+  config.api_key = ENV['ANTHROPIC_API_KEY'] # Your Anthropic API key
+  config.ai_model = "claude-3-7-sonnet-20250219" # The Claude model to use
+  config.max_tokens = 20_000 # The max tokens to use per request defaults to 20_000
+  config.test_style = "thorough" # Options: minimal, thorough, edge_cases
+end
+```
+
+NOTE: The `max_tokens` determines the length of the response. If you are seeing files cut off, then you need to increase this value.
+
+Ensure you have your Anthropic API key set as an environment variable or directly in the configuration. You can signup here https://console.anthropic.com to generate an account and get an API key.
+
+## Usage
+
+### Command Line
+
+You can use the executable to generate a test for any file in your Rails application:
+
+```ruby
+bundle exec spec_guardian app/models/user.rb
+```
+
+During testing, models files with around 200-300 lines of code cost around $0.02. This will vary of course depending on how complex your code is. If you have rubocop installed you may need to adjust specs. It also isn't 100% accurate with generating perfect code so expect to look over the files it generates and be ready to fix things.
+
+### Rake Task
+
+Alternatively, you can use the provided Rake task:
+
+```ruby
+bundle exec rake spec_guardian:generate[app/models/user.rb]
+```
+
+If you are using zsh you may need to quote your input:
+
+```ruby
+bundle exec rake spec_guardian:generate['app/models/user.rb']
+```
+
+## Features
+
+1. Automatically detects whether your application uses RSpec or Minitest
+2. Places generated test files in the appropriate location (spec/ or test/ directory)
+3. Customizes test generation based on file type (model, controller, etc.)
+4. Supports different test styles (minimal, thorough, or edge_cases)
+
+## How It Works
+
+1. The gem detects your project's test framework (RSpec, Minitest, or Rails default)
+2. It analyzes the source file to determine its type (model, controller, etc.)
+3. It sends the source code to Claude's API with a specialized prompt
+4. It creates the test file in the correct location with the AI-generated test code
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/darrenterhune/spec_guardian.

data/Rakefile

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+require 'bundler/gem_tasks'
+require 'rspec/core/rake_task'
+
+RSpec::Core::RakeTask.new(:spec)
+
+require 'rubocop/rake_task'
+
+RuboCop::RakeTask.new
+
+task default: %i[spec rubocop]

data/bin/console

@@ -0,0 +1,11 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require 'bundler/setup'
+require 'spec_guardian'
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+require 'irb'
+IRB.start(__FILE__)

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/bin/spec_guardian

@@ -0,0 +1,25 @@
+#!/usr/bin/env ruby
+
+require 'bundler/setup'
+require 'spec_guardian'
+
+# Parse command line arguments
+if ARGV.empty? || ARGV[0] == '--help' || ARGV[0] == '-h'
+  puts 'Usage: spec_guardian PATH_TO_FILE'
+  puts 'Generates a test file for the given source file using AI.'
+  exit 0
+end
+
+begin
+  # Initialize Rails environment
+  require File.expand_path('config/environment', Dir.pwd)
+
+  # Generate the test
+  SpecGuardian.generate_test(ARGV[0])
+rescue LoadError => e
+  puts 'Error: This command must be run from a Rails application root directory.'
+  exit 1
+rescue SpecGuardian::Error => e
+  puts "Error: #{e.message}"
+  exit 1
+end

data/lib/generators/spec_guardian/install_generator.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module SpecGuardian
+  module Generators
+    class InstallGenerator < Rails::Generators::Base
+      source_root File.expand_path('templates', __dir__)
+
+      def copy_config
+        template('config/initializers/spec_guardian.rb')
+      end
+    end
+  end
+end

data/lib/generators/spec_guardian/templates/config/initializers/spec_guardian.rb

@@ -0,0 +1,7 @@
+# config/initializers/spec_guardian.rb
+SpecGuardian.configure do |config|
+  config.api_key = ENV['ANTHROPIC_API_KEY'] # Your Anthropic API key
+  config.ai_model = 'claude-3-7-sonnet-20250219' # The Claude model to use
+  config.max_tokens = 20_000 # The max tokens to use per request defaults to 20_000
+  config.test_style = 'thorough' # Options: minimal, thorough, edge_cases
+end

data/lib/spec_guardian/ai_client.rb

@@ -0,0 +1,74 @@
+require 'anthropic'
+
+module SpecGuardian
+  class AiClient
+    def self.generate_test(source_code, framework, file_path)
+      client = Anthropic::Client.new(
+        access_token: SpecGuardian.configuration.api_key
+      )
+
+      # File type detection
+      file_type = detect_file_type(file_path)
+
+      # Create the prompt
+      prompt = create_prompt(source_code, framework, file_type)
+      model = SpecGuardian.configuration.ai_model
+      max_tokens = SpecGuardian.configuration.max_tokens || 20_000
+
+      # Make API call
+      response = client.messages(
+        parameters: {
+          model: model,
+          max_tokens: max_tokens,
+          messages: [
+            {
+              role: 'user',
+              content: prompt
+            }
+          ]
+        }
+      )
+
+      # Extract and return the test code
+      response['content'][0]['text']
+    end
+
+    def self.detect_file_type(file_path)
+      if file_path.include?('/models/')
+        :model
+      elsif file_path.include?('/controllers/')
+        :controller
+      elsif file_path.include?('/views/')
+        :view
+      elsif file_path.include?('/helpers/')
+        :helper
+      elsif file_path.include?('/mailers/')
+        :mailer
+      elsif file_path.include?('/jobs/')
+        :job
+      else
+        :other
+      end
+    end
+
+    def self.create_prompt(source_code, framework, file_type)
+      framework_name = framework == :rspec ? 'RSpec' : 'Minitest'
+      style = SpecGuardian.configuration.test_style
+
+      <<~PROMPT
+        Generate a comprehensive Rails #{framework_name} test for the following #{file_type} code.
+        The test should follow best practices for Rails testing with #{framework_name}.
+
+        Test style preference: #{style}
+
+        Here's the source code to test:
+
+        ```ruby
+        #{source_code}
+        ```
+
+        Please generate only the test code without any explanations. The output should be valid Ruby that can be saved directly to a test file. Remove comments. Do not include code formatting markdown like '```ruby'. Do not test rails internals like associations, scopes or table columns.
+      PROMPT
+    end
+  end
+end

data/lib/spec_guardian/configuration.rb

@@ -0,0 +1,12 @@
+module SpecGuardian
+  class Configuration
+    attr_accessor :api_key, :ai_model, :test_style, :max_tokens
+
+    def initialize
+      @api_key = ENV['ANTHROPIC_API_KEY']
+      @ai_model = 'claude-3-7-sonnet-20250219'
+      @max_tokens = 20_000
+      @test_style = 'thorough' # Options: minimal, thorough, edge_cases
+    end
+  end
+end

data/lib/spec_guardian/file_path_resolver.rb

@@ -0,0 +1,33 @@
+require 'fileutils'
+
+module SpecGuardian
+  class FilePathResolver
+    def self.resolve(file_path, framework)
+      # Remove leading ./ if present
+      file_path = file_path.sub(%r{^\./}, '')
+
+      # Extract the relative path from app/
+      relative_path = if file_path.start_with?('app/')
+                        file_path.sub(%r{^app/}, '')
+                      else
+                        file_path
+                      end
+
+      # Determine test directory based on framework
+      test_dir = framework == :rspec ? 'spec' : 'test'
+
+      # Handle different file types
+      if relative_path.start_with?('models/')
+        "#{test_dir}/#{relative_path.sub(/\.rb$/, '_spec.rb')}"
+      elsif relative_path.start_with?('controllers/')
+        "#{test_dir}/#{relative_path.sub(/\.rb$/, '_spec.rb')}"
+      elsif relative_path.start_with?('views/')
+        view_path = relative_path.sub(%r{^views/}, '')
+        "#{test_dir}/views/#{view_path.sub(/\.\w+\.erb$/, '.html_spec.rb')}"
+      else
+        # For other files, maintain the same structure
+        "#{test_dir}/#{relative_path.sub(/\.rb$/, '_spec.rb')}"
+      end
+    end
+  end
+end

data/lib/spec_guardian/railtie.rb

@@ -0,0 +1,9 @@
+require 'rails'
+
+module SpecGuardian
+  class Railtie < Rails::Railtie
+    rake_tasks do
+      load 'tasks/generate_tests.rake'
+    end
+  end
+end

data/lib/spec_guardian/test_framework_detector.rb

@@ -0,0 +1,21 @@
+module SpecGuardian
+  class TestFrameworkDetector
+    def self.detect
+      return :rspec if File.exist?('spec') && gemfile_includes?('rspec')
+
+      return :minitest if File.exist?('test') && gemfile_includes?('minitest')
+
+      if Rails.version.to_i >= 5
+        :minitest
+      else
+        :test_unit
+      end
+    end
+
+    def self.gemfile_includes?(gem_name)
+      return false unless File.exist?('Gemfile')
+
+      File.read('Gemfile').include?(gem_name)
+    end
+  end
+end

data/lib/spec_guardian/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module SpecGuardian
+  VERSION = '0.1.2'
+end

data/lib/spec_guardian.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+require_relative 'spec_guardian/version'
+require_relative 'spec_guardian/configuration'
+require_relative 'spec_guardian/test_framework_detector'
+require_relative 'spec_guardian/file_path_resolver'
+require_relative 'spec_guardian/ai_client'
+
+module SpecGuardian
+  class Error < StandardError; end
+
+  class << self
+    attr_accessor :configuration
+
+    def configure
+      self.configuration ||= Configuration.new
+      yield(configuration) if block_given?
+    end
+
+    def generate_test(file_path)
+      # Validate the file exists
+      raise Error, "File not found: #{file_path}" unless File.exist?(file_path)
+
+      # Detect the test framework
+      framework = TestFrameworkDetector.detect
+
+      # Resolve the test file path
+      test_file_path = FilePathResolver.resolve(file_path, framework)
+
+      # Generate the test content
+      source_code = File.read(file_path)
+      test_content = AiClient.generate_test(source_code, framework, file_path)
+
+      # Create the test file
+      FileUtils.mkdir_p(File.dirname(test_file_path))
+      File.write(test_file_path, test_content)
+
+      puts "Test file generated: #{test_file_path}"
+      test_file_path
+    end
+  end
+end
+
+require 'spec_guardian/railtie' if defined?(Rails)

data/lib/tasks/generate_tests.rake

@@ -0,0 +1,13 @@
+namespace :spec_guardian do
+  desc 'Generate a test file for a given source file using AI'
+  task :generate, [:file_path] => :environment do |t, args|
+    if args[:file_path].nil?
+      puts 'Error: Please provide a file path.'
+      puts 'Usage: rake spec_guardian:generate[app/models/user.rb]'
+      puts "You may need to use quotes if using zsh e.g. ['app/models/user.rb']"
+      exit 1
+    end
+
+    SpecGuardian.generate_test(args[:file_path])
+  end
+end

data/sig/spec_guardian.rbs

@@ -0,0 +1,4 @@
+module SpecGuardian
+  VERSION: String
+  # See the writing guide of rbs: https://github.com/ruby/rbs#guides
+end

data/spec_guardian.gemspec

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+require_relative 'lib/spec_guardian/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'spec_guardian'
+  spec.version       = SpecGuardian::VERSION
+  spec.platform      = Gem::Platform::RUBY
+  spec.authors       = ['Darren Terhune']
+  spec.email         = ['darrenterhune@gmail.com']
+
+  spec.summary       = 'Generate test files for Rails apps using AI'
+  spec.description   = 'A gem that uses AI to automatically generate test files for your Rails application code'
+  spec.homepage      = 'https://github.com/darrenterhune/spec_guardian'
+  spec.license       = 'MIT'
+  spec.required_ruby_version = Gem::Requirement.new('>= 3.1.6')
+
+  spec.metadata = {
+    'bug_tracker_uri' => 'https://github.com/darrenterhune/spec_guardian/issues',
+    'changelog_uri' => "#{spec.homepage}/blob/main/CHANGELOG.md",
+    'homepage_uri' => 'https://github.com/darrenterhune/spec_guardian',
+    'source_code_uri' => 'https://github.com/darrenterhune/spec_guardian'
+  }
+
+  spec.files = Dir.chdir(File.expand_path(__dir__)) do
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  end
+  spec.bindir        = 'bin'
+  spec.executables   = ['spec_guardian']
+  spec.require_paths = ['lib']
+
+  spec.add_dependency 'anthropic', '>= 0.3.2'
+  spec.add_dependency 'rails', '>= 6.0'
+  spec.add_dependency 'rake', '>= 12.0'
+  spec.add_dependency 'thor', '>= 1.2'
+  spec.add_development_dependency 'pry', '~> 0.15.2'
+  spec.add_development_dependency 'rspec', '~> 3.0'
+  spec.add_development_dependency 'rubocop', '~> 1.73', '>= 1.73.1'
+  spec.add_development_dependency 'rubocop-packaging', '~> 0.5.2'
+  spec.add_development_dependency 'rubocop-performance', '~> 1.24'
+  spec.add_development_dependency 'rubocop-rspec', '~> 3.5'
+  spec.add_development_dependency 'rubocop-shopify', '~> 2.15', '>= 2.15.1'
+  spec.add_development_dependency 'rubocop-thread_safety', '~> 0.7.0'
+end

metadata

@@ -0,0 +1,256 @@
+--- !ruby/object:Gem::Specification
+name: spec_guardian
+version: !ruby/object:Gem::Version
+  version: 0.1.2
+platform: ruby
+authors:
+- Darren Terhune
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2025-03-25 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: anthropic
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.3.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.3.2
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '6.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '6.0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '12.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '12.0'
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.2'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.15.2
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.15.2
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.73'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.73.1
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.73'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.73.1
+- !ruby/object:Gem::Dependency
+  name: rubocop-packaging
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.5.2
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.5.2
+- !ruby/object:Gem::Dependency
+  name: rubocop-performance
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.24'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.24'
+- !ruby/object:Gem::Dependency
+  name: rubocop-rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.5'
+- !ruby/object:Gem::Dependency
+  name: rubocop-shopify
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.15'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.15.1
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.15'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.15.1
+- !ruby/object:Gem::Dependency
+  name: rubocop-thread_safety
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.7.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.7.0
+description: A gem that uses AI to automatically generate test files for your Rails
+  application code
+email:
+- darrenterhune@gmail.com
+executables:
+- spec_guardian
+extensions: []
+extra_rdoc_files: []
+files:
+- ".DS_Store"
+- ".gitignore"
+- ".prettierrc"
+- ".rspec"
+- ".rubocop.yml"
+- ".ruby-version"
+- CHANGELOG.md
+- Gemfile
+- Gemfile.lock
+- LICENSE.md
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- bin/spec_guardian
+- lib/generators/spec_guardian/install_generator.rb
+- lib/generators/spec_guardian/templates/config/initializers/spec_guardian.rb
+- lib/spec_guardian.rb
+- lib/spec_guardian/ai_client.rb
+- lib/spec_guardian/configuration.rb
+- lib/spec_guardian/file_path_resolver.rb
+- lib/spec_guardian/railtie.rb
+- lib/spec_guardian/test_framework_detector.rb
+- lib/spec_guardian/version.rb
+- lib/tasks/generate_tests.rake
+- sig/spec_guardian.rbs
+- spec_guardian.gemspec
+homepage: https://github.com/darrenterhune/spec_guardian
+licenses:
+- MIT
+metadata:
+  bug_tracker_uri: https://github.com/darrenterhune/spec_guardian/issues
+  changelog_uri: https://github.com/darrenterhune/spec_guardian/blob/main/CHANGELOG.md
+  homepage_uri: https://github.com/darrenterhune/spec_guardian
+  source_code_uri: https://github.com/darrenterhune/spec_guardian
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.1.6
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.5.22
+signing_key:
+specification_version: 4
+summary: Generate test files for Rails apps using AI
+test_files: []