add CRUD for Scholarship
Joshua Herring
3 weeks ago
parent
86ebe6bb90
commit
694a246957
@ -0,0 +1,90 @@
|
||||
package dbi
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"fmt"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
)
|
||||
|
||||
type Scholarship struct {
|
||||
UID string `json:"uid"`
|
||||
Citation string `json:"citation"`
|
||||
Hyperlink string `json:"hyperlink"`
|
||||
Status string `json:"status"`
|
||||
Created string `json:"created"`
|
||||
Modified string `json:"modified"`
|
||||
Username string `json:"username"`
|
||||
}
|
||||
|
||||
func CreateScholarship(conn *sql.Conn, s *Scholarship) error {
|
||||
if s.UID == "" {
|
||||
s.UID = strings.ReplaceAll(uuid.New().String(), "-", "")
|
||||
}
|
||||
now := time.Now().Format("2006-01-02 15:04:05")
|
||||
s.Created = now
|
||||
s.Modified = now
|
||||
_, err := conn.ExecContext(context.Background(),
|
||||
`INSERT INTO scholarship (UID, Citation, Hyperlink, Status, Created, Modified, Username)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?)`,
|
||||
s.UID, s.Citation, s.Hyperlink, s.Status, s.Created, s.Modified, s.Username,
|
||||
)
|
||||
return err
|
||||
}
|
||||
|
||||
func GetScholarship(conn *sql.Conn, uid string) (Scholarship, error) {
|
||||
row := conn.QueryRowContext(context.Background(),
|
||||
`SELECT UID, Citation, Hyperlink, Status, Created, Modified, Username
|
||||
FROM scholarship WHERE UID = ?`,
|
||||
uid,
|
||||
)
|
||||
var s Scholarship
|
||||
err := row.Scan(&s.UID, &s.Citation, &s.Hyperlink, &s.Status, &s.Created, &s.Modified, &s.Username)
|
||||
if err == sql.ErrNoRows {
|
||||
return Scholarship{}, fmt.Errorf("scholarship %q not found", uid)
|
||||
}
|
||||
return s, err
|
||||
}
|
||||
|
||||
func UpdateScholarship(conn *sql.Conn, s *Scholarship) error {
|
||||
s.Modified = time.Now().Format("2006-01-02 15:04:05")
|
||||
_, err := conn.ExecContext(context.Background(),
|
||||
`UPDATE scholarship SET Citation = ?, Hyperlink = ?, Status = ?, Modified = ?, Username = ?
|
||||
WHERE UID = ?`,
|
||||
s.Citation, s.Hyperlink, s.Status, s.Modified, s.Username, s.UID,
|
||||
)
|
||||
return err
|
||||
}
|
||||
|
||||
func DeleteScholarship(conn *sql.Conn, uid string) error {
|
||||
_, err := conn.ExecContext(context.Background(),
|
||||
`DELETE FROM scholarship WHERE UID = ?`,
|
||||
uid,
|
||||
)
|
||||
return err
|
||||
}
|
||||
|
||||
func GetScholarshipsForUsername(conn *sql.Conn, username string) ([]Scholarship, error) {
|
||||
rows, err := conn.QueryContext(context.Background(),
|
||||
`SELECT UID, Citation, Hyperlink, Status, Created, Modified, Username
|
||||
FROM scholarship WHERE Username = ?`,
|
||||
username,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var results []Scholarship
|
||||
for rows.Next() {
|
||||
var s Scholarship
|
||||
if err := rows.Scan(&s.UID, &s.Citation, &s.Hyperlink, &s.Status, &s.Created, &s.Modified, &s.Username); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
results = append(results, s)
|
||||
}
|
||||
return results, rows.Err()
|
||||
}
|
||||
@ -0,0 +1,210 @@
|
||||
package dbi
|
||||
|
||||
import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
func genScholarship(username string) Scholarship {
|
||||
return Scholarship{
|
||||
Citation: randString(40),
|
||||
Hyperlink: "https://" + randString(12) + ".example.com",
|
||||
Status: "reported",
|
||||
Username: username,
|
||||
}
|
||||
}
|
||||
|
||||
func TestCreateScholarship(t *testing.T) {
|
||||
conn := testConn(t, testDB(t))
|
||||
s := genScholarship(randString(10))
|
||||
|
||||
if err := CreateScholarship(conn, &s); err != nil {
|
||||
t.Fatalf("CreateScholarship: %v", err)
|
||||
}
|
||||
|
||||
if s.UID == "" {
|
||||
t.Error("expected UID to be set after create")
|
||||
}
|
||||
if s.Created == "" {
|
||||
t.Error("expected Created to be set after create")
|
||||
}
|
||||
if s.Modified == "" {
|
||||
t.Error("expected Modified to be set after create")
|
||||
}
|
||||
|
||||
got, err := GetScholarship(conn, s.UID)
|
||||
if err != nil {
|
||||
t.Fatalf("GetScholarship after create: %v", err)
|
||||
}
|
||||
if got != s {
|
||||
t.Errorf("got %+v, want %+v", got, s)
|
||||
}
|
||||
}
|
||||
|
||||
func TestCreateScholarshipPresetUID(t *testing.T) {
|
||||
conn := testConn(t, testDB(t))
|
||||
s := genScholarship(randString(10))
|
||||
s.UID = "preset" + randString(10)
|
||||
|
||||
if err := CreateScholarship(conn, &s); err != nil {
|
||||
t.Fatalf("CreateScholarship: %v", err)
|
||||
}
|
||||
if s.UID != "preset"+s.UID[6:] {
|
||||
t.Error("CreateScholarship should not overwrite a non-empty UID")
|
||||
}
|
||||
|
||||
got, err := GetScholarship(conn, s.UID)
|
||||
if err != nil {
|
||||
t.Fatalf("GetScholarship: %v", err)
|
||||
}
|
||||
if got.UID != s.UID {
|
||||
t.Errorf("got UID %q, want %q", got.UID, s.UID)
|
||||
}
|
||||
}
|
||||
|
||||
func TestCreateScholarshipDuplicateUID(t *testing.T) {
|
||||
conn := testConn(t, testDB(t))
|
||||
s := genScholarship(randString(10))
|
||||
|
||||
if err := CreateScholarship(conn, &s); err != nil {
|
||||
t.Fatalf("first CreateScholarship: %v", err)
|
||||
}
|
||||
s2 := s
|
||||
if err := CreateScholarship(conn, &s2); err == nil {
|
||||
t.Error("expected error on duplicate UID, got nil")
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetScholarship(t *testing.T) {
|
||||
conn := testConn(t, testDB(t))
|
||||
s := genScholarship(randString(10))
|
||||
|
||||
if err := CreateScholarship(conn, &s); err != nil {
|
||||
t.Fatalf("CreateScholarship: %v", err)
|
||||
}
|
||||
|
||||
got, err := GetScholarship(conn, s.UID)
|
||||
if err != nil {
|
||||
t.Fatalf("GetScholarship: %v", err)
|
||||
}
|
||||
if got != s {
|
||||
t.Errorf("got %+v, want %+v", got, s)
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetScholarshipNotFound(t *testing.T) {
|
||||
conn := testConn(t, testDB(t))
|
||||
|
||||
_, err := GetScholarship(conn, randString(32))
|
||||
if err == nil {
|
||||
t.Error("expected error for missing scholarship, got nil")
|
||||
}
|
||||
}
|
||||
|
||||
func TestUpdateScholarship(t *testing.T) {
|
||||
conn := testConn(t, testDB(t))
|
||||
s := genScholarship(randString(10))
|
||||
|
||||
if err := CreateScholarship(conn, &s); err != nil {
|
||||
t.Fatalf("CreateScholarship: %v", err)
|
||||
}
|
||||
|
||||
originalUID := s.UID
|
||||
originalCreated := s.Created
|
||||
s.Citation = randString(40)
|
||||
s.Hyperlink = "https://" + randString(12) + ".example.com"
|
||||
s.Status = "posted"
|
||||
|
||||
if err := UpdateScholarship(conn, &s); err != nil {
|
||||
t.Fatalf("UpdateScholarship: %v", err)
|
||||
}
|
||||
|
||||
got, err := GetScholarship(conn, s.UID)
|
||||
if err != nil {
|
||||
t.Fatalf("GetScholarship after update: %v", err)
|
||||
}
|
||||
if got.UID != originalUID {
|
||||
t.Errorf("UID changed: got %q, want %q", got.UID, originalUID)
|
||||
}
|
||||
if got.Created != originalCreated {
|
||||
t.Errorf("Created changed: got %q, want %q", got.Created, originalCreated)
|
||||
}
|
||||
if got.Citation != s.Citation {
|
||||
t.Errorf("Citation not updated: got %q, want %q", got.Citation, s.Citation)
|
||||
}
|
||||
if got.Status != "posted" {
|
||||
t.Errorf("Status not updated: got %q", got.Status)
|
||||
}
|
||||
if got.Modified == "" {
|
||||
t.Error("Modified should be set after update")
|
||||
}
|
||||
}
|
||||
|
||||
func TestDeleteScholarship(t *testing.T) {
|
||||
conn := testConn(t, testDB(t))
|
||||
s := genScholarship(randString(10))
|
||||
|
||||
if err := CreateScholarship(conn, &s); err != nil {
|
||||
t.Fatalf("CreateScholarship: %v", err)
|
||||
}
|
||||
|
||||
if err := DeleteScholarship(conn, s.UID); err != nil {
|
||||
t.Fatalf("DeleteScholarship: %v", err)
|
||||
}
|
||||
|
||||
_, err := GetScholarship(conn, s.UID)
|
||||
if err == nil {
|
||||
t.Error("expected error after delete, got nil")
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetScholarshipsForUsername(t *testing.T) {
|
||||
conn := testConn(t, testDB(t))
|
||||
username := randString(10)
|
||||
|
||||
items := make([]Scholarship, 3)
|
||||
for i := range items {
|
||||
items[i] = genScholarship(username)
|
||||
if err := CreateScholarship(conn, &items[i]); err != nil {
|
||||
t.Fatalf("CreateScholarship %d: %v", i, err)
|
||||
}
|
||||
}
|
||||
// extra item belonging to a different user
|
||||
other := genScholarship(randString(10))
|
||||
if err := CreateScholarship(conn, &other); err != nil {
|
||||
t.Fatalf("CreateScholarship other: %v", err)
|
||||
}
|
||||
|
||||
got, err := GetScholarshipsForUsername(conn, username)
|
||||
if err != nil {
|
||||
t.Fatalf("GetScholarshipsForUsername: %v", err)
|
||||
}
|
||||
if len(got) != 3 {
|
||||
t.Errorf("got %d scholarships, want 3", len(got))
|
||||
}
|
||||
byUID := make(map[string]Scholarship)
|
||||
for _, s := range got {
|
||||
byUID[s.UID] = s
|
||||
}
|
||||
for _, want := range items {
|
||||
s, ok := byUID[want.UID]
|
||||
if !ok {
|
||||
t.Errorf("scholarship %q missing from results", want.UID)
|
||||
continue
|
||||
}
|
||||
if s != want {
|
||||
t.Errorf("scholarship %q: got %+v, want %+v", want.UID, s, want)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetScholarshipsForUsernameEmpty(t *testing.T) {
|
||||
conn := testConn(t, testDB(t))
|
||||
|
||||
got, err := GetScholarshipsForUsername(conn, randString(10))
|
||||
if err != nil {
|
||||
t.Fatalf("GetScholarshipsForUsername: %v", err)
|
||||
}
|
||||
if len(got) != 0 {
|
||||
t.Errorf("expected empty slice, got %d items", len(got))
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue