fix: #9 no error on trying to insert more than limit, plus some other

tests and minor fix
This commit is contained in:
2026-07-29 00:35:03 +05:30
parent 1e297a45ab
commit 9eeac4e7df
3 changed files with 72 additions and 10 deletions
+4
View File
@@ -37,6 +37,10 @@ func Run() {
fmt.Println("Syntax error, could not parse statement") fmt.Println("Syntax error, could not parse statement")
case PrepareUnrecognised: case PrepareUnrecognised:
fmt.Printf("Unrecognised keyword at the start of '%s'.\n", inputBuffer.Buffer) fmt.Printf("Unrecognised keyword at the start of '%s'.\n", inputBuffer.Buffer)
case PrepareTooManyArgs:
fmt.Printf("Too many arguments in the insert statement '%v'\n", inputBuffer.Buffer)
case PrepareValueTooLong:
fmt.Printf("Values for either username or email are too long\n")
} }
switch statement.ExecuteStatement(table) { switch statement.ExecuteStatement(table) {
+40 -10
View File
@@ -4,6 +4,8 @@ import (
"encoding/binary" "encoding/binary"
"fmt" "fmt"
"os" "os"
"slices"
"strconv"
"strings" "strings"
. "godb/internals/buffer" . "godb/internals/buffer"
@@ -21,6 +23,8 @@ type PrepareResults int
const ( const (
PrepareSuccess PrepareResults = iota PrepareSuccess PrepareResults = iota
PrepareUnrecognised PrepareUnrecognised
PrepareTooManyArgs
PrepareValueTooLong
PrepareSyntaxError PrepareSyntaxError
) )
@@ -151,19 +155,45 @@ func DoMetaCommands(inputBuffer *InputBuffer) MetaCommandResults {
return metacommand return metacommand
} }
func (statement *Statement) perpareInsert(inputBuffer *InputBuffer) PrepareResults {
inputs := strings.Split(inputBuffer.Buffer, " ")[1:]
if len(inputs) < 3 {
return PrepareSyntaxError
}
if len(inputs) > 3 {
return PrepareTooManyArgs
}
if slices.Contains(inputs, "null") || slices.Contains(inputs, "NULL") || slices.Contains(inputs, "Null") || slices.Contains(inputs, " ") {
return PrepareSyntaxError
}
tid, err := strconv.ParseUint(inputs[0], 10, 0)
if err != nil {
// fmt.Printf("supplied ID: %v isn't acceptable, please correct and retry\n", inputs[0])
return PrepareSyntaxError
}
id := uint(tid)
if len(inputs[1]) > usernameSize {
// fmt.Printf("%v is too long for username\n", inputs[1])
return PrepareValueTooLong
}
if len(inputs[2]) > emailSize {
// fmt.Printf("%v is too long for email\n", inputs[2])
return PrepareValueTooLong
}
statement.row.id = id
statement.row.username = inputs[1]
statement.row.email = inputs[2]
statement.stype = StatementInsert
return PrepareSuccess
}
func (statement *Statement) PrepareStatements(inputBuffer *InputBuffer) PrepareResults { func (statement *Statement) PrepareStatements(inputBuffer *InputBuffer) PrepareResults {
statement.stype = StatementUnrecognized statement.stype = StatementUnrecognized
if strings.HasPrefix(inputBuffer.Buffer, "insert") { if strings.HasPrefix(inputBuffer.Buffer, "insert") {
argsAssigned, err := fmt.Sscanf(inputBuffer.Buffer, "insert %d %s %s", &statement.row.id, &statement.row.username, &statement.row.email) return statement.perpareInsert(inputBuffer)
if err != nil {
fmt.Println("error parsing insert statement")
fmt.Println(err.Error())
}
if err != nil || argsAssigned > 3 {
return PrepareSyntaxError
}
statement.stype = StatementInsert
return PrepareSuccess
} }
if inputBuffer.Buffer == "select" { if inputBuffer.Buffer == "select" {
statement.stype = StatementSelect statement.stype = StatementSelect
+28
View File
@@ -106,3 +106,31 @@ func TestTableFull(t *testing.T) {
t.Errorf("Our Tables max capacity is 1300 rows, but we're able to insert %v\n", len(results)) 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)
}
}