mysql_test.go

package db

import (
  "testing"
  "github.com/kr/pretty"
  "math/rand"
  "fmt"
)

func TestMyTruncate(t *testing.T) {

  db := NewMysqlDB(&DataSource{ Host: "localhost", Database: "gotest", User: "gotest", Password: "gopass" })

  err := db.Connect()

  if err != nil {
    panic(err)
  }

  collections := db.Collections()

  for _, name := range collections {
    col := db.Collection(name)
    col.Truncate()
    if col.Count() != 0 {
      t.Errorf("Could not truncate '%s'.", name)
    }
  }


}

func TestMyAppend(t *testing.T) {

  db := NewMysqlDB(&DataSource{ Host: "localhost", Database: "gotest", User: "gotest", Password: "gopass" })

  err := db.Connect()

  if err != nil {
    panic(err)
  }

  col := db.Collection("people")

  col.Truncate()

  names := []string { "Juan", "José", "Pedro", "María", "Roberto", "Manuel", "Miguel" }

  for i := 0; i < len(names); i++ {
    col.Append(Item { "name": names[i] })
  }

  if col.Count() != len(names) {
    t.Error("Could not append all items.")
  }

}

func TestMyFind(t *testing.T) {

  db := NewMysqlDB(&DataSource{ Host: "localhost", Database: "gotest", User: "gotest", Password: "gopass" })

  err := db.Connect()

  if err != nil {
    panic(err)
  }

  col := db.Collection("people")

  result := col.Find(Where { "name": "José" })

  if result["name"] != "José" {
    t.Error("Could not find a recently appended item.")
  }

}

func TestMyDelete(t *testing.T) {
  db := NewMysqlDB(&DataSource{ Host: "localhost", Database: "gotest", User: "gotest", Password: "gopass" })

  err := db.Connect()

  if err != nil {
    panic(err)
  }

  col := db.Collection("people")

  col.Remove(Where { "name": "Juan" })

  result := col.Find(Where { "name": "Juan" })

  if len(result) > 0 {
    t.Error("Could not remove a recently appended item.")
  }
}

func TestMyUpdate(t *testing.T) {
  db := NewMysqlDB(&DataSource{ Host: "localhost", Database: "gotest", User: "gotest", Password: "gopass" })

  err := db.Connect()

  if err != nil {
    panic(err)
  }

  db.Use("test")

  col := db.Collection("people")

  col.Update(Where { "name": "José" }, Set { "name": "Joseph"})

  result := col.Find(Where { "name": "Joseph" })

  if len(result) == 0 {
    t.Error("Could not update a recently appended item.")
  }
}

func TestMyPopulate(t *testing.T) {
  var i int

  db := NewMysqlDB(&DataSource{ Host: "localhost", Database: "gotest", User: "gotest", Password: "gopass" })

  err := db.Connect()

  if err != nil {
    panic(err)
  }

  db.Use("test")

  places := []string { "Alaska", "Nebraska", "Alaska", "Acapulco", "Rome", "Singapore", "Alabama", "Cancún" }

  for i = 0; i < len(places); i++ {
    db.Collection("places").Append(Item {
      "code_id": i,
      "name": places[i],
    })
  }

  people := db.Collection("people").FindAll(
    Fields { "id", "name" },
  )

  for i = 0; i < len(people); i++ {
    person := people[i]

    // Has 5 children.
    for j := 0; j < 5; j++ {
      db.Collection("children").Append(Item {
        "name": fmt.Sprintf("%s's child %d", person["name"], j + 1),
        "parent_id": person["id"],
      })
    }

    // Lives in
    db.Collection("people").Update(
      Where { "id": person["id"] },
      Set { "place_code_id": int( rand.Float32() * float32(len(places)) ) },
    )

    // Has visited
    for k := 0; k < 3; k++ {
      place := db.Collection("places").Find(Where {
        "code_id": int( rand.Float32() * float32(len(places)) ),
      })
      db.Collection("visits").Append(Item {
        "place_id": place["id"],
        "person_id": person["id"],
      })
    }
  }

}


func TestMyRelation(t *testing.T) {
  db := NewMysqlDB(&DataSource{ Host: "localhost", Database: "gotest", User: "gotest", Password: "gopass" })

  err := db.Connect()

  if err != nil {
    panic(err)
  }

  col := db.Collection("people")

  result := col.FindAll(
    Relate {
      "lives_in": On {
        db.Collection("places"),
        Where { "code_id": "{place_code_id}" },
      },
    },
    RelateAll {
      "has_children": On {
        db.Collection("children"),
        Where { "parent_id": "{id}" },
      },
      "has_visited": On {
        db.Collection("visits"),
        Where { "person_id": "{id}" },
        Relate {
          "place": On {
            db.Collection("places"),
            Where { "id": "{place_id}" },
          },
        },
      },
    },
  )

  fmt.Printf("%# v\n", pretty.Formatter(result))
}