Function TestModelChangeModes

runtime/server/controller_test.go:105–276

(t *testing.T)

}

func TestModelChangeModes(t *testing.T) {
	// --- RESET MODE ---
	t.Run("reset", func(t *testing.T) {
		rt, instanceID := testruntime.NewInstance(t)
		createTableAsSelect(t, rt, instanceID, "duckdb", "foo", "SELECT 'US' AS country")

		// Create test resources m1, mv1
		testruntime.PutFiles(t, rt, instanceID, map[string]string{
			"m1.yaml": `
version: 1
type: model
sql: SELECT 'US' AS country
change_mode: reset
incremental: false
`,
			"mv1.yaml": `
type: metrics_view
version: 1
model: m1
dimensions:
- column: country
measures:
- expression: COUNT(*)
`,
		})
		testruntime.ReconcileParserAndWait(t, rt, instanceID)
		testruntime.RequireReconcileState(t, rt, instanceID, 3, 0, 0) // m1, mv1, project_settings

		ctrl, err := rt.Controller(context.Background(), instanceID)
		require.NoError(t, err)
		m1, err := ctrl.Get(context.Background(), &runtimev1.ResourceName{Kind: runtime.ResourceKindModel, Name: "m1"}, false)
		require.NoError(t, err)
		initialVersionM1 := int(m1.Meta.StateVersion)

		// Update m1.yaml with different SQL
		testruntime.PutFiles(t, rt, instanceID, map[string]string{
			"m1.yaml": `
version: 1
type: model
sql: SELECT 'CA' AS country
change_mode: reset
incremental: false
`,
		})
		testruntime.ReconcileParserAndWait(t, rt, instanceID)
		testruntime.RequireReconcileState(t, rt, instanceID, 3, 0, 0)

		// Verify that the model m1 was refreshed
		m1, err = ctrl.Get(context.Background(), &runtimev1.ResourceName{Kind: runtime.ResourceKindModel, Name: "m1"}, false)
		require.NoError(t, err)
		require.Greater(t, int(m1.Meta.StateVersion), initialVersionM1, "m1 was not refreshed in reset mode")
	})

	// --- MANUAL MODE ---
	t.Run("manual", func(t *testing.T) {
		rt, instanceID := testruntime.NewInstance(t)
		createTableAsSelect(t, rt, instanceID, "duckdb", "foo", "SELECT 'US' AS country")