44 ClassifierSdk ,
55 type ClassifyImageInput ,
66 ImageFormat ,
7+ parseAudience ,
78} from '../../src/index.js' ;
89import fs from 'fs' ;
910import { randomUUID } from 'crypto' ;
@@ -13,16 +14,18 @@ describe('ClassifierSdk Functional Tests', () => {
1314 it ( 'should listDeployments and return responses (smoke test)' , async ( {
1415 expect,
1516 } ) => {
16- // This is a smoke test. You must have a running gRPC server at localhost:50051 for this to pass.
17+ // This is a smoke test. The gRPC server address is taken from VITE_ATHENA_GRPC_ADDRESS,
18+ // falling back to the SDK default if the variable is not set.
1719 // You may want to mock the gRPC client for true unit testing.
1820 const sdk = new ClassifierSdk ( {
1921 deploymentId : process . env . VITE_ATHENA_DEPLOYMENT_ID ,
2022 affiliate : process . env . VITE_ATHENA_AFFILIATE ,
23+ grpcAddress : process . env . VITE_ATHENA_GRPC_ADDRESS ,
2124 authentication : {
2225 issuerUrl : process . env . VITE_OAUTH_ISSUER ,
2326 clientId : process . env . VITE_ATHENA_CLIENT_ID ,
2427 clientSecret : process . env . VITE_ATHENA_CLIENT_SECRET ,
25- scope : 'manage:classify' ,
28+ audience : parseAudience ( process . env . VITE_ATHENA_AUDIENCE ) ,
2629 } ,
2730 } ) ;
2831
@@ -46,11 +49,12 @@ describe('ClassifierSdk Functional Tests', () => {
4649 const sdk = new ClassifierSdk ( {
4750 deploymentId : process . env . VITE_ATHENA_DEPLOYMENT_ID ,
4851 affiliate : process . env . VITE_ATHENA_AFFILIATE ,
52+ grpcAddress : process . env . VITE_ATHENA_GRPC_ADDRESS ,
4953 authentication : {
5054 issuerUrl : process . env . VITE_OAUTH_ISSUER ,
5155 clientId : process . env . VITE_ATHENA_CLIENT_ID ,
5256 clientSecret : process . env . VITE_ATHENA_CLIENT_SECRET ,
53- scope : 'manage:classify' ,
57+ audience : parseAudience ( process . env . VITE_ATHENA_AUDIENCE ) ,
5458 } ,
5559 } ) ;
5660
@@ -60,8 +64,8 @@ describe('ClassifierSdk Functional Tests', () => {
6064 } ;
6165
6266 const response = await sdk . classifySingle ( input ) ;
63- expect ( response . classifications ) . toBe ( true ) ;
64- expect ( response . error ) . toBeNull ( ) ;
67+ expect ( Array . isArray ( response . classifications ) ) . toBe ( true ) ;
68+ expect ( response . error ) . toBeUndefined ( ) ;
6569 } , 10000 ) ;
6670 } ) ;
6771
@@ -70,17 +74,19 @@ describe('ClassifierSdk Functional Tests', () => {
7074 expect,
7175 annotate,
7276 } ) => {
73- // This is a smoke test. You must have a running gRPC server at localhost:50051 for this to pass.
77+ // This is a smoke test. The gRPC server address is taken from VITE_ATHENA_GRPC_ADDRESS,
78+ // falling back to the SDK default if the variable is not set.
7479 // You may want to mock the gRPC client for true unit testing.
7580 const imagePath = __dirname + '/448x448.jpg' ;
7681 const sdk = new ClassifierSdk ( {
7782 deploymentId : process . env . VITE_ATHENA_DEPLOYMENT_ID ,
7883 affiliate : process . env . VITE_ATHENA_AFFILIATE ,
84+ grpcAddress : process . env . VITE_ATHENA_GRPC_ADDRESS ,
7985 authentication : {
8086 issuerUrl : process . env . VITE_OAUTH_ISSUER ,
8187 clientId : process . env . VITE_ATHENA_CLIENT_ID ,
8288 clientSecret : process . env . VITE_ATHENA_CLIENT_SECRET ,
83- scope : 'manage:classify' ,
89+ audience : parseAudience ( process . env . VITE_ATHENA_AUDIENCE ) ,
8490 } ,
8591 } ) ;
8692
@@ -172,11 +178,12 @@ describe('ClassifierSdk Functional Tests', () => {
172178 const sdk = new ClassifierSdk ( {
173179 deploymentId : process . env . VITE_ATHENA_DEPLOYMENT_ID ,
174180 affiliate : process . env . VITE_ATHENA_AFFILIATE ,
181+ grpcAddress : process . env . VITE_ATHENA_GRPC_ADDRESS ,
175182 authentication : {
176183 issuerUrl : process . env . VITE_OAUTH_ISSUER ,
177184 clientId : process . env . VITE_ATHENA_CLIENT_ID ,
178185 clientSecret : process . env . VITE_ATHENA_CLIENT_SECRET ,
179- scope : 'manage:classify' ,
186+ audience : parseAudience ( process . env . VITE_ATHENA_AUDIENCE ) ,
180187 } ,
181188 } ) ;
182189
@@ -251,17 +258,19 @@ describe('ClassifierSdk Functional Tests', () => {
251258 expect,
252259 annotate,
253260 } ) => {
254- // This is a smoke test. You must have a running gRPC server at localhost:50051 for this to pass.
261+ // This is a smoke test. The gRPC server address is taken from VITE_ATHENA_GRPC_ADDRESS,
262+ // falling back to the SDK default if the variable is not set.
255263 // You may want to mock the gRPC client for true unit testing.
256264 const imagePath = __dirname + '/448x448.jpg' ;
257265 const sdk = new ClassifierSdk ( {
258266 deploymentId : process . env . VITE_ATHENA_DEPLOYMENT_ID ,
259267 affiliate : process . env . VITE_ATHENA_AFFILIATE ,
268+ grpcAddress : process . env . VITE_ATHENA_GRPC_ADDRESS ,
260269 authentication : {
261270 issuerUrl : process . env . VITE_OAUTH_ISSUER ,
262271 clientId : process . env . VITE_ATHENA_CLIENT_ID ,
263272 clientSecret : process . env . VITE_ATHENA_CLIENT_SECRET ,
264- scope : 'manage:classify' ,
273+ audience : parseAudience ( process . env . VITE_ATHENA_AUDIENCE ) ,
265274 } ,
266275 } ) ;
267276
0 commit comments