[BE] 6주차 alpha 서버 배포

BE/.eslintrc.js

@@ -35,5 +35,7 @@ module.exports = {
     '@typescript-eslint/no-unsafe-assignment': 'off',
     '@typescript-eslint/no-unsafe-member-access': 'off',
     '@typescript-eslint/naming-convention': 'off',
+    'no-restricted-syntax': 'off',
+    'no-await-in-loop': 'off'
   },
 };

BE/src/asset/asset-service.spec.ts

@@ -0,0 +1,203 @@
+import { Test } from '@nestjs/testing';
+import { DeepPartial } from 'typeorm';
+import { AssetService } from './asset.service';
+import { UserStockRepository } from './user-stock.repository';
+import { AssetRepository } from './asset.repository';
+import { StockDetailService } from '../stock/detail/stock-detail.service';
+import { StockPriceSocketService } from '../stockSocket/stock-price-socket.service';
+import { TradeType } from '../stock/order/enum/trade-type';
+import { StatusType } from '../stock/order/enum/status-type';
+import { Asset } from './asset.entity';
+import { AssetResponseDto } from './dto/asset-response.dto';
+import { StockElementResponseDto } from './dto/stock-element-response.dto';
+import { MypageResponseDto } from './dto/mypage-response.dto';
+
+describe('asset test', () => {
+  let assetService: AssetService;
+  let userStockRepository: UserStockRepository;
+  let assetRepository: AssetRepository;
+  let stockDetailService: StockDetailService;
+
+  beforeEach(async () => {
+    const mockUserStockRepository = {
+      findOneBy: jest.fn(),
+      findUserStockWithNameByUserId: jest.fn(),
+      findAllDistinctCode: jest.fn(),
+      find: jest.fn(),
+    };
+    const mockAssetRepository = {
+      findAllPendingOrders: jest.fn(),
+      findOneBy: jest.fn(),
+      save: jest.fn(),
+    };
+    const mockStockDetailService = { getInquirePrice: jest.fn() };
+    const mockStockPriceSocketService = { subscribeByCode: jest.fn() };
+
+    const module = await Test.createTestingModule({
+      providers: [
+        AssetService,
+        { provide: UserStockRepository, useValue: mockUserStockRepository },
+        { provide: AssetRepository, useValue: mockAssetRepository },
+        {
+          provide: StockDetailService,
+          useValue: mockStockDetailService,
+        },
+        {
+          provide: StockPriceSocketService,
+          useValue: mockStockPriceSocketService,
+        },
+      ],
+    }).compile();
+
+    assetService = module.get(AssetService);
+    userStockRepository = module.get(UserStockRepository);
+    assetRepository = module.get(AssetRepository);
+    stockDetailService = module.get(StockDetailService);
+  });
+
+  it('보유 주식과 미체결 주문을 모두 반영한 매도 가능 주식 수를 반환한다.', async () => {
+    jest.spyOn(userStockRepository, 'findOneBy').mockResolvedValue({
+      id: 1,
+      user_id: 1,
+      stock_code: '005930',
+      quantity: 1,
+      avg_price: 1000,
+      last_updated: new Date(),
+    });
+
+    jest.spyOn(assetRepository, 'findAllPendingOrders').mockResolvedValue([
+      {
+        id: 1,
+        user_id: 1,
+        stock_code: '005930',
+        trade_type: TradeType.SELL,
+        amount: 1,
+        price: 1000,
+        status: StatusType.PENDING,
+        created_at: new Date(),
+      },
+    ]);
+
+    expect(await assetService.getUserStockByCode(1, '005930')).toEqual({
+      quantity: 0,
+      avg_price: 1000,
+    });
+  });
+
+  it('보유 자산과 미체결 주문을 모두 반영한 매수 가능 금액을 반환한다.', async () => {
+    jest.spyOn(assetRepository, 'findOneBy').mockResolvedValue({
+      id: 1,
+      user_id: 1,
+      stock_balance: 0,
+      cash_balance: 1000,
+      total_asset: 1000,
+      total_profit: 0,
+      total_profit_rate: 0,
+    });
+
+    jest.spyOn(assetRepository, 'findAllPendingOrders').mockResolvedValue([
+      {
+        id: 1,
+        user_id: 1,
+        stock_code: '005930',
+        trade_type: TradeType.BUY,
+        amount: 1,
+        price: 1000,
+        status: StatusType.PENDING,
+        created_at: new Date(),
+      },
+    ]);
+
+    expect(await assetService.getCashBalance(1)).toEqual({
+      cash_balance: 0,
+    });
+  });
+
+  it('마이페이지 조회 시 종목의 현재가를 반영한 총 자산을 반환한다.', async () => {
+    jest
+      .spyOn(userStockRepository, 'findUserStockWithNameByUserId')
+      .mockResolvedValue([
+        {
+          user_stocks_id: 1,
+          user_stocks_user_id: 1,
+          user_stocks_stock_code: '005930',
+          user_stocks_quantity: 1,
+          user_stocks_avg_price: '1000',
+          user_stocks_last_updated: new Date(),
+          stocks_code: '005930',
+          stocks_name: '삼성전자',
+          stocks_market: 'KOSPI',
+        },
+      ]);
+
+    jest.spyOn(assetRepository, 'findOneBy').mockResolvedValue({
+      id: 1,
+      user_id: 1,
+      stock_balance: 0,
+      cash_balance: 1000,
+      total_asset: 1000,
+      total_profit: 0,
+      total_profit_rate: 0,
+    });
+
+    jest
+      .spyOn(userStockRepository, 'findAllDistinctCode')
+      .mockResolvedValue([{ stock_code: '005930' }]);
+
+    jest.spyOn(stockDetailService, 'getInquirePrice').mockResolvedValue({
+      hts_kor_isnm: '삼성전자',
+      stck_shrn_iscd: '005930',
+      stck_prpr: '53600',
+      prdy_vrss: '-600',
+      prdy_vrss_sign: '5',
+      prdy_ctrt: '-1.11',
+      hts_avls: '3199803',
+      per: '25.15',
+      stck_mxpr: '70400',
+      stck_llam: '38000',
+      is_bookmarked: false,
+    });
+
+    jest.spyOn(userStockRepository, 'find').mockResolvedValue([
+      {
+        id: 1,
+        user_id: 1,
+        stock_code: '005930',
+        quantity: 1,
+        avg_price: 1000,
+        last_updated: new Date(),
+      },
+    ]);
+
+    jest
+      .spyOn(assetRepository, 'save')
+      .mockImplementation((updatedAsset) =>
+        Promise.resolve(updatedAsset as DeepPartial<Asset> & Asset),
+      );
+
+    const assetResponse = new AssetResponseDto(
+      1000,
+      53600,
+      54600,
+      -9945400,
+      '-99.45',
+      false,
+    );
+    const stockElementResponse = new StockElementResponseDto(
+      '삼성전자',
+      '005930',
+      1,
+      1000,
+      '53600',
+      '-600',
+      '5',
+      '-1.11',
+    );
+
+    const expected = new MypageResponseDto();
+    expected.asset = assetResponse;
+    expected.stocks = [stockElementResponse];
+
+    expect(await assetService.getMyPage(1)).toEqual(expected);
+  });
+});

