wehub-resource-sync
50 lines
1.3 KiB
Ruby
50 lines
1.3 KiB
Ruby
#!/usr/bin/ruby
|
|
|
|
require 'sequel'
|
|
|
|
user = ARGV[0]
|
|
port = ARGV[1].to_i
|
|
|
|
DB = Sequel.connect(
|
|
adapter: 'postgres',
|
|
host: 'localhost',
|
|
port: port,
|
|
database: 'postgres',
|
|
user: user,
|
|
password: 'password'
|
|
)
|
|
|
|
# SELECT pk from test_table (set up by bats setup())
|
|
pk = DB["SELECT pk FROM test_table LIMIT 1"].first[:pk]
|
|
raise "expected pk=1, got #{pk}" unless pk == 1
|
|
|
|
# INSERT
|
|
DB.run("INSERT INTO test_table VALUES (2)")
|
|
|
|
# COUNT
|
|
count = DB["SELECT COUNT(*) FROM test_table"].first[:count]
|
|
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| DB.run(sql) }
|
|
|
|
log_count = DB["SELECT COUNT(*) FROM dolt_log"].first[:count]
|
|
raise "expected 4 dolt_log entries, got #{log_count}" unless log_count == 4
|
|
|
|
test_count = DB["SELECT COUNT(*) FROM test"].first[:count]
|
|
raise "expected 2 rows in test, got #{test_count}" unless test_count == 2
|
|
|
|
DB.disconnect
|
|
puts "Sequel test passed"
|