Updated for comments

test/helpers/ghostferry_helper.rb

@@ -31,7 +31,6 @@ class Ghostferry
 
     # Keep these in sync with integrationferry.go
     ENV_KEY_PORT = "GHOSTFERRY_INTEGRATION_PORT"
-    MAX_MESSAGE_SIZE = 256
 
     module Status
       # This should be in sync with integrationferry.go
@@ -256,9 +255,8 @@ def run(resuming_state = nil)
       raise @server_last_error unless @server_last_error.nil?
     end
 
-    # When using this method, you need to call it within the block of
-    # GhostferryIntegration::TestCase#with_state_cleanup to ensure that the
-    # integration server is shutdown properly.
+    # When using this method, you need to ensure that the datawriter has been
+    # stopped properly (if you're using stop_datawriter_during_cutover).
     def run_expecting_interrupt(resuming_state = nil)
       run(resuming_state)
     rescue GhostferryExitFailure

test/integration/interrupt_resume_test.rb

@@ -3,9 +3,32 @@
 require "json"
 
 class InterruptResumeTest < GhostferryTestCase
-  def test_interrupt_resume_with_writes_to_source
+  def setup
     seed_simple_database_with_single_table
+  end
+
+  def test_interrupt_resume_without_writes_to_source_to_check_target_state_when_interrupted
+    ghostferry = new_ghostferry(MINIMAL_GHOSTFERRY)
+
+    # Writes one batch
+    ghostferry.on_status(Ghostferry::Status::AFTER_ROW_COPY) do
+      ghostferry.send_signal("TERM")
+    end
+
+    dumped_state = ghostferry.run_expecting_interrupt
+    assert_basic_fields_exist_in_dumped_state(dumped_state)
+
+    result = target_db.query("SELECT COUNT(*) AS cnt FROM #{DEFAULT_FULL_TABLE_NAME}")
+    count = result.first["cnt"]
+    assert_equal 200, count
 
+    result = target_db.query("SELECT MAX(id) AS max_id FROM #{DEFAULT_FULL_TABLE_NAME}")
+    last_successful_id = result.first["max_id"]
+    assert last_successful_id > 0
+    assert_equal last_successful_id, dumped_state["LastSuccessfulPrimaryKeys"]["#{DEFAULT_DB}.#{DEFAULT_TABLE}"]
+  end
+
+  def test_interrupt_resume_with_writes_to_source
     # Start a ghostferry run expecting it to be interrupted.
     datawriter = new_source_datawriter
     ghostferry = new_ghostferry(MINIMAL_GHOSTFERRY)
@@ -36,8 +59,6 @@ def test_interrupt_resume_with_writes_to_source
   end
 
   def test_interrupt_resume_when_table_has_completed
-    seed_simple_database_with_single_table
-
     # Start a run of Ghostferry expecting to be interrupted
     datawriter = new_source_datawriter
     ghostferry = new_ghostferry(MINIMAL_GHOSTFERRY)
@@ -58,6 +79,9 @@ def test_interrupt_resume_when_table_has_completed
 
     start_datawriter_with_ghostferry(datawriter, ghostferry)
     stop_datawriter_during_cutover(datawriter, ghostferry)
+
     ghostferry.run(dumped_state)
+
+    assert_test_table_is_identical
   end
 end

test/lib/go/integrationferry/integrationferry.go → test/lib/go/integrationferry.go

@@ -1,4 +1,4 @@
-package integrationferry
+package main
 
 import (
 	"encoding/json"
@@ -12,13 +12,13 @@ import (
 
 	"github.com/Shopify/ghostferry"
 	"github.com/Shopify/ghostferry/testhelpers"
+	"github.com/sirupsen/logrus"
 )
 
 const (
 	// These should be kept in sync with ghostferry.rb
-	portEnvName    string        = "GHOSTFERRY_INTEGRATION_PORT"
-	timeout        time.Duration = 30 * time.Second
-	maxMessageSize int           = 256
+	portEnvName string        = "GHOSTFERRY_INTEGRATION_PORT"
+	timeout     time.Duration = 30 * time.Second
 )
 
 const (
@@ -194,3 +194,24 @@ func NewStandardConfig() (*ghostferry.Config, error) {
 
 	return config, config.ValidateConfig()
 }
+
+func main() {
+	logrus.SetFormatter(&logrus.JSONFormatter{})
+	logrus.SetLevel(logrus.DebugLevel)
+
+	config, err := NewStandardConfig()
+	if err != nil {
+		panic(err)
+	}
+
+	f := &IntegrationFerry{
+		Ferry: &ghostferry.Ferry{
+			Config: config,
+		},
+	}
+
+	err = f.Main()
+	if err != nil {
+		panic(err)
+	}
+}

test/test_helper.rb

@@ -19,7 +19,7 @@ class GhostferryTestCase < Minitest::Test
   include DbHelper
   include DataWriterHelper
 
-  MINIMAL_GHOSTFERRY = "minimal.go"
+  MINIMAL_GHOSTFERRY = "integrationferry.go"
 
   def new_ghostferry(filename)
     # Transform path to something ruby understands
@@ -83,13 +83,13 @@ def assert_test_table_is_identical
     assert target[DEFAULT_FULL_TABLE_NAME][:row_count] > 0
 
     assert_equal(
-      source[DEFAULT_FULL_TABLE_NAME][:checksum],
-      target[DEFAULT_FULL_TABLE_NAME][:checksum],
+      source[DEFAULT_FULL_TABLE_NAME][:row_count],
+      target[DEFAULT_FULL_TABLE_NAME][:row_count],
     )
 
     assert_equal(
-      source[DEFAULT_FULL_TABLE_NAME][:sample_row],
-      target[DEFAULT_FULL_TABLE_NAME][:sample_row],
+      source[DEFAULT_FULL_TABLE_NAME][:checksum],
+      target[DEFAULT_FULL_TABLE_NAME][:checksum],
     )
   end