small test

chirst · Jun 27, 2024 · b6d6187 · b6d6187
1 parent 6897000
commit b6d6187
Showing 1 changed file with 81 additions and 48 deletions.
diff --git a/db/db_test.go b/db/db_test.go
@@ -5,52 +5,85 @@ import (
 )
 
 func TestExecute(t *testing.T) {
-	db, err := New(true)
-	if err != nil {
-		t.Fatal(err)
-	}
-	createSql := "CREATE TABLE person (id INTEGER, first_name TEXT, last_name TEXT, age INTEGER)"
-	createRes := db.Execute(createSql)
-	if createRes.Err != nil {
-		t.Fatal(createRes.Err.Error())
-	}
-	selectSchemaSql := "SELECT * FROM cdb_schema"
-	schemaRes := db.Execute(selectSchemaSql)
-	if schemaRes.Err != nil {
-		t.Fatal(schemaRes.Err.Error())
-	}
-	schemaSelectExpectations := []string{
-		"1",
-		"table",
-		"person",
-		"person",
-		"2",
-		"{\"columns\":[{\"name\":\"id\",\"type\":\"INTEGER\"},{\"name\":\"first_name\",\"type\":\"TEXT\"},{\"name\":\"last_name\",\"type\":\"TEXT\"},{\"name\":\"age\",\"type\":\"INTEGER\"}]}",
-	}
-	for i, s := range schemaSelectExpectations {
-		if c := *schemaRes.ResultRows[1][i]; c != s {
-			t.Fatalf("expected %s got %s", s, c)
-		}
-	}
-	insertSql := "INSERT INTO person (first_name, last_name, age) VALUES ('John', 'Smith', 50)"
-	insertRes := db.Execute(insertSql)
-	if insertRes.Err != nil {
-		t.Fatal(insertRes.Err.Error())
-	}
-	selectPersonSql := "SELECT * FROM person"
-	selectPersonRes := db.Execute(selectPersonSql)
-	if selectPersonRes.Err != nil {
-		t.Fatal(selectPersonRes.Err.Error())
-	}
-	selectPersonExpectations := []string{
-		"1",
-		"John",
-		"Smith",
-		"50",
-	}
-	for i, s := range selectPersonExpectations {
-		if c := *selectPersonRes.ResultRows[1][i]; c != s {
-			t.Fatalf("expected %s got %s", s, c)
-		}
-	}
+	t.Run("create, select schema, insert, select", func(t *testing.T) {
+		db, err := New(true)
+		if err != nil {
+			t.Fatal(err.Error())
+		}
+		createSql := "CREATE TABLE person (id INTEGER, first_name TEXT, last_name TEXT, age INTEGER)"
+		createRes := db.Execute(createSql)
+		if createRes.Err != nil {
+			t.Fatal(createRes.Err.Error())
+		}
+		selectSchemaSql := "SELECT * FROM cdb_schema"
+		schemaRes := db.Execute(selectSchemaSql)
+		if schemaRes.Err != nil {
+			t.Fatal(schemaRes.Err.Error())
+		}
+		schemaSelectExpectations := []string{
+			"1",
+			"table",
+			"person",
+			"person",
+			"2",
+			"{\"columns\":[{\"name\":\"id\",\"type\":\"INTEGER\"},{\"name\":\"first_name\",\"type\":\"TEXT\"},{\"name\":\"last_name\",\"type\":\"TEXT\"},{\"name\":\"age\",\"type\":\"INTEGER\"}]}",
+		}
+		for i, s := range schemaSelectExpectations {
+			if c := *schemaRes.ResultRows[1][i]; c != s {
+				t.Fatalf("expected %s got %s", s, c)
+			}
+		}
+		insertSql := "INSERT INTO person (first_name, last_name, age) VALUES ('John', 'Smith', 50)"
+		insertRes := db.Execute(insertSql)
+		if insertRes.Err != nil {
+			t.Fatal(insertRes.Err.Error())
+		}
+		selectPersonSql := "SELECT * FROM person"
+		selectPersonRes := db.Execute(selectPersonSql)
+		if selectPersonRes.Err != nil {
+			t.Fatal(selectPersonRes.Err.Error())
+		}
+		selectPersonExpectations := []string{
+			"1",
+			"John",
+			"Smith",
+			"50",
+		}
+		for i, s := range selectPersonExpectations {
+			if c := *selectPersonRes.ResultRows[1][i]; c != s {
+				t.Fatalf("expected %s got %s", s, c)
+			}
+		}
+	})
+
+	t.Run("bulk insert", func(t *testing.T) {
+		db, err := New(true)
+		if err != nil {
+			t.Fatal(err.Error())
+		}
+		createSql := "CREATE TABLE test (id INTEGER, junk TEXT)"
+		createRes := db.Execute(createSql)
+		if createRes.Err != nil {
+			t.Fatal(createRes.Err.Error())
+		}
+		expectedTotal := 5000
+		for i := 0; i < expectedTotal; i += 1 {
+			insertSql := "INSERT INTO test (junk) VALUES ('asdf')"
+			insertRes := db.Execute(insertSql)
+			if insertRes.Err != nil {
+				t.Fatal(insertRes.Err.Error())
+			}
+		}
+		selectSql := "SELECT * FROM test"
+		selectRes := db.Execute(selectSql)
+		if selectRes.Err != nil {
+			t.Fatal(selectRes.Err.Error())
+		}
+		// minus 1 due to the result header being part. TODO the result header
+		// should be on its own.
+		if gotT := len(selectRes.ResultRows) - 1; expectedTotal != gotT {
+			t.Fatalf("expected %d got %d", expectedTotal, gotT)
+		}
+
+	})
 }