Skip to content

Commit

Permalink
Update test
Browse files Browse the repository at this point in the history
  • Loading branch information
VaniHaripriya committed Jan 7, 2025
1 parent 5e091a2 commit eadba71
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/pipeline_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -80,7 +80,7 @@ func (suite *IntegrationTestSuite) TestAPIServerDeployment() {
expectedDisplayName := "[Instructlab] Multi-Phase Training Pipeline"

// Retrieve pipelines
pipelines, err := RetrievePipelines(t, httpClient, apiServerURL)
pipelines, err := RetrievePipelines(t, suite.Clientmgr.httpClient, APIServerURL)
require.NoError(t, err, "Failed to retrieve pipelines")

found := false
Expand All @@ -90,7 +90,7 @@ func (suite *IntegrationTestSuite) TestAPIServerDeployment() {
break
}
}
if suite.DSPA.Spec.APIServer.EnableInstructlabPipeline {
if suite.DSPA.Spec.APIServer.EnableInstructLabPipeline {
expectedCount := 4
assert.Equal(t, expectedCount, len(pipelines.Pipelines), "Pipeline count should match when Instructlab is enabled")
assert.True(t, found, "Instructlab pipeline should exist when the flag is enabled")
Expand Down

0 comments on commit eadba71

Please sign in to comment.