1
- import { BuildHandlerArguments , Logger , MiddlewareStack } from "@aws-sdk/types" ;
1
+ import { Logger , MiddlewareStack } from "@aws-sdk/types" ;
2
2
3
3
import { getLoggerPlugin , loggerMiddleware , loggerMiddlewareOptions } from "./loggerMiddleware" ;
4
4
@@ -19,9 +19,18 @@ describe("getLoggerPlugin", () => {
19
19
} ) ;
20
20
21
21
describe ( "loggerMiddleware" , ( ) => {
22
- const next = jest . fn ( ) ;
22
+ const mockNext = jest . fn ( ) ;
23
+
24
+ const mockArgs = {
25
+ input : {
26
+ inputKey : "inputValue" ,
27
+ } ,
28
+ request : {
29
+ method : "GET" ,
30
+ headers : { } ,
31
+ } ,
32
+ } ;
23
33
24
- const args = { } ;
25
34
const mockResponse = {
26
35
output : {
27
36
$metadata : {
@@ -33,23 +42,23 @@ describe("loggerMiddleware", () => {
33
42
} ;
34
43
35
44
beforeEach ( ( ) => {
36
- next . mockResolvedValueOnce ( mockResponse ) ;
45
+ mockNext . mockResolvedValueOnce ( mockResponse ) ;
37
46
} ) ;
38
47
39
48
afterEach ( ( ) => {
40
49
jest . clearAllMocks ( ) ;
41
50
} ) ;
42
51
43
52
it ( "returns without logging if context.logger is not defined" , async ( ) => {
44
- const response = await loggerMiddleware ( ) ( next , { } ) ( args as BuildHandlerArguments < any > ) ;
45
- expect ( next ) . toHaveBeenCalledTimes ( 1 ) ;
53
+ const response = await loggerMiddleware ( ) ( mockNext , { } ) ( mockArgs ) ;
54
+ expect ( mockNext ) . toHaveBeenCalledTimes ( 1 ) ;
46
55
expect ( response ) . toStrictEqual ( mockResponse ) ;
47
56
} ) ;
48
57
49
58
it ( "returns without logging if context.logger doesn't have info function" , async ( ) => {
50
59
const logger = { } as Logger ;
51
- const response = await loggerMiddleware ( ) ( next , { logger } ) ( args as BuildHandlerArguments < any > ) ;
52
- expect ( next ) . toHaveBeenCalledTimes ( 1 ) ;
60
+ const response = await loggerMiddleware ( ) ( mockNext , { logger } ) ( mockArgs ) ;
61
+ expect ( mockNext ) . toHaveBeenCalledTimes ( 1 ) ;
53
62
expect ( response ) . toStrictEqual ( mockResponse ) ;
54
63
} ) ;
55
64
@@ -60,8 +69,8 @@ describe("loggerMiddleware", () => {
60
69
logger,
61
70
} ;
62
71
63
- const response = await loggerMiddleware ( ) ( next , context ) ( args as BuildHandlerArguments < any > ) ;
64
- expect ( next ) . toHaveBeenCalledTimes ( 1 ) ;
72
+ const response = await loggerMiddleware ( ) ( mockNext , context ) ( mockArgs ) ;
73
+ expect ( mockNext ) . toHaveBeenCalledTimes ( 1 ) ;
65
74
expect ( response ) . toStrictEqual ( mockResponse ) ;
66
75
67
76
expect ( logger . info ) . toHaveBeenCalledTimes ( 1 ) ;
0 commit comments