如何更改控制器中的字段以进行测试?

时间:2018-01-29 22:14:15

标签: java spring-mvc spring-boot spring-mvc-test

我有下一个控制器:

@EnableWebMvc
@RequestMapping("experiences")
@RestController("experience_controller")
public class ExperienceController {

    @Autowired
    private ExperienceManager mExperienceManager;

    protected boolean mIsTest = false; //I need to replace by 'true'

    @PostMapping(value = "new")
    public ResponseEntity<ExperienceModel> add(
            @RequestBody NewExperienceModel newExperience
    ) {
        if (!mIsTest) {
            ExperienceUtils.validate(newExperience);
        }

        Experience experience = newExperience.toExperience();

        if (!mExperienceManager.add(experience)) {
            throw new InternalServerException(ServerError.INTERNAL);
        }

        return new ResponseEntity<>(new ExperienceModel(experience),
                HttpStatus.CREATED);
    }
}

我的应用测试类:

@TestPropertySource(properties = { "com.contedevel.virto.experience.controllers.ExperienceController.mIsTest=true" })
@RunWith(SpringRunner.class)
@SpringBootTest
@AutoConfigureMockMvc
public class ExperienceControllerTest {

    @Autowired
    private MockMvc mMockMvc;
    private ObjectMapper mMapper;
    private NewExperienceModel mModel;
    private String mExperienceId;

    @Before
    public void setUp() {
        mMapper = new ObjectMapper();
        UUID userId = UUID.randomUUID();
        UUID gameId = UUID.randomUUID();
        UUID achievementId = UUID.randomUUID();
        float value = 50f;
        mModel = new NewExperienceModel(userId, gameId, achievementId, value);
    }

    @Test()
    public void testOrder() throws Exception {
        testPost();
    }

    private void testPost() throws Exception {
        String json = mMapper.writeValueAsString(mModel);
        final String url = "/experiences/new";

        MvcResult result = mMockMvc.perform(post(url)
                .accept(MediaType.APPLICATION_JSON)
                .contentType(MediaType.APPLICATION_JSON)
                .content(json))
                .andDo(print())
                .andExpect(status().isCreated())
                .andReturn();
        String response = result.getResponse()
                .getContentAsString();
        JSONObject obj = new JSONObject(response);
        mExperienceId = obj.getString(ExperienceKeys.ID);
    }
}

我需要在我的控制器中设置mIsTest = true ...当然,我知道这段代码是错误的但是如何制作呢?或者我每次都需要手动更改它?

1 个答案:

答案 0 :(得分:1)

如何使用@Value?

public class ExperienceController {
    ...
    @Value("${flag.test:false}")
    protected boolean mIsTest;
    ...
}

测试代码

...
@TestPropertySource(properties = {"flag.test = true"})
public class ExperienceControllerTest {
   ...
}

mIsTest的默认值为false,您可以使用属性flag.test(或其他)更改mIsTest的值。您可以使用属性文件和活动配置文件更改其值。

如果您想快速了解@Value,请参阅此处:http://www.baeldung.com/spring-value-annotation