Skip to content

Commit

Permalink
Add AI API tests
Browse files Browse the repository at this point in the history
  • Loading branch information
PasanT9 committed Dec 10, 2024
1 parent 7492bcb commit 6209c86
Show file tree
Hide file tree
Showing 12 changed files with 2,505 additions and 11 deletions.

Large diffs are not rendered by default.

Large diffs are not rendered by default.

Original file line number Diff line number Diff line change
@@ -0,0 +1,185 @@
/*
* WSO2 API Manager - Admin
* This document specifies a **RESTful API** for WSO2 **API Manager** - **Admin Portal**. Please see [full OpenAPI Specification](https://raw.githubusercontent.com/wso2/carbon-apimgt/v6.7.206/components/apimgt/org.wso2.carbon.apimgt.rest.api.admin.v1/src/main/resources/admin-api.yaml) of the API which is written using [OAS 3.0](http://swagger.io/) specification. # Authentication Our REST APIs are protected using OAuth2 and access control is achieved through scopes. Before you start invoking the the API you need to obtain an access token with the required scopes. This guide will walk you through the steps that you will need to follow to obtain an access token. First you need to obtain the consumer key/secret key pair by calling the dynamic client registration (DCR) endpoint. You can add your preferred grant types in the payload. A sample payload is shown below. ``` { \"callbackUrl\":\"www.google.lk\", \"clientName\":\"rest_api_admin\", \"owner\":\"admin\", \"grantType\":\"client_credentials password refresh_token\", \"saasApp\":true } ``` Create a file (payload.json) with the above sample payload, and use the cURL shown bellow to invoke the DCR endpoint. Authorization header of this should contain the base64 encoded admin username and password. **Format of the request** ``` curl -X POST -H \"Authorization: Basic Base64(admin_username:admin_password)\" -H \"Content-Type: application/json\" \\ -d @payload.json https://<host>:<servlet_port>/client-registration/v0.17/register ``` **Sample request** ``` curl -X POST -H \"Authorization: Basic YWRtaW46YWRtaW4=\" -H \"Content-Type: application/json\" \\ -d @payload.json https://localhost:9443/client-registration/v0.17/register ``` Following is a sample response after invoking the above curl. ``` { \"clientId\": \"fOCi4vNJ59PpHucC2CAYfYuADdMa\", \"clientName\": \"rest_api_admin\", \"callBackURL\": \"www.google.lk\", \"clientSecret\": \"a4FwHlq0iCIKVs2MPIIDnepZnYMa\", \"isSaasApplication\": true, \"appOwner\": \"admin\", \"jsonString\": \"{\\\"grant_types\\\":\\\"client_credentials password refresh_token\\\",\\\"redirect_uris\\\":\\\"www.google.lk\\\",\\\"client_name\\\":\\\"rest_api_admin\\\"}\", \"jsonAppAttribute\": \"{}\", \"tokenType\": null } ``` Next you must use the above client id and secret to obtain the access token. We will be using the password grant type for this, you can use any grant type you desire. You also need to add the proper **scope** when getting the access token. All possible scopes for Admin REST API can be viewed in **OAuth2 Security** section of this document and scope for each resource is given in **authorizations** section of resource documentation. Following is the format of the request if you are using the password grant type. ``` curl -k -d \"grant_type=password&username=<admin_username>&password=<admin_passowrd>&scope=<scopes seperated by space>\" \\ -H \"Authorization: Basic base64(cliet_id:client_secret)\" \\ https://<host>:<gateway_port>/token ``` **Sample request** ``` curl https://localhost:8243/token -k \\ -H \"Authorization: Basic Zk9DaTR2Tko1OVBwSHVjQzJDQVlmWXVBRGRNYTphNEZ3SGxxMGlDSUtWczJNUElJRG5lcFpuWU1h\" \\ -d \"grant_type=password&username=admin&password=admin&scope=apim:admin apim:tier_view\" ``` Shown below is a sample response to the above request. ``` { \"access_token\": \"e79bda48-3406-3178-acce-f6e4dbdcbb12\", \"refresh_token\": \"a757795d-e69f-38b8-bd85-9aded677a97c\", \"scope\": \"apim:admin apim:tier_view\", \"token_type\": \"Bearer\", \"expires_in\": 3600 } ``` Now you have a valid access token, which you can use to invoke an API. Navigate through the API descriptions to find the required API, obtain an access token as described above and invoke the API with the authentication header. If you use a different authentication mechanism, this process may change. # Try out in Postman If you want to try-out the embedded postman collection with \"Run in Postman\" option, please follow the guidelines listed below. * All of the OAuth2 secured endpoints have been configured with an Authorization Bearer header with a parameterized access token. Before invoking any REST API resource make sure you run the `Register DCR Application` and `Generate Access Token` requests to fetch an access token with all required scopes. * Make sure you have an API Manager instance up and running. * Update the `basepath` parameter to match the hostname and port of the APIM instance. [![Run in Postman](https://run.pstmn.io/button.svg)](https://app.getpostman.com/run-collection/f5ac2ca9fb22afef6ed6)
*
* The version of the OpenAPI document: v4
* Contact: [email protected]
*
* NOTE: This class is auto generated by OpenAPI Generator (https://openapi-generator.tech).
* https://openapi-generator.tech
* Do not edit the class manually.
*/


