add testing for update via condition bean (go-xorm#918)

y0unghe · Apr 29, 2018 · d1e0fe9 · d1e0fe9
1 parent 44623fc
commit d1e0fe9
Showing 1 changed file with 54 additions and 0 deletions.
diff --git a/session_update_test.go b/session_update_test.go
@@ -1229,3 +1229,57 @@ func TestUpdateMapContent(t *testing.T) {
 	assert.EqualValues(t, true, c3.IsMan)
 	assert.EqualValues(t, 1, c3.Gender)
 }
+
+func TestUpdateCondiBean(t *testing.T) {
+	type NeedUpdateBean struct {
+		Id   int64
+		Name string
+	}
+
+	type NeedUpdateCondiBean struct {
+		Name string
+	}
+
+	assert.NoError(t, prepareEngine())
+	assertSync(t, new(NeedUpdateBean))
+
+	cnt, err := testEngine.Insert(&NeedUpdateBean{
+		Name: "name1",
+	})
+	assert.NoError(t, err)
+	assert.EqualValues(t, 1, cnt)
+
+	has, err := testEngine.Exist(&NeedUpdateBean{
+		Name: "name1",
+	})
+	assert.NoError(t, err)
+	assert.True(t, has)
+
+	cnt, err = testEngine.Update(&NeedUpdateBean{
+		Name: "name2",
+	}, &NeedUpdateCondiBean{
+		Name: "name1",
+	})
+	assert.NoError(t, err)
+	assert.EqualValues(t, 1, cnt)
+
+	has, err = testEngine.Exist(&NeedUpdateBean{
+		Name: "name2",
+	})
+	assert.NoError(t, err)
+	assert.True(t, has)
+
+	cnt, err = testEngine.Update(&NeedUpdateBean{
+		Name: "name1",
+	}, NeedUpdateCondiBean{
+		Name: "name2",
+	})
+	assert.NoError(t, err)
+	assert.EqualValues(t, 1, cnt)
+
+	has, err = testEngine.Exist(&NeedUpdateBean{
+		Name: "name1",
+	})
+	assert.NoError(t, err)
+	assert.True(t, has)
+}