package example;

import com.amazonaws.services.lambda.runtime.ClientContext;

import com.amazonaws.services.lambda.runtime.CognitoIdentity;

import com.amazonaws.services.lambda.runtime.Context;

import com.amazonaws.services.lambda.runtime.LambdaLogger;

/**

* A simple mock implementation of the {@code Context} interface. Default

* values are stubbed out, and setters are provided so you can customize

* the context before passing it to your function.

*/

public class TestContext implements Context {

private String awsRequestId = "EXAMPLE";

private ClientContext clientContext;

private String functionName = "EXAMPLE";

private CognitoIdentity identity;

private String logGroupName = "EXAMPLE";

private String logStreamName = "EXAMPLE";

private LambdaLogger logger = new TestLogger();

private int memoryLimitInMB = 128;

private int remainingTimeInMillis = 10000;

private String functionVersion = "EXAMPLE";

private String invokedFunctionArn = "EXAMPLE";

@Override

public String getAwsRequestId() {

return awsRequestId;

}

public void setAwsRequestId(String value) {

awsRequestId = value;

}

@Override

public ClientContext getClientContext() {

return clientContext;

}

public void setClientContext(ClientContext value) {

clientContext = value;

}

@Override

public String getFunctionName() {

return functionName;

}

public void setFunctionName(String value) {

functionName = value;

}

@Override

public CognitoIdentity getIdentity() {

return identity;

}

public void setIdentity(CognitoIdentity value) {

identity = value;

}

@Override

public String getLogGroupName() {

return logGroupName;

}

public void setLogGroupName(String value) {

logGroupName = value;

}

@Override

public String getLogStreamName() {

return logStreamName;

}

public void setLogStreamName(String value) {

logStreamName = value;

}

@Override

public LambdaLogger getLogger() {

return logger;

}

public void setLogger(LambdaLogger value) {

logger = value;

}

@Override

public int getMemoryLimitInMB() {

return memoryLimitInMB;

}

public void setMemoryLimitInMB(int value) {

memoryLimitInMB = value;

}

@Override

public int getRemainingTimeInMillis() {

return remainingTimeInMillis;

}

public void setRemainingTimeInMillis(int value) {

remainingTimeInMillis = value;

}

@Override

public String getFunctionVersion() {

return functionVersion;

}

public void setFunctionVersion(String value) {

functionVersion = value;

}

@Override

public String getInvokedFunctionArn() {

return invokedFunctionArn;

}

public void setInvokedFunctionArn(String value) {

invokedFunctionArn = value;

}

/**

* A simple {@code LambdaLogger} that prints everything to stderr.

*/

private static class TestLogger implements LambdaLogger {

@Override

public void log(String message) {

System.err.println(message);

}

@Override

public void log(byte[] bytes) {

System.err.println(new String(bytes));

}

}

}