package org.wso2.am.integration.clients.admin.api.dto;

import java.util.Objects;
import java.util.Arrays;
import com.google.gson.TypeAdapter;
import com.google.gson.annotations.JsonAdapter;
import com.google.gson.annotations.SerializedName;
import com.google.gson.stream.JsonReader;
import com.google.gson.stream.JsonWriter;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import java.io.IOException;
import com.fasterxml.jackson.annotation.JsonCreator;
/**
* AIAPIQuotaLimitAllOfDTO
*/

public class AIAPIQuotaLimitAllOfDTO {
public static final String SERIALIZED_NAME_REQUEST_COUNT = "requestCount";
@SerializedName(SERIALIZED_NAME_REQUEST_COUNT)
private Long requestCount;

public static final String SERIALIZED_NAME_TOTAL_TOKEN_COUNT = "totalTokenCount";
@SerializedName(SERIALIZED_NAME_TOTAL_TOKEN_COUNT)
private Long totalTokenCount;

public static final String SERIALIZED_NAME_PROMPT_TOKEN_COUNT = "promptTokenCount";
@SerializedName(SERIALIZED_NAME_PROMPT_TOKEN_COUNT)
private Long promptTokenCount;

public static final String SERIALIZED_NAME_COMPLETION_TOKEN_COUNT = "completionTokenCount";
@SerializedName(SERIALIZED_NAME_COMPLETION_TOKEN_COUNT)
private Long completionTokenCount;


public AIAPIQuotaLimitAllOfDTO requestCount(Long requestCount) {

this.requestCount = requestCount;
return this;
}

/**
* Maximum number of requests allowed
* @return requestCount
**/
@ApiModelProperty(example = "300", required = true, value = "Maximum number of requests allowed")

public Long getRequestCount() {
return requestCount;
}


public void setRequestCount(Long requestCount) {
this.requestCount = requestCount;
}


public AIAPIQuotaLimitAllOfDTO totalTokenCount(Long totalTokenCount) {

this.totalTokenCount = totalTokenCount;
return this;
}

/**
* Maximum number of total tokens allowed
* @return totalTokenCount
**/
@javax.annotation.Nullable
@ApiModelProperty(example = "800", value = "Maximum number of total tokens allowed")

public Long getTotalTokenCount() {
return totalTokenCount;
}


public void setTotalTokenCount(Long totalTokenCount) {
this.totalTokenCount = totalTokenCount;
}


public AIAPIQuotaLimitAllOfDTO promptTokenCount(Long promptTokenCount) {

this.promptTokenCount = promptTokenCount;
return this;
}

/**
* Maximum number of prompt tokens allowed
* @return promptTokenCount
**/
@javax.annotation.Nullable
@ApiModelProperty(example = "400", value = "Maximum number of prompt tokens allowed")

public Long getPromptTokenCount() {
return promptTokenCount;
}


public void setPromptTokenCount(Long promptTokenCount) {
this.promptTokenCount = promptTokenCount;
}


public AIAPIQuotaLimitAllOfDTO completionTokenCount(Long completionTokenCount) {

this.completionTokenCount = completionTokenCount;
return this;
}

/**
* Maximum number of completion tokens allowed
* @return completionTokenCount
**/
@javax.annotation.Nullable
@ApiModelProperty(example = "500", value = "Maximum number of completion tokens allowed")

public Long getCompletionTokenCount() {
return completionTokenCount;
}


public void setCompletionTokenCount(Long completionTokenCount) {
this.completionTokenCount = completionTokenCount;
}


@Override
public boolean equals(Object o) {
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
AIAPIQuotaLimitAllOfDTO aiAPIQuotaLimitAllOf = (AIAPIQuotaLimitAllOfDTO) o;
return Objects.equals(this.requestCount, aiAPIQuotaLimitAllOf.requestCount) &&
Objects.equals(this.totalTokenCount, aiAPIQuotaLimitAllOf.totalTokenCount) &&
Objects.equals(this.promptTokenCount, aiAPIQuotaLimitAllOf.promptTokenCount) &&
Objects.equals(this.completionTokenCount, aiAPIQuotaLimitAllOf.completionTokenCount);
}

@Override
public int hashCode() {
return Objects.hash(requestCount, totalTokenCount, promptTokenCount, completionTokenCount);
}


@Override
public String toString() {
StringBuilder sb = new StringBuilder();
sb.append("class AIAPIQuotaLimitAllOfDTO {\n");
sb.append(" requestCount: ").append(toIndentedString(requestCount)).append("\n");
sb.append(" totalTokenCount: ").append(toIndentedString(totalTokenCount)).append("\n");
sb.append(" promptTokenCount: ").append(toIndentedString(promptTokenCount)).append("\n");
sb.append(" completionTokenCount: ").append(toIndentedString(completionTokenCount)).append("\n");
sb.append("}");
return sb.toString();
}

/**
* Convert the given object to string with each line indented by 4 spaces
* (except the first line).
*/
private String toIndentedString(Object o) {
if (o == null) {
return "null";
}
return o.toString().replace("\n", "\n ");
}

}

Loading

0 comments on commit 6209c86

Please sign in to comment.