Update endpoints (#50)

* Update endpoints

* Add Rspec

app/controllers/assertion_results_controller.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+class AssertionResultsController < ApplicationController
+  before_action :set_test_result
+
+  def index
+    @assertion_results = @test_result.assertion_results
+  end
+
+  private
+
+  def set_test_result
+    @test_result = current_user.test_results.find(params[:test_result_id])
+  end
+end

app/views/assertion_results/_assertion_result.json.jbuilder

@@ -0,0 +1,3 @@
+# frozen_string_literal: true
+
+json.extract! assertion_result, :id, :test_result_id, :assertion_id, :state, :created_at, :updated_at

app/views/assertion_results/index.json.jbuilder

@@ -0,0 +1,3 @@
+# frozen_string_literal: true
+
+json.array! @assertion_results, partial: 'assertion_results/assertion_result', as: :assertion_result

app/views/test_results/_test_result.json.jbuilder

@@ -1,4 +1,4 @@
 # frozen_string_literal: true
 
-json.extract! test_result, :id, :result, :content, :time, :rating, :status, :created_at, :updated_at
+json.extract! test_result, :id, :result, :time, :rating, :status, :created_at, :updated_at
 json.url test_result_url(test_result, format: :json)

app/views/test_results/show.json.jbuilder

@@ -0,0 +1,4 @@
+# frozen_string_literal: true
+
+json.partial! 'test_results/test_result', test_result: @test_result
+json.assertion_results @test_result.assertion_result_ids

app/views/test_runs/_test_run.json.jbuilder

@@ -1,4 +1,5 @@
 # frozen_string_literal: true
 
 json.extract! test_run, :id, :name, :created_at, :updated_at
+json.test_result_ids test_run.test_result_ids
 json.url prompt_url(test_run, format: :json)

bin/docker-entrypoint

@@ -14,6 +14,6 @@ generate_credentials() {
 generate_credentials
 
 ./bin/rails db:prepare
-./bin/rails data:migrate          
+./bin/rails data:migrate
 
 exec "${@}"

config/routes.rb

@@ -19,7 +19,9 @@
     end
   end
 
-  resources :test_results, only: %i[index update show]
+  resources :test_results, only: %i[index update show] do
+    resources :assertion_results, only: %i[index]
+  end
 
   resources :models do
     resources :model_versions

spec/controllers/assertion_results_controller_spec.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+require 'rails_helper'
+
+RSpec.describe AssertionResultsController, type: :controller do
+  let(:user) { create(:user) }
+  let(:prompt) { create(:prompt, user:) }
+  let(:test_run) { create(:test_run, prompt:) }
+  let(:model) { create(:model, user:) }
+  let(:model_version) { create(:model_version, model:) }
+  let(:test_model_version_run) { create(:test_model_version_run, test_run:, model_version:) }
+  let(:test_result) { create(:test_result, test_model_version_run:) }
+
+  before do
+    sign_in user
+  end
+
+  describe 'GET #index' do
+    it 'returns a success response' do
+      get :index, params: { test_result_id: test_result.id }, format: :json
+      expect(response).to be_successful
+    end
+
+    it 'sets the correct test_result' do
+      get :index, params: { test_result_id: test_result.id }, format: :json
+      expect(assigns(:test_result)).to eq(test_result)
+    end
+
+    it 'sets the correct assertion_results' do
+      get :index, params: { test_result_id: test_result.id }, format: :json
+      expect(assigns(:assertion_results)).to eq(test_result.assertion_results)
+    end
+  end
+end