Fixed previousRoundRankRepository.test.ts issue with new take snapsho…

…t procedure
Giveth · Jul 31, 2023 · c25218b · c25218b
1 parent 58a84da
commit c25218b
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/repositories/previousRoundRankRepository.test.ts b/src/repositories/previousRoundRankRepository.test.ts
@@ -17,10 +17,7 @@ import {
   insertSinglePowerBoosting,
   takePowerBoostingSnapshot,
 } from './powerBoostingRepository';
-import {
-  findInCompletePowerSnapShots,
-  insertSinglePowerBalanceSnapshot,
-} from './powerSnapshotRepository';
+import { findInCompletePowerSnapShots } from './powerSnapshotRepository';
 import { getPowerRound, setPowerRound } from './powerRoundRepository';
 import {
   getProjectPowers,
@@ -32,6 +29,7 @@ import { assert } from 'chai';
 import { ProjectPowerView } from '../views/projectPowerView';
 import { findProjectById } from './projectRepository';
 import { PowerRound } from '../entities/powerRound';
+import { addOrUpdatePowerSnapshotBalances } from './powerBalanceSnapshotRepository';
 
 describe(
   'copyProjectRanksToPreviousRoundRankTable test cases',
@@ -82,7 +80,7 @@ const createSomeSampleProjectsAndPowerViews = async () => {
   snapshot.roundNumber = roundNumber;
   await snapshot.save();
 
-  await insertSinglePowerBalanceSnapshot({
+  await addOrUpdatePowerSnapshotBalances({
     userId: user.id,
     powerSnapshotId: snapshot.id,
     balance: 100,
@@ -146,7 +144,7 @@ function projectsThatTheirRanksHaveChangedTestCases() {
     snapshot.roundNumber = roundNumber;
     await snapshot.save();
 
-    await insertSinglePowerBalanceSnapshot({
+    await addOrUpdatePowerSnapshotBalances({
       userId: user.id,
       powerSnapshotId: snapshot.id,