You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

207 lines
4.9 KiB
Go

package dbi
import (
"testing"
)
func genActivity(username string) Activity {
return Activity{
Title: randString(20),
Description: randString(60),
Hyperlink: "https://" + randString(12) + ".example.com",
Status: "reported",
Username: username,
}
}
func TestCreateActivity(t *testing.T) {
conn := testConn(t, testDB(t))
a := genActivity(randString(10))
if err := CreateActivity(conn, &a); err != nil {
t.Fatalf("CreateActivity: %v", err)
}
if a.UID == "" {
t.Error("expected UID to be set after create")
}
if a.Created == "" {
t.Error("expected Created to be set after create")
}
if a.Modified == "" {
t.Error("expected Modified to be set after create")
}
got, err := GetActivity(conn, a.UID)
if err != nil {
t.Fatalf("GetActivity after create: %v", err)
}
if got != a {
t.Errorf("got %+v, want %+v", got, a)
}
}
func TestCreateActivityPresetUID(t *testing.T) {
conn := testConn(t, testDB(t))
a := genActivity(randString(10))
a.UID = "preset" + randString(10)
presetUID := a.UID
if err := CreateActivity(conn, &a); err != nil {
t.Fatalf("CreateActivity: %v", err)
}
if a.UID != presetUID {
t.Error("CreateActivity should not overwrite a non-empty UID")
}
}
func TestCreateActivityDuplicateUID(t *testing.T) {
conn := testConn(t, testDB(t))
a := genActivity(randString(10))
if err := CreateActivity(conn, &a); err != nil {
t.Fatalf("first CreateActivity: %v", err)
}
a2 := a
if err := CreateActivity(conn, &a2); err == nil {
t.Error("expected error on duplicate UID, got nil")
}
}
func TestGetActivity(t *testing.T) {
conn := testConn(t, testDB(t))
a := genActivity(randString(10))
if err := CreateActivity(conn, &a); err != nil {
t.Fatalf("CreateActivity: %v", err)
}
got, err := GetActivity(conn, a.UID)
if err != nil {
t.Fatalf("GetActivity: %v", err)
}
if got != a {
t.Errorf("got %+v, want %+v", got, a)
}
}
func TestGetActivityNotFound(t *testing.T) {
conn := testConn(t, testDB(t))
_, err := GetActivity(conn, randString(32))
if err == nil {
t.Error("expected error for missing activity, got nil")
}
}
func TestUpdateActivity(t *testing.T) {
conn := testConn(t, testDB(t))
a := genActivity(randString(10))
if err := CreateActivity(conn, &a); err != nil {
t.Fatalf("CreateActivity: %v", err)
}
originalUID := a.UID
originalCreated := a.Created
a.Title = randString(20)
a.Description = randString(60)
a.Hyperlink = "https://" + randString(12) + ".example.com"
a.Status = "posted"
if err := UpdateActivity(conn, &a); err != nil {
t.Fatalf("UpdateActivity: %v", err)
}
got, err := GetActivity(conn, a.UID)
if err != nil {
t.Fatalf("GetActivity 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.Title != a.Title {
t.Errorf("Title not updated: got %q, want %q", got.Title, a.Title)
}
if got.Description != a.Description {
t.Errorf("Description not updated: got %q, want %q", got.Description, a.Description)
}
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 TestDeleteActivity(t *testing.T) {
conn := testConn(t, testDB(t))
a := genActivity(randString(10))
if err := CreateActivity(conn, &a); err != nil {
t.Fatalf("CreateActivity: %v", err)
}
if err := DeleteActivity(conn, a.UID); err != nil {
t.Fatalf("DeleteActivity: %v", err)
}
_, err := GetActivity(conn, a.UID)
if err == nil {
t.Error("expected error after delete, got nil")
}
}
func TestGetActivitiesForUsername(t *testing.T) {
conn := testConn(t, testDB(t))
username := randString(10)
items := make([]Activity, 3)
for i := range items {
items[i] = genActivity(username)
if err := CreateActivity(conn, &items[i]); err != nil {
t.Fatalf("CreateActivity %d: %v", i, err)
}
}
other := genActivity(randString(10))
if err := CreateActivity(conn, &other); err != nil {
t.Fatalf("CreateActivity other: %v", err)
}
got, err := GetActivitiesForUsername(conn, username)
if err != nil {
t.Fatalf("GetActivitiesForUsername: %v", err)
}
if len(got) != 3 {
t.Errorf("got %d activities, want 3", len(got))
}
byUID := make(map[string]Activity)
for _, a := range got {
byUID[a.UID] = a
}
for _, want := range items {
a, ok := byUID[want.UID]
if !ok {
t.Errorf("activity %q missing from results", want.UID)
continue
}
if a != want {
t.Errorf("activity %q: got %+v, want %+v", want.UID, a, want)
}
}
}
func TestGetActivitiesForUsernameEmpty(t *testing.T) {
conn := testConn(t, testDB(t))
got, err := GetActivitiesForUsername(conn, randString(10))
if err != nil {
t.Fatalf("GetActivitiesForUsername: %v", err)
}
if len(got) != 0 {
t.Errorf("expected empty slice, got %d items", len(got))
}
}