1use strict;
2use warnings;
3use lib "t/lib";
4use SQLiteTest qw/connect_ok/;
5use Test::More;
6use if -d ".git", "Test::FailWarnings";
7
8for my $quote ('', qw/' " ` []/) {
9	my ($begin_quote, $end_quote) = (substr($quote, 0, 1), substr($quote, -1, 1));
10	my $dbh = connect_ok( RaiseError => 1 );
11	ok $dbh->do(
12		"create table ${begin_quote}foo${end_quote} (${begin_quote}id${end_quote} integer primary key)"
13	);
14	my $sth = $dbh->primary_key_info(undef, undef, 'foo');
15	my $pk = $sth->fetchrow_hashref;
16	ok $pk->{TABLE_NAME} eq 'foo'; # dequoted
17	ok $pk->{COLUMN_NAME} eq 'id'; # dequoted
18
19	($pk) = $dbh->primary_key(undef, undef, 'foo');
20	ok $pk eq 'id';
21}
22
23{
24	my $dbh = connect_ok();
25	$dbh->do("create table foo (id integer primary key)");
26	$dbh->do("attach database ':memory:' as remote");
27	$dbh->do("create table remote.bar (name text, primary key(name))");
28	$dbh->do("create temporary table baz (tmp primary key)");
29
30	{
31		my $sth = $dbh->primary_key_info(undef, undef, 'foo');
32		my @pk_info;
33		while(my $row = $sth->fetchrow_hashref) { push @pk_info, $row };
34		is @pk_info => 1, "found 1 pk in a table";
35		is $pk_info[0]{TABLE_SCHEM} => 'main', "scheme is correct";
36		is $pk_info[0]{COLUMN_NAME} => 'id', "pk name is correct";
37	}
38
39	{
40		my $sth = $dbh->primary_key_info(undef, 'main', undef);
41		my @pk_info;
42		while(my $row = $sth->fetchrow_hashref) { push @pk_info, $row };
43		is @pk_info => 1, "found 1 pk in a table";
44		is $pk_info[0]{TABLE_SCHEM} => 'main', "scheme is correct";
45		is $pk_info[0]{COLUMN_NAME} => 'id', "pk name is correct";
46	}
47
48	{
49		my $sth = $dbh->primary_key_info(undef, undef, 'bar');
50		my @pk_info;
51		while(my $row = $sth->fetchrow_hashref) { push @pk_info, $row };
52		is @pk_info => 1, "found 1 pk in an attached table";
53		is $pk_info[0]{TABLE_SCHEM} => 'remote', "scheme is correct";
54		is $pk_info[0]{COLUMN_NAME} => 'name', "pk name is correct";
55	}
56
57	{
58		my $sth = $dbh->primary_key_info(undef, 'remote', undef);
59		my @pk_info;
60		while(my $row = $sth->fetchrow_hashref) { push @pk_info, $row };
61		is @pk_info => 1, "found 1 pk in an attached table";
62		is $pk_info[0]{TABLE_SCHEM} => 'remote', "scheme is correct";
63		is $pk_info[0]{COLUMN_NAME} => 'name', "pk name is correct";
64	}
65
66	{
67		my $sth = $dbh->primary_key_info(undef, 'temp', undef);
68		my @pk_info;
69		while(my $row = $sth->fetchrow_hashref) { push @pk_info, $row };
70		is @pk_info => 1, "found 1 pk in a table";
71		is $pk_info[0]{TABLE_SCHEM} => 'temp', "scheme is correct";
72		is $pk_info[0]{COLUMN_NAME} => 'tmp', "pk name is correct";
73	}
74
75	{
76		my $sth = $dbh->primary_key_info(undef, undef, 'baz');
77		my @pk_info;
78		while(my $row = $sth->fetchrow_hashref) { push @pk_info, $row };
79		is @pk_info => 1, "found 1 pk in an attached table";
80		is $pk_info[0]{TABLE_SCHEM} => 'temp', "scheme is correct";
81		is $pk_info[0]{COLUMN_NAME} => 'tmp', "pk name is correct";
82	}
83}
84
85{
86	my $dbh = connect_ok();
87	$dbh->do("create table foo (id integer, text text, primary key (id))");
88
89	{
90		my $sth = $dbh->primary_key_info(undef, undef, 'foo');
91		my @pk_info;
92		while(my $row = $sth->fetchrow_hashref) { push @pk_info, $row };
93		is @pk_info => 1, "found 1 pk in a table";
94		is $pk_info[0]{TABLE_SCHEM} => 'main', "scheme is correct";
95		is $pk_info[0]{COLUMN_NAME} => 'id', "pk name is correct";
96		is $pk_info[0]{PK_NAME} => 'PRIMARY KEY', "pk name is correct";
97	}
98}
99
100{
101	my $dbh = connect_ok();
102	$dbh->do("create table foo (id integer, text text, constraint bar primary key (id))");
103
104	{
105		my $sth = $dbh->primary_key_info(undef, undef, 'foo');
106		my @pk_info;
107		while(my $row = $sth->fetchrow_hashref) { push @pk_info, $row };
108		is @pk_info => 1, "found 1 pk in a table";
109		is $pk_info[0]{TABLE_SCHEM} => 'main', "scheme is correct";
110		is $pk_info[0]{COLUMN_NAME} => 'id', "pk name is correct";
111		is $pk_info[0]{PK_NAME} => 'bar', "pk name is correct";
112	}
113}
114
115done_testing;
116