def test_get_split_names(self, mocker): """Test getting all split names.""" uwsgi = get_uwsgi(True) storage = UWSGISplitStorage(uwsgi) from_raw_mock = self._get_from_raw_mock(mocker) mocker.patch('splitio.models.splits.from_raw', new=from_raw_mock) split_1 = from_raw_mock({'name': 'some_split_1', 'trafficTypeName': 'user'}) split_2 = from_raw_mock({'name': 'some_split_2', 'trafficTypeName': 'user'}) storage.put(split_1) storage.put(split_2) assert set(storage.get_split_names()) == set(['some_split_1', 'some_split_2']) storage.remove('some_split_1') assert storage.get_split_names() == ['some_split_2']
def test_get_split_names(self, mocker): """Test getting all split names.""" uwsgi = get_uwsgi(True) storage = UWSGISplitStorage(uwsgi) split_1 = mocker.Mock(spec=Split) split_1.to_json.return_value = '{"name": "split1"}' split_name_1 = mocker.PropertyMock() split_name_1.return_value = 'some_split_1' type(split_1).name = split_name_1 split_2 = mocker.Mock(spec=Split) split_2.to_json.return_value = '{"name": "split2"}' split_name_2 = mocker.PropertyMock() split_name_2.return_value = 'some_split_2' type(split_2).name = split_name_2 storage.put(split_1) storage.put(split_2) assert set(storage.get_split_names()) == set(['some_split_1', 'some_split_2']) storage.remove('some_split_1') assert storage.get_split_names() == ['some_split_2']