コード例 #1
0
ファイル: pgdriver_test.go プロジェクト: 23inhouse/rebecca
func TestSaveUpdates(t *testing.T) {
	setup(t)

	p := &Person{Name: "John", Age: 31}
	if err := rebecca.Save(p); err != nil {
		t.Fatal(err)
	}

	expected := &Person{}
	if err := rebecca.Get(p.ID, expected); err != nil {
		t.Fatal(err)
	}

	expected.Name = "John Smith"
	if err := rebecca.Save(expected); err != nil {
		t.Fatal(err)
	}

	actual := &Person{}
	if err := rebecca.Get(p.ID, actual); err != nil {
		t.Fatal(err)
	}

	if !reflect.DeepEqual(actual, expected) {
		t.Errorf("Expected %+v to equal %+v", actual, expected)
	}
}
コード例 #2
0
ファイル: examples_test.go プロジェクト: 23inhouse/rebecca
func ExampleGet() {
	type Person struct {
		// ...
	}

	person := &Person{}
	if err := rebecca.Get(25, person); err != nil {
		panic(err)
	}
	// At this point person contains record with primary key equal to 25.
	fmt.Print(person)
}
コード例 #3
0
ファイル: pg_test.go プロジェクト: waterlink/rebecca
func TestSaveCreates(t *testing.T) {
	setup(t)

	expected := &Person{Name: "John", Age: 31}
	if err := rebecca.Save(expected); err != nil {
		t.Fatal(err)
	}

	actual := &Person{}
	if err := rebecca.Get(actual, expected.ID); err != nil {
		t.Fatal(err)
	}

	if !reflect.DeepEqual(actual, expected) {
		t.Errorf("Expected %+v to equal %+v", actual, expected)
	}
}
コード例 #4
0
ファイル: examples_test.go プロジェクト: 23inhouse/rebecca
func ExampleRemove() {
	type Person struct {
		// ...
	}

	// First lets find person with primary key = 25
	person := &Person{}
	if err := rebecca.Get(25, person); err != nil {
		panic(err)
	}

	// And then, lets remove it
	if err := rebecca.Remove(person); err != nil {
		panic(err)
	}

	// At this point person with primary key = 25 was removed from database.
}
コード例 #5
0
ファイル: transaction_test.go プロジェクト: waterlink/rebecca
func TestTransact(t *testing.T) {
	now := time.Now()
	var p *Post

	examples := map[string]struct {
		handler func(tx *rebecca.Transaction) error
		verify  func(err error) error
	}{

		"when handler succeeds": {
			handler: func(tx *rebecca.Transaction) error {
				p = &Post{Title: "Hello", Content: "World", CreatedAt: now}
				return tx.Save(p)
			},

			verify: func(err error) error {
				if err != nil {
					return fmt.Errorf("Expected handler to not fail, but got: %s", err)
				}

				actual := &Post{}
				if err := rebecca.Get(actual, p.ID); err != nil {
					return fmt.Errorf("Expected rebecca.Get(actual, %d) to not fail, but got: %s", p.ID, err)
				}

				if !actual.Equal(p) {
					return fmt.Errorf("Expected %#v to equal %#v", actual, p)
				}

				return nil
			},
		},

		"when handler fails": {
			handler: func(tx *rebecca.Transaction) error {
				p = &Post{Title: "Hello", Content: "World", CreatedAt: now}
				tx.Save(p)
				return errors.New("I have failed")
			},

			verify: func(err error) error {
				if err == nil {
					return errors.New("Expected handler to fail, but got nil")
				}

				if err.Error() != "I have failed" {
					return fmt.Errorf("Expected error to be 'I have failed', but got: '%s'", err)
				}

				actual := &Post{}
				if err := rebecca.Get(actual, p.ID); err == nil {
					return fmt.Errorf("Expected rebecca.Get(actual, %d) to fail, but got nil", p.ID)
				}

				return nil
			},
		},

		"when handler panics": {
			handler: func(tx *rebecca.Transaction) error {
				p = &Post{Title: "Hello", Content: "World", CreatedAt: now}
				tx.Save(p)
				panic("I have a panic!")
			},

			verify: func(err error) error {
				if err == nil {
					return errors.New("Expected handler to fail, but got nil")
				}

				if err.Error() != "I have a panic! (recovered)" {
					return fmt.Errorf("Expected error to be 'I have a panic! (recovered)', but got: '%s'", err)
				}

				actual := &Post{}
				if err := rebecca.Get(actual, p.ID); err == nil {
					return fmt.Errorf("Expected rebecca.Get(actual, %d) to fail, but got nil", p.ID)
				}

				return nil
			},
		},
	}

	for info, e := range examples {
		t.Log(info)
		p = nil
		if err := e.verify(rebecca.Transact(e.handler)); err != nil {
			t.Error(err)
		}
	}
}