wehub-resource-sync
53 lines
1.5 KiB
Ruby
53 lines
1.5 KiB
Ruby
#!/usr/bin/ruby
|
|
|
|
require 'active_record'
|
|
|
|
user = ARGV[0]
|
|
port = ARGV[1].to_i
|
|
|
|
ActiveRecord::Base.establish_connection(
|
|
adapter: 'postgresql',
|
|
host: 'localhost',
|
|
port: port,
|
|
database: 'postgres',
|
|
username: user,
|
|
password: 'password'
|
|
)
|
|
ActiveRecord::Base.logger = nil
|
|
|
|
conn = ActiveRecord::Base.connection
|
|
|
|
# SELECT pk from test_table (set up by bats setup())
|
|
pk = conn.execute("SELECT pk FROM test_table LIMIT 1").first["pk"].to_i
|
|
raise "expected pk=1, got #{pk}" unless pk == 1
|
|
|
|
# INSERT
|
|
conn.execute("INSERT INTO test_table VALUES (2)")
|
|
|
|
# COUNT
|
|
count = conn.execute("SELECT COUNT(*) FROM test_table").first["count"].to_i
|
|
raise "expected count=2, got #{count}" unless count == 2
|
|
|
|
# Dolt workflow
|
|
[
|
|
"DROP TABLE IF EXISTS test",
|
|
"CREATE TABLE test (pk int, value int, PRIMARY KEY(pk))",
|
|
"INSERT INTO test (pk, value) VALUES (0, 0)",
|
|
"SELECT dolt_add('-A')",
|
|
"SELECT dolt_commit('-m', 'added table test')",
|
|
"SELECT dolt_checkout('-b', 'mybranch')",
|
|
"INSERT INTO test VALUES (1, 1)",
|
|
"SELECT dolt_commit('-a', '-m', 'updated test')",
|
|
"SELECT dolt_checkout('main')",
|
|
"SELECT dolt_merge('mybranch')",
|
|
].each { |sql| conn.execute(sql) }
|
|
|
|
log_count = conn.execute("SELECT COUNT(*) FROM dolt_log").first["count"].to_i
|
|
raise "expected 4 dolt_log entries, got #{log_count}" unless log_count == 4
|
|
|
|
test_count = conn.execute("SELECT COUNT(*) FROM test").first["count"].to_i
|
|
raise "expected 2 rows in test, got #{test_count}" unless test_count == 2
|
|
|
|
ActiveRecord::Base.connection_pool.disconnect!
|
|
puts "ActiveRecord test passed"
|