|
|
@@ -0,0 +1,50 @@
|
|
|
+package com.izouma.walkchina.repo;
|
|
|
+
|
|
|
+import com.izouma.walkchina.constant.StepRecordType;
|
|
|
+import com.izouma.walkchina.domain.StepRecord;
|
|
|
+import org.junit.Test;
|
|
|
+import org.junit.runner.RunWith;
|
|
|
+import org.springframework.beans.factory.annotation.Autowired;
|
|
|
+import org.springframework.boot.test.context.SpringBootTest;
|
|
|
+import org.springframework.test.context.junit4.SpringRunner;
|
|
|
+
|
|
|
+import static org.junit.Assert.*;
|
|
|
+
|
|
|
+@RunWith(SpringRunner.class)
|
|
|
+@SpringBootTest
|
|
|
+public class WalkDataRepoTest {
|
|
|
+ @Autowired
|
|
|
+ private WalkDataRepo walkDataRepo;
|
|
|
+ @Autowired
|
|
|
+ private StepRecordRepo stepRecordRepo;
|
|
|
+ @Autowired
|
|
|
+ private TeamMemberRepository teamMemberRepository;
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void test() {
|
|
|
+
|
|
|
+ walkDataRepo.findAll().stream().forEach(walkData -> {
|
|
|
+ stepRecordRepo.save(StepRecord.builder()
|
|
|
+ .userId(walkData.getUserId())
|
|
|
+ .fromUserId(walkData.getUserId())
|
|
|
+ .type(StepRecordType.SELF)
|
|
|
+ .date(walkData.getDate())
|
|
|
+ .step(Math.toIntExact(walkData.getSteps()))
|
|
|
+ .build());
|
|
|
+ });
|
|
|
+
|
|
|
+
|
|
|
+ teamMemberRepository.findAll().forEach(teamMember -> {
|
|
|
+ walkDataRepo.findAllByDateGreaterThanEqualAndDateLessThanEqualAndUserIdEquals(teamMember.getStartTime().toLocalDate(), teamMember.getEndTime().toLocalDate(), teamMember.getUserId())
|
|
|
+ .forEach(walkData -> {
|
|
|
+ stepRecordRepo.save(StepRecord.builder()
|
|
|
+ .userId(teamMember.getLeader())
|
|
|
+ .fromUserId(walkData.getUserId())
|
|
|
+ .type(StepRecordType.TEAM)
|
|
|
+ .date(walkData.getDate())
|
|
|
+ .step(Math.toIntExact(walkData.getSteps()))
|
|
|
+ .build());
|
|
|
+ });
|
|
|
+ });
|
|
|
+ }
|
|
|
+}
|