@@ -42,4 +42,29 @@ public void works() throws IOException, URISyntaxException {
4242 assertThat (client .getTreatmentWithConfig ("user_e" , "split_2" ).treatment (), is (equalTo ("off" )));
4343 assertThat (client .getTreatmentWithConfig ("user_e" , "split_2" ).config (), is (equalTo ("{ \" size\" : 55 }" )));
4444 }
45+
46+ @ Test
47+ public void worksYML () throws IOException , URISyntaxException {
48+
49+ String file = getClass ().getClassLoader ().getResource ("split_compact.yml" ).getFile ();
50+
51+ SplitClientConfig config = SplitClientConfig .builder ().splitFile (file ).build ();
52+ SplitFactory splitFactory = SplitFactoryBuilder .build ("localhost" , config );
53+ SplitClient client = splitFactory .client ();
54+
55+ assertThat (client .getTreatment (null , "foo" ), is (equalTo (Treatments .CONTROL )));
56+ assertThat (client .getTreatment ("user_c" , "foo" ), is (equalTo (Treatments .CONTROL )));
57+
58+ assertThat (client .getTreatment ("user_c" , "split_1" ), is (equalTo ("off" )));
59+ assertThat (client .getTreatmentWithConfig ("user_c" , "split_1" ).treatment (), is (equalTo ("off" )));
60+ assertThat (client .getTreatmentWithConfig ("user_c" , "split_1" ).config (), is (equalTo ("{ \" size\" : 10 }" )));
61+
62+ assertThat (client .getTreatment ("user_d" , "split_1" ), is (equalTo ("on" )));
63+ assertThat (client .getTreatmentWithConfig ("user_d" , "split_1" ).treatment (), is (equalTo ("on" )));
64+ assertThat (client .getTreatmentWithConfig ("user_d" , "split_1" ).config (), is (nullValue ()));
65+
66+ assertThat (client .getTreatment ("user_e" , "split_2" ), is (equalTo ("off" )));
67+ assertThat (client .getTreatmentWithConfig ("user_e" , "split_2" ).treatment (), is (equalTo ("off" )));
68+ assertThat (client .getTreatmentWithConfig ("user_e" , "split_2" ).config (), is (equalTo ("{ \" size\" : 55 }" )));
69+ }
4570}
0 commit comments