137 lines
3.2 KiB
Go
137 lines
3.2 KiB
Go
package tests
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"io"
|
|
"os/exec"
|
|
"slices"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func runner(commands []string, t *testing.T) []string {
|
|
executablePath := "../tmp/godb"
|
|
t.Logf("starting godb from %v\n", executablePath)
|
|
|
|
executable := exec.Command(executablePath)
|
|
|
|
results := make([]string, 0)
|
|
|
|
stdin, err := executable.StdinPipe()
|
|
if err != nil {
|
|
t.Errorf("Failed to create stdin %q\n", err)
|
|
}
|
|
|
|
stdout, err := executable.StdoutPipe()
|
|
if err != nil {
|
|
t.Errorf("Failed to create stdout %q\n", err)
|
|
}
|
|
|
|
if err := executable.Start(); err != nil {
|
|
t.Errorf("Failed to start executable %q\n", err)
|
|
}
|
|
|
|
go func() {
|
|
defer stdin.Close()
|
|
for _, c := range commands {
|
|
_, err = io.WriteString(stdin, c+"\n")
|
|
if err != nil {
|
|
t.Errorf("Failed to execute %v because of %q\n", c, err)
|
|
break
|
|
}
|
|
}
|
|
}()
|
|
|
|
reader := bufio.NewReader(stdout)
|
|
for {
|
|
// fmt.Printf("reading command outputs...\n")
|
|
line, err := reader.ReadString('\n')
|
|
if line != "" {
|
|
formattedline := fmt.Sprintf("%v", strings.Trim(strings.ReplaceAll(line, "\x00", ""), "\n"))
|
|
// fmt.Printf("formatting and appending result: %v\n", formattedline)
|
|
results = append(results, formattedline)
|
|
}
|
|
if err != nil {
|
|
t.Logf("stoped reading due to %q\n", err)
|
|
break
|
|
}
|
|
|
|
}
|
|
|
|
if err := executable.Wait(); err != nil {
|
|
t.Logf("Error occured while waiting for cleanup %q\n", err)
|
|
}
|
|
|
|
return results
|
|
}
|
|
|
|
func TestMain(t *testing.T) {
|
|
commands := []string{
|
|
"insert 1 user1 person1@example.com",
|
|
"select",
|
|
".exit",
|
|
}
|
|
|
|
expected := []string{
|
|
"godb > Executed.",
|
|
"godb > {1 user1 person1@example.com}",
|
|
"Executed.",
|
|
"godb > bye!",
|
|
}
|
|
results := runner(commands, t)
|
|
|
|
for i := range results {
|
|
results[i] = strings.ReplaceAll(results[i], "\n", "")
|
|
}
|
|
|
|
if !slices.Equal(expected, results) {
|
|
t.Errorf("Output Mismatch\nExpected: %v\nGot: %v\n", expected, results)
|
|
}
|
|
}
|
|
|
|
func TestTableFull(t *testing.T) {
|
|
maxRows := 1400
|
|
commands := make([]string, 0)
|
|
|
|
for i := range maxRows {
|
|
c := fmt.Sprintf("insert %d user%d person%d@example.com", i, i, i)
|
|
commands = append(commands, c)
|
|
}
|
|
commands = append(commands, ".exit")
|
|
|
|
results := runner(commands, t)
|
|
|
|
if results[1300] != "godb > Error: Table is full" {
|
|
t.Errorf("Our Tables max capacity is 1300 rows, but we're able to insert %v\n", len(results))
|
|
}
|
|
}
|
|
|
|
func TestMaxLengthForValues(t *testing.T) {
|
|
commands := []string{"insert 0 " + strings.Repeat("n", 32) + " " + strings.Repeat("n", 255), ".exit"}
|
|
expected := []string{"godb > Executed.", "godb > bye!"}
|
|
|
|
results := runner(commands, t)
|
|
for i := range results {
|
|
results[i] = strings.ReplaceAll(results[i], "\n", "")
|
|
}
|
|
|
|
if !slices.Equal(expected, results) {
|
|
t.Errorf("Output Mismatch\nExpected: %v\nGot: %v\n", expected, results)
|
|
}
|
|
}
|
|
|
|
func TestOverMaxLengthValues(t *testing.T) {
|
|
commands := []string{"insert 0 " + strings.Repeat("n", 42) + " " + strings.Repeat("n", 275), ".exit"}
|
|
expected := []string{"godb > Values for either username or email are too long", "godb > bye!"}
|
|
|
|
results := runner(commands, t)
|
|
for i := range results {
|
|
results[i] = strings.ReplaceAll(results[i], "\n", "")
|
|
}
|
|
|
|
if !slices.Equal(expected, results) {
|
|
t.Errorf("Output Mismatch\nExpected: %v\nGot: %v\n", expected, results)
|
|
}
|
|
}
|