BE/src/asset/asset.service.ts

@@ -148,17 +148,19 @@ export class AssetService {
     const userStocks: UserStock[] =
       await this.userStockRepository.findAllDistinctCode(userId);
 
-    userStocks.map((userStock) =>
-      this.stockPriceSocketService.subscribeByCode(userStock.stock_code),
+    await Promise.all(
+      userStocks.map((userStock) =>
+        this.stockPriceSocketService.subscribeByCode(userStock.stock_code),
+      ),
     );
   }
 
   async unsubscribeMyStocks(userId: number) {
     const userStocks: UserStock[] =
       await this.userStockRepository.findAllDistinctCode(userId);
 
-    userStocks.map((userStock) =>
-      this.stockPriceSocketService.unsubscribeByCode(userStock.stock_code),
+    await this.stockPriceSocketService.unsubscribeByCode(
+      userStocks.map((userStock) => userStock.stock_code),
     );
   }
 

BE/src/auth/auth.controller.ts

@@ -15,6 +15,7 @@ import { Request, Response } from 'express';
 import { ConfigService } from '@nestjs/config';
 import { AuthService } from './auth.service';
 import { AuthCredentialsDto } from './dto/auth-credentials.dto';
+import { OptionalAuthGuard } from './optional-auth-guard';
 
 @Controller('/api/auth')
 export class AuthController {
@@ -83,13 +84,14 @@ export class AuthController {
 
   @ApiOperation({ summary: '로그인 상태 확인 API' })
   @Get('/check')
-  @UseGuards(AuthGuard('jwt'))
+  @UseGuards(OptionalAuthGuard)
   @ApiResponse({
     status: 200,
     description: '로그인 상태 조회 성공',
     example: { isLogin: true },
   })
-  check() {
+  check(@Req() req: Request) {
+    if (!req.user) return { isLogin: false };
     return { isLogin: true };
   }
 

BE/src/common/redis/redis.domain-service.ts

@@ -4,19 +4,20 @@ import Redis from 'ioredis';
 @Injectable()
 export class RedisDomainService {
   constructor(
-    @Inject('REDIS_CLIENT')
-    private readonly redis: Redis,
+    @Inject('REDIS_CLIENT') private readonly redis: Redis,
+    @Inject('REDIS_PUBLISHER') private readonly publisher: Redis,
+    @Inject('REDIS_SUBSCRIBER') private readonly subscriber: Redis,
   ) {}
 
   async exists(key: string): Promise<number> {
     return this.redis.exists(key);
   }
 
-  async get(key: string): Promise<string | null> {
+  async get(key: string): Promise<string | number | null> {
     return this.redis.get(key);
   }
 
-  async set(key: string, value: string, expires?: number): Promise<'OK'> {
+  async set(key: string, value: number, expires?: number): Promise<'OK'> {
     if (expires) {
       return this.redis.set(key, value, 'EX', expires);
     }
@@ -62,4 +63,30 @@ export class RedisDomainService {
   async expire(key: string, seconds: number): Promise<number> {
     return this.redis.expire(key, seconds);
   }
+
+  async publish(channel: string, message: string) {
+    return this.publisher.publish(channel, message);
+  }
+
+  async subscribe(channel: string) {
+    await this.subscriber.subscribe(channel);
+  }
+
+  on(callback: (message: string) => void) {
+    this.redis.on('message', (message) => {
+      callback(message);
+    });
+  }
+
+  async unsubscribe(channel: string) {
+    return this.redis.unsubscribe(channel);
+  }
+
+  async increment(key: string) {
+    return this.redis.incr(key);
+  }
+
+  async decrement(key: string) {
+    return this.redis.decr(key);
+  }
 }

BE/src/common/redis/redis.module.ts

@@ -15,8 +15,31 @@ import { RedisDomainService } from './redis.domain-service';
         });
       },
     },
+    {
+      provide: 'REDIS_PUBLISHER',
+      useFactory: () => {
+        return new Redis({
+          host: process.env.REDIS_HOST || 'redis',
+          port: Number(process.env.REDIS_PORT || 6379),
+        });
+      },
+    },
+    {
+      provide: 'REDIS_SUBSCRIBER',
+      useFactory: () => {
+        return new Redis({
+          host: process.env.REDIS_HOST || 'redis',
+          port: Number(process.env.REDIS_PORT || 6379),
+        });
+      },
+    },
+    RedisDomainService,
+  ],
+  exports: [
     RedisDomainService,
+    'REDIS_CLIENT',
+    'REDIS_PUBLISHER',
+    'REDIS_SUBSCRIBER',
   ],
-  exports: [RedisDomainService, 'REDIS_CLIENT'],
 })
 export class RedisModule {}