io.github.jsonSnapshot.SnapshotFile Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of json-snapshot Show documentation

Snapshot Testing for Java

There is a newer version: 1.0.17

package io.github.jsonSnapshot;

import lombok.Getter;
import org.apache.commons.lang3.StringUtils;

import java.io.*;
import java.util.Set;
import java.util.TreeSet;
import java.util.stream.Collectors;
import java.util.stream.Stream;

public class SnapshotFile {

    private static final String SPLIT_STRING = "\n\n\n";

    private String fileName;

    @Getter
    private Set rawSnapshots;

    SnapshotFile(String filePath, String fileName) throws IOException {

        this.fileName = filePath + fileName;

        StringBuilder fileContent = new StringBuilder();

        try (BufferedReader br = new BufferedReader(new FileReader(this.fileName))) {

            String sCurrentLine;

            while ((sCurrentLine = br.readLine()) != null) {
                fileContent.append(sCurrentLine + "\n");
            }

            String fileText = fileContent.toString();
            if (StringUtils.isNotBlank(fileText)) {
                rawSnapshots = Stream.of(fileContent.toString().split(SPLIT_STRING)).map(String::trim).collect(Collectors.toCollection(
                        TreeSet::new));
            }
            else {
                rawSnapshots = new TreeSet<>();
            }
        } catch (IOException e) {
            createFile(this.fileName);
            rawSnapshots = new TreeSet<>();
        }
    }

    private File createFile(String fileName) throws IOException {
        File file = new File(fileName);
        file.getParentFile().mkdirs();
        file.createNewFile();
        return file;
    }

    public void push(String snapshot) {

        rawSnapshots.add(snapshot);

        File file = null;

        try {
            file = createFile(fileName);
        } catch (IOException e) {
            e.printStackTrace();
        }

        try (FileOutputStream fileStream = new FileOutputStream(file, false)) {
            byte[] myBytes = StringUtils.join(rawSnapshots, SPLIT_STRING).getBytes();
            fileStream.write(myBytes);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}